create_device_test.go 1.3 KB
package device

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() {
	Describe("提交数据创建设备服务", func() {
		Context("提交正确的新设备数据", func() {
			It("返回设备数据", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				body := map[string]interface{}{
					"companyId":    "int",
					"orgId":        "int",
					"deviceCode":   "string",
					"deviceName":   "string",
					"deviceModel":  "string",
					"deviceType":   "string",
					"workshopId":   "int",
					"lineId":       "int",
					"sectionId":    "int",
					"brand":        "string",
					"deviceStatus": "int",
					"riskLevel":    "int",
				}
				httpExpect.POST("/devices/").
					WithJSON(body).
					Expect().
					Status(http.StatusOK).
					JSON().
					Object().
					ContainsKey("code").ValueEqual("code", 0).
					ContainsKey("msg").ValueEqual("msg", "ok").
					ContainsKey("data").Value("data").Object().
					ContainsKey("deviceId").ValueNotEqual("deviceId", BeZero())
			})
		})
	})
	AfterEach(func() {
		_, err := pG.DB.Exec("DELETE FROM devices WHERE true")
		Expect(err).NotTo(HaveOccurred())
	})
})