package company

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-user/pkg/infrastructure/pg"
)

var _ = Describe("更新自定义菜单", func() {
	var companyId int64
	BeforeEach(func() {
		_, err := pG.DB.QueryOne(
			pg.Scan(&companyId),
			"INSERT INTO company (company_id, company_config, company_info, status, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?) RETURNING company_id",
			"testCompanyId", "testCompanyConfig", "testCompanyInfo", "testStatus", "testCreatedAt", "testUpdatedAt")
		Expect(err).NotTo(HaveOccurred())
	})
	Describe("更新自定义菜单", func() {
		Context("", func() {
			It("", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				body := map[string]interface{}{
					"menuId":    "int64",
					"menuName":  "string",
					"menuAlias": "string",
					"sort":      "int",
				}
				httpExpect.PUT("/company/{companyId}/customize-menus").
					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 company WHERE true")
		Expect(err).NotTo(HaveOccurred())
	})
})