get_project_belong_test.go 1.2 KB
package project_belong

import (
	"fmt"
	"net/http"
	"strconv"

	"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 projectBelongId int64
	BeforeEach(func() {
		_, err := pG.DB.QueryOne(
			pg.Scan(&projectBelongId),
			"INSERT INTO project_belongs (project_belong_name, company_id) VALUES (?, ?) RETURNING id",
			"能力展示", 101)
		Expect(err).NotTo(HaveOccurred())
	})
	Describe("根据projectBelongId参数返回项目归属", func() {
		Context("传入有效的projectBelongId", func() {
			It("返回项目归属数据", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				httpExpect.GET(fmt.Sprintf("/project-belongs/%s", strconv.FormatInt(projectBelongId, 10))).
					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 project_belongs WHERE true")
		Expect(err).NotTo(HaveOccurred())
	})
})