Test: aria2 / filesystem.image

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)

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
+	}
+}

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)
+	}
+}