update_unit_conversion_test.go 2.2 KB
package unit_conversion

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 unitConversionId int64
	BeforeEach(func() {
		_, err := pG.DB.QueryOne(
			pg.Scan(&unitConversionId),
			"INSERT INTO unit_conversions (unit_conversion_id, company_id, org_id, work_station, material, from_unit_quantity, to_unit_quantity, intelligent_weighing_flag, created_at, updated_at, deleted_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) RETURNING unit_conversion_id",
			"testUnitConversionId", "testCompanyId", "testOrgId", "testWorkStation", "testMaterial", "testFromUnitQuantity", "testToUnitQuantity", "testIntelligentWeighingFlag", "testCreatedAt", "testUpdatedAt", "testDeletedAt")
		Expect(err).NotTo(HaveOccurred())
	})
	Describe("提交数据更新单位换算服务", func() {
		Context("提交正确的单位换算数据", func() {
			It("返回更新后的单位换算数据", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				body := map[string]interface{}{
					"companyId":               "int",
					"orgId":                   "int",
					"workshopId":              "int",
					"lineId":                  "int",
					"sectionId":               "int",
					"materialName":            "string",
					"materialCategory":        "string",
					"quantity":                "float64",
					"unit":                    "string",
					"unitWeight":              "float64",
					"weight":                  "float64",
					"intelligentWeighingFlag": "boolean",
				}
				httpExpect.PUT("/unit-conversions/{unitConversionId}").
					WithJSON(body).
					Expect().
					Status(http.StatusOK).
					JSON().
					Object().
					ContainsKey("code").ValueEqual("code", 0).
					ContainsKey("msg").ValueEqual("msg", "ok").
					ContainsKey("data").Value("data").Object().
					ContainsKey("unitConversionId").ValueEqual("unitConversionId", unitConversionId)
			})
		})
	})
	AfterEach(func() {
		_, err := pG.DB.Exec("DELETE FROM unit_conversions WHERE true")
		Expect(err).NotTo(HaveOccurred())
	})
})