get_device_test.go 1.5 KB
package device

import (
	"github.com/go-pg/pg/v10"
	"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 deviceId int64
	BeforeEach(func() {
		_, err := pG.DB.QueryOne(
			pg.Scan(&deviceId),
			"INSERT INTO devices (device_id, company_id, org_id, device_code, device_name, device_model, device_type, brand, device_status, risk_level, created_at, updated_at, deleted_at, work_station) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) RETURNING device_id",
			"testDeviceId", "testCompanyId", "testOrgId", "testDeviceCode", "testDeviceName", "testDeviceModel", "testDeviceType", "testBrand", "testDeviceStatus", "testRiskLevel", "testCreatedAt", "testUpdatedAt", "testDeletedAt", "testWorkStation")
		Expect(err).NotTo(HaveOccurred())
	})
	Describe("根据deviceId参数返回设备", func() {
		Context("传入有效的deviceId", func() {
			It("返回设备数据", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				httpExpect.GET("/devices/{deviceId}").
					Expect().
					Status(http.StatusOK).
					JSON().
					Object().
					ContainsKey("code").ValueEqual("code", 0).
					ContainsKey("msg").ValueEqual("msg", "ok").
					ContainsKey("data").Value("data").Object()
			})
		})
	})
	AfterEach(func() {
		_, err := pG.DB.Exec("DELETE FROM devices WHERE true")
		Expect(err).NotTo(HaveOccurred())
	})
})