package dividends_estimate

import (
	"net/http"

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

var _ = Describe("更新分红预算服务", func() {
	var dividendsEstimateId int64
	BeforeEach(func() {
		_, err := pG.DB.QueryOne(
			pg.Scan(&dividendsEstimateId),
			"INSERT INTO dividends_estimates (dividends_estimate_id, dividends_account_status, dividends_amount, dividends_estimate_order_number, dividends_estimate_time, dividends_participate_type, dividends_type, order_or_returned_order_num, cooperation_project_number, dividends_user, org, company, operator, operate_time, created_at, deleted_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) RETURNING dividends_estimate_id",
			"testDividendsEstimateId", "testDividendsAccountStatus", "testDividendsAmount", "testDividendsEstimateOrderNumber", "testDividendsEstimateTime", "testDividendsParticipateType", "testDividendsType", "testOrderOrReturnedOrderNum", "testCooperationProjectNumber", "testDividendsUser", "testOrg", "testCompany", "testOperator", "testOperateTime", "testCreatedAt", "testDeletedAt", "testUpdatedAt")
		Expect(err).NotTo(HaveOccurred())
	})
	Describe("提交数据更新分红预算服务", func() {
		Context("提交正确的分红预算实体数据", func() {
			It("返回更新后的分红预算实体数据", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				body := map[string]interface{}{
					"companyId": "int64",
					"orgId":     "int64",
					"userId":    "int64",
				}
				httpExpect.PUT("/dividends-estimates/{dividendsEstimateId}").
					WithJSON(body).
					Expect().
					Status(http.StatusOK).
					JSON().
					Object().
					ContainsKey("code").ValueEqual("code", 0).
					ContainsKey("msg").ValueEqual("msg", "ok").
					ContainsKey("data").Value("data").Object().
					ContainsKey("dividendsEstimateId").ValueEqual("dividendsEstimateId", dividendsEstimateId)
			})
		})
	})
	AfterEach(func() {
		_, err := pG.DB.Exec("DELETE FROM dividends_estimates WHERE true")
		Expect(err).NotTo(HaveOccurred())
	})
})