Test: User login router

models/setting.go

@@ -15,6 +15,11 @@ type Setting struct {
 // settingCache 设置项缓存
 var settingCache = make(map[string]string)
 
+// ClearCache 清空设置缓存
+func ClearCache() {
+	settingCache = make(map[string]string)
+}
+
 // IsTrueVal 返回设置的值是否为真
 func IsTrueVal(val string) bool {
 	return val == "1" || val == "true"

routers/router_test.go

@@ -1,12 +1,32 @@
 package routers
 
 import (
+	"bytes"
+	"cloudreve/models"
+	"database/sql"
+	"github.com/DATA-DOG/go-sqlmock"
+	"github.com/jinzhu/gorm"
 	"github.com/stretchr/testify/assert"
 	"net/http"
 	"net/http/httptest"
 	"testing"
 )
 
+var mock sqlmock.Sqlmock
+
+// TestMain 初始化数据库Mock
+func TestMain(m *testing.M) {
+	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.DB, _ = gorm.Open("mysql", db)
+	defer db.Close()
+	m.Run()
+}
+
 func TestPing(t *testing.T) {
 	asserts := assert.New(t)
 	router := InitRouter()
@@ -18,3 +38,80 @@ func TestPing(t *testing.T) {
 	assert.Equal(t, 200, w.Code)
 	asserts.Contains(w.Body.String(), "Pong")
 }
+
+func TestUserSession(t *testing.T) {
+	asserts := assert.New(t)
+	router := InitRouter()
+	w := httptest.NewRecorder()
+
+	testCases := []struct {
+		settingRows *sqlmock.Rows
+		userRows    *sqlmock.Rows
+		reqBody     string
+		expected    string
+	}{
+		// 登录信息正确，不需要验证码
+		{
+			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", "{}"),
+			reqBody:  `{"userName":"admin@cloudreve.org","captchaCode":"captchaCode","Password":"admin"}`,
+			expected: `{"code":0,"data":{"ID":0,"CreatedAt":"0001-01-01T00:00:00Z","UpdatedAt":"0001-01-01T00:00:00Z","DeletedAt":null,"Email":"admin@cloudreve.org","Nick":"admin","Status":0,"Group":0,"PrimaryGroup":0,"Storage":0,"LastNotify":null,"Delay":0,"Avatar":"","OptionsSerialized":{"profile_on":0,"webdav_key":""}},"msg":""}`,
+		},
+		// 邮箱正确密码错误
+		{
+			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", "{}"),
+			reqBody:  `{"userName":"admin@cloudreve.org","captchaCode":"captchaCode","Password":"admin123"}`,
+			expected: `{"code":401, "msg":"用户邮箱或密码错误"}`,
+		},
+		//邮箱格式不正确
+		{
+			reqBody:  `{"userName":"admin@cloudreve","captchaCode":"captchaCode","Password":"admin123"}`,
+			expected: `{"code":40001, "error":"Key: 'UserLoginService.UserName' Error:Field validation for 'UserName' failed on the 'email' tag", "msg":"邮箱格式不正确"}`,
+		},
+		// 用户被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),
+			reqBody:  `{"userName":"admin@cloudreve.org","captchaCode":"captchaCode","Password":"admin"}`,
+			expected: `{"code":403, "msg":"该账号已被封禁"}`,
+		},
+		// 用户未激活
+		{
+			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),
+			reqBody:  `{"userName":"admin@cloudreve.org","captchaCode":"captchaCode","Password":"admin"}`,
+			expected: `{"code":403, "msg":"该账号未激活"}`,
+		},
+	}
+
+	for _, testCase := range testCases {
+		if testCase.settingRows != nil {
+			mock.ExpectQuery("^SELECT (.+)").WillReturnRows(testCase.settingRows)
+		}
+		if testCase.userRows != nil {
+			mock.ExpectQuery("^SELECT (.+)").WillReturnRows(testCase.userRows)
+		}
+		req, _ := http.NewRequest(
+			"POST",
+			"/Api/V3/User/Session",
+			bytes.NewReader([]byte(testCase.reqBody)),
+		)
+		router.ServeHTTP(w, req)
+
+		assert.Equal(t, 200, w.Code)
+		asserts.JSONEq(testCase.expected, w.Body.String())
+		w.Body.Reset()
+		asserts.NoError(mock.ExpectationsWereMet())
+		model.ClearCache()
+	}
+
+}