list_user_test.go 1.3 KB
package user

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/mmm-go-pp/partner01/pkg/infrastructure/pg"
)

var _ = Describe("返回列表", func() {
	var userId int64
	BeforeEach(func() {
		_, err := pG.DB.QueryOne(
			pg.Scan(&userId),
			"INSERT INTO users (user_id, user_type, permissions, company_id, user_info, partner_info, status, admin_type) VALUES (?, ?, ?, ?, ?, ?, ?, ?) RETURNING user_id",
			1, 1, pg.Array([]int{}), 1, `{}`, `{}`, 1, 1)
		Expect(err).NotTo(HaveOccurred())
	})
	Describe("根据参数返回用户实体列表", func() {
		Context("传入有效的参数", func() {
			It("返回用户实体数据列表", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				httpExpect.GET("/users/").
					WithQuery("offset", 0).
					WithQuery("limit", 10).
					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("users").Value("users").Array()
			})
		})
	})
	AfterEach(func() {
		_, err := pG.DB.Exec("DELETE FROM users WHERE true")
		Expect(err).NotTo(HaveOccurred())
	})
})