Add unit test for routes/node

2026-01-22 17:31:03 +01:00 · 2019-08-19 18:06:49 +08:00
parent 3fa7b535f4
commit 22c599c9dd
1 changed files with 152 additions and 0 deletions
--- a/backend/routes/node_test.go
+++ b/backend/routes/node_test.go
@@ -0,0 +1,152 @@
+package routes
+
+import (
+	"crawlab/config"
+	"crawlab/database"
+	"encoding/json"
+	"github.com/apex/log"
+	"github.com/gin-gonic/gin"
+	. "github.com/smartystreets/goconvey/convey"
+	"github.com/spf13/viper"
+	"net/http"
+	"net/http/httptest"
+	"runtime/debug"
+	"testing"
+)
+
+var app *gin.Engine
+// 本测试依赖MongoDB的服务，所以在测试之前需要启动MongoDB及相关服务
+func init() {
+	app = gin.Default()
+
+	// 初始化配置
+	if err := config.InitConfig("../conf/config.yml"); err != nil {
+		panic(err)
+	}
+	log.Info("初始化配置成功")
+
+	// 初始化日志设置
+	logLevel := viper.GetString("log.level")
+	if logLevel != "" {
+		log.SetLevelFromString(logLevel)
+	}
+	log.Info("初始化日志设置成功")
+
+	// 初始化Mongodb数据库
+	if err := database.InitMongo(); err != nil {
+		debug.PrintStack()
+		panic(err)
+	}
+	log.Info("初始化Mongodb数据库成功")
+
+	// 初始化Redis数据库
+	if err := database.InitRedis(); err != nil {
+		debug.PrintStack()
+		panic(err)
+	}
+	log.Info("初始化Redis数据库成功")
+
+	// 路由
+	// 节点相关的API
+	app.GET("/ping", Ping)
+	app.GET("/nodes", GetNodeList)               // 节点列表
+	app.GET("/nodes/:id", GetNode)               // 节点详情
+	app.POST("/nodes/:id", PostNode)             // 修改节点
+	app.GET("/nodes/:id/tasks", GetNodeTaskList) // 节点任务列表
+	app.GET("/nodes/:id/system", GetSystemInfo)  // 节点任务列表
+	app.DELETE("/nodes/:id", DeleteNode)         // 删除节点
+	// 爬虫
+}
+
+//先测试GetNodeList得到一个节点的ID，再继续测试下面的API
+func TestGetNodeList(t *testing.T) {
+	var resp Response
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("GET", "/nodes", nil)
+	app.ServeHTTP(w, req)
+	err := json.Unmarshal([]byte(w.Body.String()), &resp)
+	if err != nil {
+		t.Fatal("Unmarshal resp failed")
+	}
+	t.Log(resp.Data)
+	Convey("Test API GetNodeList", t, func() {
+		Convey("Test response status", func() {
+			So(resp.Status, ShouldEqual, "ok")
+			So(resp.Message, ShouldEqual, "success")
+		})
+	})
+}
+
+//依赖MongoDB中的数据,_id=5d429e6c19f7abede924fee2,实际测试时需替换
+func TestGetNode(t *testing.T) {
+	var resp Response
+	var mongoId = "5d5a658319f7ab423585b0b0"
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("GET", "/nodes/"+mongoId, nil)
+	app.ServeHTTP(w, req)
+	err := json.Unmarshal([]byte(w.Body.String()), &resp)
+	if err != nil {
+		t.Fatal("Unmarshal resp failed")
+	}
+	t.Log(resp.Data)
+	Convey("Test API GetNode", t, func() {
+		Convey("Test response status", func() {
+			So(resp.Status, ShouldEqual, "ok")
+			So(resp.Message, ShouldEqual, "success")
+			So(resp.Data.(map[string]interface{})["_id"], ShouldEqual, mongoId)
+		})
+	})
+}
+
+func TestPing(t *testing.T) {
+	var resp Response
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("GET", "/ping", nil)
+	app.ServeHTTP(w, req)
+	err := json.Unmarshal([]byte(w.Body.String()), &resp)
+	if err != nil {
+		t.Fatal("Unmarshal resp failed")
+	}
+	Convey("Test API ping", t, func() {
+		Convey("Test response status", func() {
+			So(resp.Status, ShouldEqual, "ok")
+			So(resp.Message, ShouldEqual, "success")
+		})
+	})
+}
+
+func TestGetNodeTaskList(t *testing.T) {
+	var resp Response
+	var mongoId = "5d5a658319f7ab423585b0b0"
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("GET", "nodes/"+mongoId+"/tasks", nil)
+	app.ServeHTTP(w, req)
+	err := json.Unmarshal([]byte(w.Body.String()), &resp)
+	if err != nil {
+		t.Fatal("Unmarshal resp failed")
+	}
+	Convey("Test API GetNodeTaskList", t, func() {
+		Convey("Test response status", func() {
+			So(resp.Status, ShouldEqual, "ok")
+			So(resp.Message, ShouldEqual, "success")
+		})
+	})
+}
+
+func TestDeleteNode(t *testing.T) {
+	var resp Response
+	var mongoId = "5d5a658319f7ab423585b0b0"
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("DELETE", "nodes/"+mongoId, nil)
+	app.ServeHTTP(w, req)
+	err := json.Unmarshal([]byte(w.Body.String()), &resp)
+	if err != nil {
+		t.Fatal("Unmarshal resp failed")
+	}
+	Convey("Test API DeleteNode", t, func() {
+		Convey("Test response status", func() {
+			So(resp.Status, ShouldEqual, "ok")
+			So(resp.Message, ShouldEqual, "success")
+		})
+	})
+}