update_device_collection_test.go 1.7 KB
package device_collection

import (
	"net/http"

	"github.com/gavv/httpexpect"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
	pG "gitlab.fjmaimaimai.com/allied-creation/allied-creation-manufacture/pkg/infrastructure/pg"
)

var _ = Describe("更新", func() {
	var deviceCollectionId int64
	BeforeEach(func() {
		_, err := pG.DB.QueryOne(
			pg.Scan(&deviceCollectionId),
			"INSERT INTO device_collections (device_collection_id, work_shop_name, device_sn, startup_status, com_status, values) VALUES (?, ?, ?, ?, ?, ?) RETURNING device_collection_id",
			"testDeviceCollectionId", "testWorkShopName", "testDeviceSn", "testStartupStatus", "testComStatus", "testValues")
		Expect(err).NotTo(HaveOccurred())
	})
	Describe("提交数据更新", func() {
		Context("提交正确的设备采集数据数据", func() {
			It("返回更新后的设备采集数据数据", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				body := map[string]interface{}{
					"deviceCollectionId": "int64",
					"workShopName":       "string",
					"startupStatus":      "int64",
					"deviceSn":           "string",
					"comStatus":          "int64",
					"values":             "string",
				}
				httpExpect.PUT("/device-collections/{Id}").
					WithJSON(body).
					Expect().
					Status(http.StatusOK).
					JSON().
					Object().
					ContainsKey("code").ValueEqual("code", 0).
					ContainsKey("msg").ValueEqual("msg", "ok").
					ContainsKey("data").Value("data").Object().
					ContainsKey("deviceCollectionId").ValueEqual("deviceCollectionId", deviceCollectionId)
			})
		})
	})
	AfterEach(func() {
		_, err := pG.DB.Exec("DELETE FROM device_collections WHERE true")
		Expect(err).NotTo(HaveOccurred())
	})
})