Test: captcha

routers/router_test.go

@@ -11,6 +11,7 @@ import (
 	"github.com/DATA-DOG/go-sqlmock"
 	"github.com/gin-gonic/gin"
 	"github.com/jinzhu/gorm"
+	"github.com/mojocn/base64Captcha"
 	"github.com/stretchr/testify/assert"
 	"net/http"
 	"net/http/httptest"
@@ -47,11 +48,41 @@ func TestPing(t *testing.T) {
 	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) {
 	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
@@ -70,6 +101,15 @@ func TestUserSession(t *testing.T) {
 				Nick:  "admin",
 			}),
 		},
+		// 登录信息正确，需要验证码,验证码错误
+		{
+			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", "{}"),
+			reqBody:  `{"userName":"admin@cloudreve.org","captchaCode":"captchaCode","Password":"admin"}`,
+			expected: serializer.ParamErr("验证码错误", nil),
+		},
 		// 邮箱正确密码错误
 		{
 			settingRows: sqlmock.NewRows([]string{"name", "value", "type"}).
@@ -104,7 +144,7 @@ func TestUserSession(t *testing.T) {
 		},
 	}
 
-	for _, testCase := range testCases {
+	for k, testCase := range testCases {
 		if testCase.settingRows != nil {
 			mock.ExpectQuery("^SELECT (.+)").WillReturnRows(testCase.settingRows)
 		}
@@ -120,7 +160,7 @@ func TestUserSession(t *testing.T) {
 
 		asserts.Equal(200, w.Code)
 		expectedJSON, _ := json.Marshal(testCase.expected)
-		asserts.JSONEq(string(expectedJSON), w.Body.String())
+		asserts.JSONEq(string(expectedJSON), w.Body.String(), "测试用例：%d", k)
 
 		w.Body.Reset()
 		asserts.NoError(mock.ExpectationsWereMet())