Test: options / WebDAV middleware

middleware/auth_test.go

@@ -148,6 +148,8 @@ func TestWebDAVAuth(t *testing.T) {
 			WillReturnRows(
 				sqlmock.NewRows([]string{"id", "password", "email", "options"}).AddRow(1, "123", "who@cloudreve.org", "{}"),
 			)
+		// 查找密码
+		mock.ExpectQuery("SELECT(.+)webdav(.+)").WillReturnRows(sqlmock.NewRows([]string{"id"}))
 		AuthFunc(c)
 		asserts.NoError(mock.ExpectationsWereMet())
 		asserts.Equal(c.Writer.Status(), http.StatusUnauthorized)
@@ -172,6 +174,8 @@ func TestWebDAVAuth(t *testing.T) {
 					),
 			)
 		mock.ExpectQuery("SELECT(.+)groups(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "web_dav_enabled"}).AddRow(1, false))
+		// 查找密码
+		mock.ExpectQuery("SELECT(.+)webdav(.+)").WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(1))
 		AuthFunc(c)
 		asserts.NoError(mock.ExpectationsWereMet())
 		asserts.Equal(c.Writer.Status(), http.StatusForbidden)
@@ -196,6 +200,8 @@ func TestWebDAVAuth(t *testing.T) {
 					),
 			)
 		mock.ExpectQuery("SELECT(.+)groups(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "web_dav_enabled"}).AddRow(1, true))
+		// 查找密码
+		mock.ExpectQuery("SELECT(.+)webdav(.+)").WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(1))
 		AuthFunc(c)
 		asserts.NoError(mock.ExpectationsWereMet())
 		asserts.Equal(c.Writer.Status(), 200)

middleware/option_test.go

@@ -1,6 +1,7 @@
 package middleware
 
 import (
+	"github.com/HFO4/cloudreve/pkg/cache"
 	"github.com/HFO4/cloudreve/pkg/hashid"
 	"github.com/gin-gonic/gin"
 	"github.com/stretchr/testify/assert"
@@ -48,3 +49,31 @@ func TestHashID(t *testing.T) {
 		asserts.False(c.IsAborted())
 	}
 }
+
+func TestIsFunctionEnabled(t *testing.T) {
+	asserts := assert.New(t)
+	rec := httptest.NewRecorder()
+	TestFunc := IsFunctionEnabled("TestIsFunctionEnabled")
+
+	// 未开启
+	{
+		cache.Set("setting_TestIsFunctionEnabled", "0", 0)
+		c, _ := gin.CreateTestContext(rec)
+		c.Params = []gin.Param{}
+		c.Request, _ = http.NewRequest("POST", "/api/v3/file/dellete/1", nil)
+		TestFunc(c)
+		asserts.NoError(mock.ExpectationsWereMet())
+		asserts.True(c.IsAborted())
+	}
+	// 开启
+	{
+		cache.Set("setting_TestIsFunctionEnabled", "1", 0)
+		c, _ := gin.CreateTestContext(rec)
+		c.Params = []gin.Param{}
+		c.Request, _ = http.NewRequest("POST", "/api/v3/file/dellete/1", nil)
+		TestFunc(c)
+		asserts.NoError(mock.ExpectationsWereMet())
+		asserts.False(c.IsAborted())
+	}
+
+}