update_device_test.go
1.9 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
46
47
48
49
50
51
52
53
54
55
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)
body := map[string]interface{}{
"deviceCode": "string",
"deviceName": "string",
"deviceModel": "string",
"deviceType": "string",
"workshopId": "int",
"lineId": "int",
"sectionId": "int",
"brand": "string",
"deviceStatus": "int",
"riskLevel": "int",
}
httpExpect.PUT("/devices/{deviceId}").
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").ValueEqual("deviceId", deviceId)
})
})
})
AfterEach(func() {
_, err := pG.DB.Exec("DELETE FROM devices WHERE true")
Expect(err).NotTo(HaveOccurred())
})
})