update_employee_test.go 1.6 KB
package employee

import (
	"net/http"

	"github.com/gavv/httpexpect"
	"github.com/go-pg/pg"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
	pG "gitlab.fjmaimaimai.com/linmadan/mmm-worth/pkg/infrastructure/pg"
)

var _ = Describe("更新员工", func() {
	var employeeId int64
	BeforeEach(func() {
		_, err := pG.DB.QueryOne(
			pg.Scan(&employeeId),
			"INSERT INTO employees (id, company_id, uid, employee_name, employee_account, employee_avatar_url, su_money, status, permissions) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?) RETURNING id",
			1, 101, 2499036607974745088, "employee_name", "employee_account", "employee_avatar_url", 1000.00, 1, pg.Array([]int{1, 3}))
		Expect(err).NotTo(HaveOccurred())
	})
	Describe("提交数据更新员工", func() {
		Context("提交正确的员工数据", func() {
			It("返回更新后的员工数据", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				body := map[string]interface{}{
					"employeeName":      "employeeName",
					"employeeAccount":   "employeeAccount",
					"employeeAvatarUrl": "employeeAvatarUrl",
					"permissions":       []int{1, 2},
				}
				httpExpect.PUT("/employees/2499036607974745088").
					WithJSON(body).
					Expect().
					Status(http.StatusOK).
					JSON().
					Object().
					ContainsKey("code").ValueEqual("code", 0).
					ContainsKey("msg").ValueEqual("msg", "ok").
					ContainsKey("data").Value("data").Object().
					ContainsKey("employeeId").ValueEqual("employeeId", employeeId)
			})
		})
	})
	AfterEach(func() {
		_, err := pG.DB.Exec("DELETE FROM employees WHERE true")
		Expect(err).NotTo(HaveOccurred())
	})
})