create_task_test.go 7.1 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() {
	BeforeEach(func() {
		_, err := pG.DB.QueryOne(
			pg.Scan(),
			"INSERT INTO employees (id, uid, employee_name, employee_account, su_money) VALUES (?, ?, ?, ?, ?)",
			1, 2499036607974745088, "testEmployeeName", "testEmployeeAccount", 0)
		Expect(err).NotTo(HaveOccurred())
		_, err1 := 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, create_time, release_time) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
			2, 101, "抢单任务2", 1, &domain.EmployeeInfo{
				Uid: 2499036607974745088,
			}, 2, &domain.ReferenceResource{
				ReferenceResourceType: 1,
				ReferenceResourceItems: []*domain.ReferenceResourceItem{
					{
						ReferenceResourceId: 4,
					},
				},
			}, 1, pg.Array([]int{1, 2}), 1, 1000.00, "验收标准1", "任务描述1", pg.Array([]string{}), false, "null", "null", "", pg.Array([]string{}), time.Now(), time.Now())
		Expect(err1).NotTo(HaveOccurred())
	})
	Describe("提交数据创建新任务", func() {
		Context("提交正确的抢单类型任务数据", func() {
			It("返回抢单类型任务数据", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				body := map[string]interface{}{
					"companyId":             101,
					"taskName":              "抢单任务",
					"taskType":              1,
					"sponsor":               2499036607974745088,
					"referenceResourceType": 1,
					"referenceResourceItems": []map[string]interface{}{
						{"referenceResourceId": 1, "serialNumber": 1, "title": "问题标题1"},
						{"referenceResourceId": 2, "serialNumber": 2, "title": "问题标题2"},
						{"referenceResourceId": 3, "serialNumber": 3, "title": "问题标题3"},
					},
					"projectBelong": 1,
					"customerValues": []int{
						1,
						2,
						3,
					},
					"taskNature":         1,
					"suMoney":            1000.00,
					"acceptanceStandard": "验收标准",
					"taskDescription":    "任务描述",
					"taskPictureUrls": []string{
						"url-1",
						"url-2",
						"url-3",
					},
					"isRewardTake": false,
				}
				httpExpect.POST("/tasks/").
					WithJSON(body).
					Expect().
					Status(http.StatusOK).
					JSON().
					Object().
					ContainsKey("code").ValueEqual("code", 0).
					ContainsKey("msg").ValueEqual("msg", "ok").
					ContainsKey("data").Value("data").Object().
					ContainsKey("taskId").ValueNotEqual("taskId", BeZero())
			})
		})
		Context("提交正确的竞标类型任务数据", func() {
			It("返回竞标类型任务数据", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				dayAfter, _ := time.ParseDuration("72h")
				body := map[string]interface{}{
					"companyId":             101,
					"taskName":              "竞标任务",
					"taskType":              2,
					"sponsor":               2499036607974745088,
					"referenceResourceType": 2,
					"referenceResourceItems": []map[string]interface{}{
						{"serialNumber": 1, "title": "机会标题1"},
						{"serialNumber": 2, "title": "机会标题2"},
						{"serialNumber": 3, "title": "机会标题3"},
					},
					"projectBelong": 1,
					"customerValues": []int{
						1,
						2,
						3,
					},
					"taskNature":         1,
					"suMoney":            1000.00,
					"acceptanceStandard": "验收标准",
					"taskDescription":    "任务描述",
					"taskPictureUrls": []string{
						"url-1",
						"url-2",
						"url-3",
					},
					"isRewardTake": true,
					"bidStartTime": time.Now(),
					"bidEndTime":   time.Now().Add(dayAfter),
				}
				httpExpect.POST("/tasks/").
					WithJSON(body).
					Expect().
					Status(http.StatusOK).
					JSON().
					Object().
					ContainsKey("code").ValueEqual("code", 0).
					ContainsKey("msg").ValueEqual("msg", "ok").
					ContainsKey("data").Value("data").Object().
					ContainsKey("taskId").ValueNotEqual("taskId", BeZero())
			})
		})
		Context("提交正确的指派类型任务数据", func() {
			It("返回指派类型任务数据", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				body := map[string]interface{}{
					"companyId":             101,
					"taskName":              "竞标任务",
					"taskType":              2,
					"sponsor":               2499036607974745088,
					"referenceResourceType": 2,
					"referenceResourceItems": []map[string]interface{}{
						{"serialNumber": 1, "title": "机会标题1"},
						{"serialNumber": 2, "title": "机会标题2"},
						{"serialNumber": 3, "title": "机会标题3"},
					},
					"projectBelong": 1,
					"customerValues": []int{
						1,
						2,
						3,
					},
					"taskNature":         1,
					"suMoney":            1000.00,
					"acceptanceStandard": "验收标准",
					"taskDescription":    "任务描述",
					"taskPictureUrls": []string{
						"url-1",
						"url-2",
						"url-3",
					},
					"assignedPerson":        2499036607974745088,
					"plannedCompletionTime": time.Now(),
				}
				httpExpect.POST("/tasks/").
					WithJSON(body).
					Expect().
					Status(http.StatusOK).
					JSON().
					Object().
					ContainsKey("code").ValueEqual("code", 0).
					ContainsKey("msg").ValueEqual("msg", "ok").
					ContainsKey("data").Value("data").Object().
					ContainsKey("taskId").ValueNotEqual("taskId", BeZero())
			})
		})
		Context("提交已经被引用的问题ID任务数据", func() {
			It("创建失败", func() {
				httpExpect := httpexpect.New(GinkgoT(), server.URL)
				body := map[string]interface{}{
					"companyId":             101,
					"taskName":              "抢单任务",
					"taskType":              1,
					"sponsor":               2499036607974745088,
					"referenceResourceType": 1,
					"referenceResourceItems": []map[string]interface{}{
						{"referenceResourceId": 4, "serialNumber": 4, "title": "问题标题4"},
					},
					"projectBelong": 1,
					"customerValues": []int{
						1,
						2,
						3,
					},
					"taskNature":         1,
					"suMoney":            1000.00,
					"acceptanceStandard": "验收标准",
					"taskDescription":    "任务描述",
					"taskPictureUrls": []string{
						"url-1",
						"url-2",
						"url-3",
					},
					"isRewardTake": false,
				}
				httpExpect.POST("/tasks/").
					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 employees WHERE true")
		Expect(err1).NotTo(HaveOccurred())
	})
})