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.
cloudreve/routers/router_test.go

250 lines
7.5 KiB

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/site/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")
}