Test: aria2 / filesystem.image

5 years ago · c74ed10dbd
parent 297fc8d5bb
commit c74ed10dbd
5 changed files with 238 additions and 3 deletions
--- a/middleware/auth_test.go
+++ b/middleware/auth_test.go
@ -622,3 +622,128 @@ func TestUpyunCallbackAuth(t *testing.T) {
 		asserts.False(c.IsAborted())
 	}
 }
+
+func TestOneDriveCallbackAuth(t *testing.T) {
+	asserts := assert.New(t)
+	rec := httptest.NewRecorder()
+	AuthFunc := OneDriveCallbackAuth()
+
+	// Callback Key 相关验证失败
+	{
+		c, _ := gin.CreateTestContext(rec)
+		c.Params = []gin.Param{
+			{"key", "testUpyunBackRemote"},
+		}
+		c.Request, _ = http.NewRequest("POST", "/api/v3/callback/upyun/testUpyunBackRemote", nil)
+		AuthFunc(c)
+		asserts.True(c.IsAborted())
+	}
+
+	// 成功
+	{
+		cache.Set(
+			"callback_testCallBackUpyun",
+			serializer.UploadSession{
+				UID:         1,
+				PolicyID:    512,
+				VirtualPath: "/",
+			},
+			0,
+		)
+		cache.Deletes([]string{"1"}, "policy_")
+		mock.ExpectQuery("SELECT(.+)users(.+)").
+			WillReturnRows(sqlmock.NewRows([]string{"id", "group_id"}).AddRow(1, 1))
+		mock.ExpectQuery("SELECT(.+)groups(.+)").
+			WillReturnRows(sqlmock.NewRows([]string{"id", "policies"}).AddRow(1, "[522]"))
+		mock.ExpectQuery("SELECT(.+)policies(.+)").
+			WillReturnRows(sqlmock.NewRows([]string{"id", "access_key", "secret_key"}).AddRow(2, "123", "123"))
+		c, _ := gin.CreateTestContext(rec)
+		c.Params = []gin.Param{
+			{"key", "testCallBackUpyun"},
+		}
+		c.Request, _ = http.NewRequest("POST", "/api/v3/callback/upyun/testCallBackUpyun", ioutil.NopCloser(strings.NewReader("1")))
+		AuthFunc(c)
+		asserts.NoError(mock.ExpectationsWereMet())
+		asserts.False(c.IsAborted())
+	}
+}
+
+func TestCOSCallbackAuth(t *testing.T) {
+	asserts := assert.New(t)
+	rec := httptest.NewRecorder()
+	AuthFunc := COSCallbackAuth()
+
+	// Callback Key 相关验证失败
+	{
+		c, _ := gin.CreateTestContext(rec)
+		c.Params = []gin.Param{
+			{"key", "testUpyunBackRemote"},
+		}
+		c.Request, _ = http.NewRequest("POST", "/api/v3/callback/upyun/testUpyunBackRemote", nil)
+		AuthFunc(c)
+		asserts.True(c.IsAborted())
+	}
+
+	// 成功
+	{
+		cache.Set(
+			"callback_testCallBackUpyun",
+			serializer.UploadSession{
+				UID:         1,
+				PolicyID:    512,
+				VirtualPath: "/",
+			},
+			0,
+		)
+		cache.Deletes([]string{"1"}, "policy_")
+		mock.ExpectQuery("SELECT(.+)users(.+)").
+			WillReturnRows(sqlmock.NewRows([]string{"id", "group_id"}).AddRow(1, 1))
+		mock.ExpectQuery("SELECT(.+)groups(.+)").
+			WillReturnRows(sqlmock.NewRows([]string{"id", "policies"}).AddRow(1, "[522]"))
+		mock.ExpectQuery("SELECT(.+)policies(.+)").
+			WillReturnRows(sqlmock.NewRows([]string{"id", "access_key", "secret_key"}).AddRow(2, "123", "123"))
+		c, _ := gin.CreateTestContext(rec)
+		c.Params = []gin.Param{
+			{"key", "testCallBackUpyun"},
+		}
+		c.Request, _ = http.NewRequest("POST", "/api/v3/callback/upyun/testCallBackUpyun", ioutil.NopCloser(strings.NewReader("1")))
+		AuthFunc(c)
+		asserts.NoError(mock.ExpectationsWereMet())
+		asserts.False(c.IsAborted())
+	}
+}
+
+func TestIsAdmin(t *testing.T) {
+	asserts := assert.New(t)
+	rec := httptest.NewRecorder()
+	testFunc := IsAdmin()
+
+	// 非管理员
+	{
+		c, _ := gin.CreateTestContext(rec)
+		c.Set("user", &model.User{})
+		testFunc(c)
+		asserts.True(c.IsAborted())
+	}
+
+	// 是管理员
+	{
+		c, _ := gin.CreateTestContext(rec)
+		user := &model.User{}
+		user.Group.ID = 1
+		c.Set("user", user)
+		testFunc(c)
+		asserts.False(c.IsAborted())
+	}
+
+	// 初始用户，非管理组
+	{
+		c, _ := gin.CreateTestContext(rec)
+		user := &model.User{}
+		user.Group.ID = 2
+		user.ID = 1
+		c.Set("user", user)
+		testFunc(c)
+		asserts.False(c.IsAborted())
+	}
+}
--- a/models/policy_test.go
+++ b/models/policy_test.go
@ -149,6 +149,14 @@ func TestPolicy_GenerateFileName(t *testing.T) {
 		testPolicy.Type = "upyun"
 		testPolicy.FileNameRule = "{uid}123{originname}"
 		asserts.Equal("1123{filename}{.suffix}", testPolicy.GenerateFileName(1, ""))
+
+		testPolicy.Type = "qiniu"
+		testPolicy.FileNameRule = "{uid}123{originname}"
+		asserts.Equal("1123$(fname)", testPolicy.GenerateFileName(1, ""))
+
+		testPolicy.Type = "local"
+		testPolicy.FileNameRule = "{uid}123{originname}"
+		asserts.Equal("1123", testPolicy.GenerateFileName(1, ""))
 	}

 }
