Test: local policy file uploading

routers/controllers/main_test.go

@@ -1,14 +1,51 @@
 package controllers
 
 import (
+	"database/sql"
 	"encoding/json"
 	"errors"
+	"github.com/DATA-DOG/go-sqlmock"
+	model "github.com/HFO4/cloudreve/models"
 	"github.com/HFO4/cloudreve/pkg/serializer"
+	"github.com/gin-gonic/gin"
+	"github.com/jinzhu/gorm"
 	"github.com/stretchr/testify/assert"
 	"gopkg.in/go-playground/validator.v8"
 	"testing"
 )
 
+var mock sqlmock.Sqlmock
+var memDB *gorm.DB
+
+// TestMain 初始化数据库Mock
+func TestMain(m *testing.M) {
+	// 设置gin为测试模式
+	gin.SetMode(gin.TestMode)
+
+	// 初始化sqlmock
+	var db *sql.DB
+	var err error
+	db, mock, err = sqlmock.New()
+	if err != nil {
+		panic("An error was not expected when opening a stub database connection")
+	}
+
+	// 初始话内存数据库
+	model.Init()
+	memDB = model.DB
+
+	model.DB, _ = gorm.Open("mysql", db)
+	defer db.Close()
+
+	switchToMemDB()
+
+	m.Run()
+}
+
+func switchToMemDB() {
+	model.DB = memDB
+}
+
 // 测试 ErrorResponse
 func TestErrorResponse(t *testing.T) {
 	asserts := assert.New(t)