mirror of https://github.com/cloudreve/Cloudreve
250 lines
7.5 KiB
Go
250 lines
7.5 KiB
Go
package routers
|
|
|
|
import (
|
|
"github.com/HFO4/cloudreve/models"
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/stretchr/testify/assert"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
func TestPing(t *testing.T) {
|
|
asserts := assert.New(t)
|
|
router := InitRouter()
|
|
|
|
w := httptest.NewRecorder()
|
|
req, _ := http.NewRequest("GET", "/api/v3/site/ping", nil)
|
|
router.ServeHTTP(w, req)
|
|
|
|
assert.Equal(t, 200, w.Code)
|
|
asserts.Contains(w.Body.String(), "Pong")
|
|
}
|
|
|
|
func TestCaptcha(t *testing.T) {
|
|
asserts := assert.New(t)
|
|
router := InitRouter()
|
|
w := httptest.NewRecorder()
|
|
|
|
req, _ := http.NewRequest(
|
|
"GET",
|
|
"/api/v3/captcha",
|
|
nil,
|
|
)
|
|
|
|
router.ServeHTTP(w, req)
|
|
|
|
asserts.Equal(200, w.Code)
|
|
asserts.Contains(w.Body.String(), "base64")
|
|
}
|
|
|
|
//func TestUserSession(t *testing.T) {
|
|
// mutex.Lock()
|
|
// defer mutex.Unlock()
|
|
// switchToMockDB()
|
|
// asserts := assert.New(t)
|
|
// router := InitRouter()
|
|
// w := httptest.NewRecorder()
|
|
//
|
|
// // 创建测试用验证码
|
|
// var configD = base64Captcha.ConfigDigit{
|
|
// Height: 80,
|
|
// Width: 240,
|
|
// MaxSkew: 0.7,
|
|
// DotCount: 80,
|
|
// CaptchaLen: 1,
|
|
// }
|
|
// idKeyD, _ := base64Captcha.GenerateCaptcha("", configD)
|
|
// middleware.ContextMock = map[string]interface{}{
|
|
// "captchaID": idKeyD,
|
|
// }
|
|
//
|
|
// testCases := []struct {
|
|
// settingRows *sqlmock.Rows
|
|
// userRows *sqlmock.Rows
|
|
// policyRows *sqlmock.Rows
|
|
// reqBody string
|
|
// expected interface{}
|
|
// }{
|
|
// // 登录信息正确，不需要验证码
|
|
// {
|
|
// settingRows: sqlmock.NewRows([]string{"name", "value", "type"}).
|
|
// AddRow("login_captcha", "0", "login"),
|
|
// userRows: sqlmock.NewRows([]string{"email", "nick", "password", "options"}).
|
|
// AddRow("admin@cloudreve.org", "admin", "CKLmDKa1C9SD64vU:76adadd4fd4bad86959155f6f7bc8993c94e7adf", "{}"),
|
|
// expected: serializer.BuildUserResponse(model.User{
|
|
// Email: "admin@cloudreve.org",
|
|
// Nick: "admin",
|
|
// Policy: model.Policy{
|
|
// Type: "local",
|
|
// OptionsSerialized: model.PolicyOption{FileType: []string{}},
|
|
// },
|
|
// }),
|
|
// },
|
|
// // 登录信息正确，需要验证码,验证码错误
|
|
// {
|
|
// settingRows: sqlmock.NewRows([]string{"name", "value", "type"}).
|
|
// AddRow("login_captcha", "1", "login"),
|
|
// userRows: sqlmock.NewRows([]string{"email", "nick", "password", "options"}).
|
|
// AddRow("admin@cloudreve.org", "admin", "CKLmDKa1C9SD64vU:76adadd4fd4bad86959155f6f7bc8993c94e7adf", "{}"),
|
|
// expected: serializer.ParamErr("验证码错误", nil),
|
|
// },
|
|
// // 邮箱正确密码错误
|
|
// {
|
|
// settingRows: sqlmock.NewRows([]string{"name", "value", "type"}).
|
|
// AddRow("login_captcha", "0", "login"),
|
|
// userRows: sqlmock.NewRows([]string{"email", "nick", "password", "options"}).
|
|
// AddRow("admin@cloudreve.org", "admin", "CKLmDKa1C9SD64vU:76adadd4fd4bad86959155f6f7bc8993c94e7adf", "{}"),
|
|
// expected: serializer.Err(401, "用户邮箱或密码错误", nil),
|
|
// },
|
|
// //邮箱格式不正确
|
|
// {
|
|
// reqBody: `{"userName":"admin@cloudreve","captchaCode":"captchaCode","Password":"admin123"}`,
|
|
// expected: serializer.Err(40001, "邮箱格式不正确", errors.New("Key: 'UserLoginService.UserName' Error:Field validation for 'UserName' failed on the 'email' tag")),
|
|
// },
|
|
// // 用户被Ban
|
|
// {
|
|
// settingRows: sqlmock.NewRows([]string{"name", "value", "type"}).
|
|
// AddRow("login_captcha", "0", "login"),
|
|
// userRows: sqlmock.NewRows([]string{"email", "nick", "password", "options", "status"}).
|
|
// AddRow("admin@cloudreve.org", "admin", "CKLmDKa1C9SD64vU:76adadd4fd4bad86959155f6f7bc8993c94e7adf", "{}", model.Baned),
|
|
// expected: serializer.Err(403, "该账号已被封禁", nil),
|
|
// },
|
|
// // 用户未激活
|
|
// {
|
|
// settingRows: sqlmock.NewRows([]string{"name", "value", "type"}).
|
|
// AddRow("login_captcha", "0", "login"),
|
|
// userRows: sqlmock.NewRows([]string{"email", "nick", "password", "options", "status"}).
|
|
// AddRow("admin@cloudreve.org", "admin", "CKLmDKa1C9SD64vU:76adadd4fd4bad86959155f6f7bc8993c94e7adf", "{}", model.NotActivicated),
|
|
// expected: serializer.Err(403, "该账号未激活", nil),
|
|
// },
|
|
// }
|
|
//
|
|
// for k, testCase := range testCases {
|
|
// if testCase.settingRows != nil {
|
|
// mock.ExpectQuery("^SELECT (.+)").WillReturnRows(testCase.settingRows)
|
|
// }
|
|
// if testCase.userRows != nil {
|
|
// mock.ExpectQuery("^SELECT (.+)").WillReturnRows(testCase.userRows)
|
|
// }
|
|
// if testCase.policyRows != nil {
|
|
// mock.ExpectQuery("^SELECT \\* FROM `(.+)` WHERE `(.+)`\\.`deleted_at` IS NULL AND \\(\\(`policies`.`id` = 1\\)\\)(.+)$").WillReturnRows(testCase.policyRows)
|
|
// }
|
|
// req, _ := http.NewRequest(
|
|
// "POST",
|
|
// "/api/v3/user/session",
|
|
// bytes.NewReader([]byte(testCase.reqBody)),
|
|
// )
|
|
// router.ServeHTTP(w, req)
|
|
//
|
|
// asserts.Equal(200, w.Code)
|
|
// expectedJSON, _ := json.Marshal(testCase.expected)
|
|
// asserts.JSONEq(string(expectedJSON), w.Body.String(), "测试用例：%d", k)
|
|
//
|
|
// w.Body.Reset()
|
|
// asserts.NoError(mock.ExpectationsWereMet())
|
|
// model.ClearCache()
|
|
// }
|
|
//
|
|
//}
|
|
//
|
|
//func TestSessionAuthCheck(t *testing.T) {
|
|
// mutex.Lock()
|
|
// defer mutex.Unlock()
|
|
// switchToMockDB()
|
|
// asserts := assert.New(t)
|
|
// router := InitRouter()
|
|
// w := httptest.NewRecorder()
|
|
//
|
|
// mock.ExpectQuery("^SELECT (.+)").WillReturnRows(sqlmock.NewRows([]string{"email", "nick", "password", "options"}).
|
|
// AddRow("admin@cloudreve.org", "admin", "CKLmDKa1C9SD64vU:76adadd4fd4bad86959155f6f7bc8993c94e7adf", "{}"))
|
|
// expectedUser, _ := model.GetUserByID(1)
|
|
//
|
|
// testCases := []struct {
|
|
// userRows *sqlmock.Rows
|
|
// sessionMock map[string]interface{}
|
|
// contextMock map[string]interface{}
|
|
// expected interface{}
|
|
// }{
|
|
// // 未登录
|
|
// {
|
|
// expected: serializer.CheckLogin(),
|
|
// },
|
|
// // 登录正常
|
|
// {
|
|
// userRows: sqlmock.NewRows([]string{"email", "nick", "password", "options"}).
|
|
// AddRow("admin@cloudreve.org", "admin", "CKLmDKa1C9SD64vU:76adadd4fd4bad86959155f6f7bc8993c94e7adf", "{}"),
|
|
// sessionMock: map[string]interface{}{"user_id": 1},
|
|
// expected: serializer.BuildUserResponse(expectedUser),
|
|
// },
|
|
// // UID不存在
|
|
// {
|
|
// userRows: sqlmock.NewRows([]string{"email", "nick", "password", "options"}),
|
|
// sessionMock: map[string]interface{}{"user_id": -1},
|
|
// expected: serializer.CheckLogin(),
|
|
// },
|
|
// }
|
|
//
|
|
// for _, testCase := range testCases {
|
|
// req, _ := http.NewRequest(
|
|
// "GET",
|
|
// "/api/v3/user/me",
|
|
// nil,
|
|
// )
|
|
// if testCase.userRows != nil {
|
|
// mock.ExpectQuery("^SELECT (.+)").WillReturnRows(testCase.userRows)
|
|
// }
|
|
// middleware.ContextMock = testCase.contextMock
|
|
// middleware.SessionMock = testCase.sessionMock
|
|
// router.ServeHTTP(w, req)
|
|
// expectedJSON, _ := json.Marshal(testCase.expected)
|
|
//
|
|
// asserts.Equal(200, w.Code)
|
|
// asserts.JSONEq(string(expectedJSON), w.Body.String())
|
|
// asserts.NoError(mock.ExpectationsWereMet())
|
|
//
|
|
// w.Body.Reset()
|
|
// }
|
|
//
|
|
//}
|
|
|
|
func TestSiteConfigRoute(t *testing.T) {
|
|
switchToMemDB()
|
|
asserts := assert.New(t)
|
|
router := InitRouter()
|
|
w := httptest.NewRecorder()
|
|
|
|
req, _ := http.NewRequest(
|
|
"GET",
|
|
"/api/v3/site/config",
|
|
nil,
|
|
)
|
|
router.ServeHTTP(w, req)
|
|
asserts.Equal(200, w.Code)
|
|
asserts.Contains(w.Body.String(), "Cloudreve")
|
|
|
|
w.Body.Reset()
|
|
|
|
// 消除无效值
|
|
model.DB.Model(&model.Setting{
|
|
Model: gorm.Model{
|
|
ID: 2,
|
|
},
|
|
}).UpdateColumn("name", "siteName_b")
|
|
|
|
req, _ = http.NewRequest(
|
|
"GET",
|
|
"/api/v3/site/config",
|
|
nil,
|
|
)
|
|
router.ServeHTTP(w, req)
|
|
asserts.Equal(200, w.Code)
|
|
asserts.Contains(w.Body.String(), "\"title\":\"\"")
|
|
|
|
model.DB.Model(&model.Setting{
|
|
Model: gorm.Model{
|
|
ID: 2,
|
|
},
|
|
}).UpdateColumn("name", "siteName")
|
|
}
|