@ -179,8 +187,20 @@ func TestPolicy_GetUploadURL(t *testing.T) {

 	// OSS
 	{
-		policy := Policy{Type: "oss", BaseURL: "base", Server: "http://127.0.0.1"}
-		asserts.Equal("base", policy.GetUploadURL())
+		policy := Policy{Type: "oss", BucketName: "base", Server: "127.0.0.1"}
+		asserts.Equal("https://base.127.0.0.1", policy.GetUploadURL())
+	}
+
+	// cos
+	{
+		policy := Policy{Type: "cos", BaseURL: "base", Server: "http://127.0.0.1"}
+		asserts.Equal("http://127.0.0.1", policy.GetUploadURL())
+	}
+
+	// upyun
+	{
+		policy := Policy{Type: "upyun", BucketName: "base", Server: "http://127.0.0.1"}
+		asserts.Equal("https://v0.api.upyun.com/base", policy.GetUploadURL())
 	}

 	// 未知
--- a/pkg/aria2/aria2_test.go
+++ b/pkg/aria2/aria2_test.go
@ -67,7 +67,7 @@ func TestInit(t *testing.T) {

 	// 连接失败
 	{
-		cache.Set("setting_aria2_options", "[]", 0)
+		cache.Set("setting_aria2_options", "{}", 0)
 		cache.Set("setting_aria2_rpcurl", "http://127.0.0.1:1234", 0)
 		cache.Set("setting_aria2_call_timeout", "1", 0)
 		cache.Set("setting_aria2_interval", "100", 0)
--- a/pkg/aria2/notification_test.go
+++ b/pkg/aria2/notification_test.go
@ -0,0 +1,51 @@
+package aria2
+
+import (
+	"github.com/HFO4/cloudreve/pkg/aria2/rpc"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestNotifier_Notify(t *testing.T) {
+	asserts := assert.New(t)
+	notifier2 := &Notifier{}
+	notifyChan := make(chan StatusEvent, 10)
+	notifier2.Subscribe(notifyChan, "1")
+
+	// 未订阅
+	{
+		notifier2.Notify([]rpc.Event{rpc.Event{Gid: ""}}, 1)
+		asserts.Len(notifyChan, 0)
+	}
+
+	// 订阅
+	{
+		notifier2.Notify([]rpc.Event{{Gid: "1"}}, 1)
+		asserts.Len(notifyChan, 1)
+		<-notifyChan
+
+		notifier2.OnBtDownloadComplete([]rpc.Event{{Gid: "1"}})
+		asserts.Len(notifyChan, 1)
+		<-notifyChan
+
+		notifier2.OnDownloadStart([]rpc.Event{{Gid: "1"}})
+		asserts.Len(notifyChan, 1)
+		<-notifyChan
+
+		notifier2.OnDownloadPause([]rpc.Event{{Gid: "1"}})
+		asserts.Len(notifyChan, 1)
+		<-notifyChan
+
+		notifier2.OnDownloadStop([]rpc.Event{{Gid: "1"}})
+		asserts.Len(notifyChan, 1)
+		<-notifyChan
+
+		notifier2.OnDownloadComplete([]rpc.Event{{Gid: "1"}})
+		asserts.Len(notifyChan, 1)
+		<-notifyChan
+
+		notifier2.OnDownloadError([]rpc.Event{{Gid: "1"}})
+		asserts.Len(notifyChan, 1)
+		<-notifyChan
+	}
+}
--- a/pkg/filesystem/image_test.go
+++ b/pkg/filesystem/image_test.go
@ -0,0 +1,39 @@
+package filesystem
+
+import (
+	"context"
+	model "github.com/HFO4/cloudreve/models"
+	"github.com/HFO4/cloudreve/pkg/cache"
+	"github.com/HFO4/cloudreve/pkg/filesystem/response"
+	"github.com/stretchr/testify/assert"
+	testMock "github.com/stretchr/testify/mock"
+	"testing"
+)
+
+func TestFileSystem_GetThumb(t *testing.T) {
+	asserts := assert.New(t)
+	fs := &FileSystem{User: &model.User{}}
+
+	// 非图像文件
+	{
+		fs.SetTargetFile(&[]model.File{{}})
+		_, err := fs.GetThumb(context.Background(), 1)
+		asserts.Equal(err, ErrObjectNotExist)
+	}
+
+	// 成功
+	{
+		cache.Set("setting_thumb_width", "10", 0)
+		cache.Set("setting_thumb_height", "10", 0)
+		cache.Set("setting_preview_timeout", "50", 0)
+		testHandller2 := new(FileHeaderMock)
+		testHandller2.On("Thumb", testMock.Anything, "").Return(&response.ContentResponse{}, nil)
+		fs.CleanTargets()
+		fs.SetTargetFile(&[]model.File{{PicInfo: "1,1", Policy: model.Policy{Type: "mock"}}})
+		fs.FileTarget[0].Policy.ID = 1
+		fs.Handler = testHandller2
+		res, err := fs.GetThumb(context.Background(), 1)
+		asserts.NoError(err)
+		asserts.EqualValues(50, res.MaxAge)
+	}
+}