You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
122 lines
2.4 KiB
122 lines
2.4 KiB
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)
|
|
|
|
type Test struct {
|
|
UserName string `validate:"required,min=10,max=20"`
|
|
}
|
|
|
|
testCase1 := Test{}
|
|
validate := validator.New(&validator.Config{TagName: "validate"})
|
|
errs := validate.Struct(testCase1)
|
|
res := ErrorResponse(errs)
|
|
asserts.Equal(serializer.ParamErr(ParamErrorMsg("UserName", "required"), errs), res)
|
|
|
|
testCase2 := Test{
|
|
UserName: "a",
|
|
}
|
|
validate2 := validator.New(&validator.Config{TagName: "validate"})
|
|
errs = validate2.Struct(testCase2)
|
|
res = ErrorResponse(errs)
|
|
asserts.Equal(serializer.ParamErr(ParamErrorMsg("UserName", "min"), errs), res)
|
|
|
|
type JsonTest struct {
|
|
UserName string `json:"UserName"`
|
|
}
|
|
testCase3 := JsonTest{}
|
|
errs = json.Unmarshal([]byte("{\"UserName\":1}"), &testCase3)
|
|
res = ErrorResponse(errs)
|
|
asserts.Equal(serializer.ParamErr("JSON类型不匹配", errs), res)
|
|
|
|
errs = errors.New("Unknown error")
|
|
res = ErrorResponse(errs)
|
|
asserts.Equal(serializer.ParamErr("参数错误", errs), res)
|
|
}
|
|
|
|
// 测试 ParamErrorMs
|
|
func TestParamErrorMsg(t *testing.T) {
|
|
asserts := assert.New(t)
|
|
testCase := []struct {
|
|
field string
|
|
tag string
|
|
expect string
|
|
}{
|
|
{
|
|
"UserName",
|
|
"required",
|
|
"邮箱不能为空",
|
|
},
|
|
{
|
|
"Password",
|
|
"min",
|
|
"密码太短",
|
|
},
|
|
{
|
|
"Password",
|
|
"max",
|
|
"密码太长",
|
|
},
|
|
{
|
|
"something unexpected",
|
|
"max",
|
|
"",
|
|
},
|
|
{
|
|
"",
|
|
"",
|
|
"",
|
|
},
|
|
}
|
|
for _, value := range testCase {
|
|
asserts.Equal(value.expect, ParamErrorMsg(value.field, value.tag), "case %v", value)
|
|
}
|
|
}
|