list_device_test.go
1.7 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
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("根据参数返回设备列表", func() {
Context("传入有效的参数", func() {
It("返回设备数据列表", func() {
httpExpect := httpexpect.New(GinkgoT(), server.URL)
httpExpect.GET("/devices/").
WithQuery("offset", "int").
WithQuery("limit", "int").
Expect().
Status(http.StatusOK).
JSON().
Object().
ContainsKey("code").ValueEqual("code", 0).
ContainsKey("msg").ValueEqual("msg", "ok").
ContainsKey("data").Value("data").Object().
ContainsKey("count").ValueEqual("count", 1).
ContainsKey("devices").Value("devices").Array()
})
})
})
AfterEach(func() {
_, err := pG.DB.Exec("DELETE FROM devices WHERE true")
Expect(err).NotTo(HaveOccurred())
})
})