update_cooperator_test.go 3.0 KB
package user

import (
	"github.com/go-pg/pg/v10"
	"net/http"
	"time"

	"github.com/gavv/httpexpect"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
	pG "gitlab.fjmaimaimai.com/allied-creation/allied-creation-user/pkg/infrastructure/pg"
)

var _ = Describe("更新共创用户", func() {
	var userId int64
	BeforeEach(func() {
		var err error
		_, err = pG.DB.QueryOne(
			pg.Scan(&userId),
			"INSERT INTO \"users\".\"user_base\"(\"user_base_id\", \"user_info\", \"account\", \"password\", \"im\", \"related_user\", \"status\", \"created_at\", \"updated_at\") VALUES (999, '{\"phone\": \"phone\", \"userName\": \"string\"}', 'phone', 'string', '{\"accid\": \"\", \"imToken\": \"\", \"csAccountId\": \"\"}', '{999}', 1, '2021-07-24 10:16:17.680805+08', '2021-07-24 10:16:17.680805+08') RETURNING user_base_id;",
		)
		Expect(err).NotTo(HaveOccurred())

		_, err = pG.DB.QueryOne(
			pg.Scan(&userId),
			"INSERT INTO \"users\".\"user\"(\"user_id\", \"company_id\", \"user_base_id\", \"user_type\", \"user_code\", \"organization_id\", \"department_id\", \"favorite_menus\", \"cooperation_info\", \"enable_status\", \"ext\", \"created_at\", \"updated_at\") VALUES (999, 5, 999, 1025, 'ADMIN01', 5, 5, '[]', '{\"cooperationCompany\": \"\", \"cooperationDeadline\": \"0001-01-01T00:00:00Z\"}', 1, '{\"phone\": \"18860183031\", \"depName\": \"string1\", \"orgName\": \"string1\"}', '2021-07-24 10:16:17.680805+08', '2021-07-24 10:16:17.680805+08');\n",
		)
		Expect(err).NotTo(HaveOccurred())

		_, err = pG.DB.QueryOne(
			pg.Scan(&userId),
			"INSERT INTO users.org (org_id,company_id,created_at,updated_at,deleted_at,org_code,org_name,ext,org_status,is_org,parent_id,parent_path) VALUES (5,999,'2021-07-26 08:06:29.3101584+00:00:00','2021-07-26 08:06:29.3101584+00:00:00','0001-01-01 00:00:00+00:00:00','ENTERPRISE01','string1','{}',1,1,0,'');",
		)
		Expect(err).NotTo(HaveOccurred())
	})
	Describe("更新共创用户", func() {
		Context("", func() {
			It("", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				body := map[string]interface{}{
					"cooperationCompany":  "string",
					"cooperationDeadline": time.Now(),
					"email":               "string",
					"enableStatus":        1,
					"userCode":            "string",
					"userName":            "string",
					"avatar":              "string",
					"orgId":               5,
					"phone":               "phone",
				}
				httpExpect.PUT("/user/cooperator/999").
					WithJSON(body).
					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 users.user_base WHERE user_base_id = 999")
		Expect(err).NotTo(HaveOccurred())
		_, err = pG.DB.Exec(`DELETE FROM users."user" WHERE user_id = 999`)
		Expect(err).NotTo(HaveOccurred())
		_, err = pG.DB.Exec(`DELETE FROM users."org" WHERE org_id = 999`)
		Expect(err).NotTo(HaveOccurred())
	})
})