Update test cases

zzy404 · Dec 15, 2019 · 0387bbd · 0387bbd
1 parent ce7d523
commit 0387bbd
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/backend/mock/schedule.go b/backend/mock/schedule.go
@@ -10,12 +10,15 @@ import (
 	"time"
 )
 
+var NodeIdss = []bson.ObjectId{bson.ObjectIdHex("5d429e6c19f7abede924fee2"),
+	bson.ObjectIdHex("5d429e6c19f7abede924fee1")}
+
 var scheduleList = []model.Schedule{
 	{
 		Id:       bson.ObjectId("5d429e6c19f7abede924fee2"),
 		Name:     "test schedule",
 		SpiderId: "123",
-		NodeId:   bson.ObjectId("5d429e6c19f7abede924fee2"),
+		NodeIds:  NodeIdss,
 		Cron:     "***1*",
 		EntryId:  10,
 		// 前端展示
@@ -29,7 +32,7 @@ var scheduleList = []model.Schedule{
 		Id:       bson.ObjectId("xx429e6c19f7abede924fee2"),
 		Name:     "test schedule2",
 		SpiderId: "234",
-		NodeId:   bson.ObjectId("5d429e6c19f7abede924fee2"),
+		NodeIds:  NodeIdss,
 		Cron:     "***1*",
 		EntryId:  10,
 		// 前端展示
@@ -100,8 +103,10 @@ func PutSchedule(c *gin.Context) {
 	}
 
 	// 如果node_id为空，则置为空ObjectId
-	if item.NodeId == "" {
-		item.NodeId = bson.ObjectIdHex(constants.ObjectIdNull)
+	for _, NodeId := range item.NodeIds {
+		if NodeId == "" {
+			NodeId = bson.ObjectIdHex(constants.ObjectIdNull)
+		}
 	}
 
 	c.JSON(http.StatusOK, Response{

diff --git a/backend/mock/schedule_test.go b/backend/mock/schedule_test.go
@@ -75,7 +75,7 @@ func TestPostSchedule(t *testing.T) {
 		Id:       bson.ObjectIdHex("5d429e6c19f7abede924fee2"),
 		Name:     "test schedule",
 		SpiderId: bson.ObjectIdHex("5d429e6c19f7abede924fee2"),
-		NodeId:   bson.ObjectIdHex("5d429e6c19f7abede924fee2"),
+		NodeIds:  NodeIdss,
 		Cron:     "***1*",
 		EntryId:  10,
 		// 前端展示
@@ -112,7 +112,7 @@ func TestPutSchedule(t *testing.T) {
 		Id:       bson.ObjectIdHex("5d429e6c19f7abede924fee2"),
 		Name:     "test schedule",
 		SpiderId: bson.ObjectIdHex("5d429e6c19f7abede924fee2"),
-		NodeId:   bson.ObjectIdHex("5d429e6c19f7abede924fee2"),
+		NodeIds:  NodeIdss,
 		Cron:     "***1*",
 		EntryId:  10,
 		// 前端展示