Test: remote handler.source

pkg/filesystem/remote/handler_test.go

@@ -5,8 +5,10 @@ import (
 	model "github.com/HFO4/cloudreve/models"
 	"github.com/HFO4/cloudreve/pkg/auth"
 	"github.com/HFO4/cloudreve/pkg/cache"
+	"github.com/HFO4/cloudreve/pkg/filesystem/fsctx"
 	"github.com/HFO4/cloudreve/pkg/serializer"
 	"github.com/stretchr/testify/assert"
+	"net/url"
 	"testing"
 )
 
@@ -41,3 +43,45 @@ func TestHandler_Token(t *testing.T) {
 	}
 
 }
+
+func TestHandler_Source(t *testing.T) {
+	asserts := assert.New(t)
+	auth.General = auth.HMACAuth{SecretKey: []byte("test")}
+
+	// 无法获取上下文
+	{
+		handler := Handler{}
+		ctx := context.Background()
+		res, err := handler.Source(ctx, "", url.URL{}, 0, true, 0)
+		asserts.Error(err)
+		asserts.Empty(res)
+	}
+
+	// 成功
+	{
+		handler := Handler{
+			Policy: &model.Policy{Server: "/"},
+		}
+		file := model.File{
+			SourceName: "1.txt",
+		}
+		ctx := context.WithValue(context.Background(), fsctx.FileModelCtx, file)
+		res, err := handler.Source(ctx, "", url.URL{}, 10, true, 0)
+		asserts.NoError(err)
+		asserts.Contains(res, "api/v3/slave/download/0")
+	}
+
+	// 成功 预览
+	{
+		handler := Handler{
+			Policy: &model.Policy{Server: "/"},
+		}
+		file := model.File{
+			SourceName: "1.txt",
+		}
+		ctx := context.WithValue(context.Background(), fsctx.FileModelCtx, file)
+		res, err := handler.Source(ctx, "", url.URL{}, 10, false, 0)
+		asserts.NoError(err)
+		asserts.Contains(res, "api/v3/slave/source/0")
+	}
+}