rob_task_test.go 5.4 KB
package task

import (
	"github.com/go-pg/pg"
	"gitlab.fjmaimaimai.com/linmadan/mmm-worth/pkg/domain"
	"net/http"
	"time"

	"github.com/gavv/httpexpect"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
	pG "gitlab.fjmaimaimai.com/linmadan/mmm-worth/pkg/infrastructure/pg"
)

var _ = Describe("对任务进行抢单", func() {
	Describe("对任务进行抢单", func() {
		Context("领取人对已发布的抢单类型任务进行抢单", func() {
			BeforeEach(func() {
				dayAfter, _ := time.ParseDuration("72h")
				_, err := pG.DB.QueryOne(
					pg.Scan(),
					"INSERT INTO tasks (id, company_id, task_name, task_type, sponsor, task_status, reference_resource, project_belong, customer_values, task_nature, su_money, acceptance_standard, task_description, task_picture_urls, is_reward_take, participators, task_percentage, solve_report, solve_picture_urls, receiver_uid, create_time, release_time) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
					1, 101, "抢单任务1", 1, &domain.EmployeeInfo{
						Uid: 2499036607974745088,
					}, 2, "null", 1, pg.Array([]int{1, 2, 3, 4}), 1, 1000.00, "验收标准1", "任务描述1", pg.Array([]string{}), true, []*domain.EmployeeInfo{
						{
							Uid: 2499036607974745077,
						},
						{
							Uid: 2499036607974745066,
						},
					}, "null", "", pg.Array([]string{}), 0, time.Now(), time.Now().Add(dayAfter))
				Expect(err).NotTo(HaveOccurred())
				_, err1 := pG.DB.QueryOne(
					pg.Scan(),
					"INSERT INTO employees (id, company_id, uid, employee_name, employee_account, su_money) VALUES (?, ?, ?, ?, ?, ?)",
					1, 101, 2499036607974745088, "testEmployeeName", "testEmployeeAccount", 0)
				Expect(err1).NotTo(HaveOccurred())
				_, err2 := pG.DB.QueryOne(
					pg.Scan(),
					"INSERT INTO employees (id, company_id, uid, employee_name, employee_account, su_money) VALUES (?, ?, ?, ?, ?, ?)",
					2, 101, 2499036607974745099, "testEmployeeName", "testEmployeeAccount", 0)
				Expect(err2).NotTo(HaveOccurred())
			})
			It("抢单成功", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				body := map[string]interface{}{
					"receiver": 2499036607974745099,
				}
				httpExpect.POST("/tasks/1/rob").
					WithJSON(body).
					Expect().
					Status(http.StatusOK).
					JSON().
					Object().
					ContainsKey("code").ValueEqual("code", 0).
					ContainsKey("msg").ValueEqual("msg", "ok").
					ContainsKey("data").Value("data").Object().
					ContainsKey("taskStatus").ValueEqual("taskStatus", 7).
					ContainsKey("robInfo").Value("robInfo").Object().
					ContainsKey("receiver").Value("receiver").Object().
					ContainsKey("uid").ValueEqual("uid", 2499036607974745099)
			})
		})
		Context("领取人对已经被领取的抢单类型任务进行抢单", func() {
			BeforeEach(func() {
				dayAfter, _ := time.ParseDuration("72h")
				_, err := pG.DB.QueryOne(
					pg.Scan(),
					"INSERT INTO tasks (id, company_id, task_name, task_type, sponsor, task_status, reference_resource, project_belong, customer_values, task_nature, su_money, acceptance_standard, task_description, task_picture_urls, is_reward_take, participators, task_percentage, solve_report, solve_picture_urls, receiver_uid, create_time, release_time) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
					1, 101, "抢单任务1", 1, &domain.EmployeeInfo{
						Uid: 2499036607974745088,
					}, 2, "null", 1, pg.Array([]int{1, 2, 3, 4}), 1, 1000.00, "验收标准1", "任务描述1", pg.Array([]string{}), true, []*domain.EmployeeInfo{
						{
							Uid: 2499036607974745077,
						},
						{
							Uid: 2499036607974745066,
						},
					}, "null", "", pg.Array([]string{}), 0, time.Now(), time.Now().Add(dayAfter))
				Expect(err).NotTo(HaveOccurred())
				_, err1 := pG.DB.QueryOne(
					pg.Scan(),
					"INSERT INTO employees (id, company_id, uid, employee_name, employee_account, su_money) VALUES (?, ?, ?, ?, ?, ?)",
					1, 101, 2499036607974745088, "testEmployeeName", "testEmployeeAccount", 0)
				Expect(err1).NotTo(HaveOccurred())
				_, err2 := pG.DB.QueryOne(
					pg.Scan(),
					"INSERT INTO employees (id, company_id, uid, employee_name, employee_account, su_money) VALUES (?, ?, ?, ?, ?, ?)",
					2, 101, 2499036607974745099, "testEmployeeName", "testEmployeeAccount", 0)
				Expect(err2).NotTo(HaveOccurred())
				_, err3 := pG.DB.QueryOne(
					pg.Scan(),
					"INSERT INTO rob_infos (id, task_id) VALUES (?, ?)",
					1, 1)
				Expect(err3).NotTo(HaveOccurred())
			})
			It("抢单失败", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				body := map[string]interface{}{
					"receiver": 2499036607974745099,
				}
				httpExpect.POST("/tasks/1/rob").
					WithJSON(body).
					Expect().
					Status(http.StatusOK).
					JSON().
					Object().
					ContainsKey("code").ValueEqual("code", 501).
					ContainsKey("msg").ValueEqual("msg", "内部服务出错:抢单失败,任务可能已经被人领取")
			})
		})
	})
	AfterEach(func() {
		_, err := pG.DB.Exec("DELETE FROM tasks WHERE true")
		Expect(err).NotTo(HaveOccurred())
		_, err1 := pG.DB.Exec("DELETE FROM bid_infos WHERE true")
		Expect(err1).NotTo(HaveOccurred())
		_, err2 := pG.DB.Exec("DELETE FROM bidder_infos WHERE true")
		Expect(err2).NotTo(HaveOccurred())
		_, err3 := pG.DB.Exec("DELETE FROM employees WHERE true")
		Expect(err3).NotTo(HaveOccurred())
		_, err4 := pG.DB.Exec("DELETE FROM rob_infos WHERE true")
		Expect(err4).NotTo(HaveOccurred())
	})
})