wsy182/Cloudreve
1
0
Fork 0
mirror of https://github.com/halejohn/Cloudreve.git synced 2026-01-26 09:34:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: failed test due to modifications

Browse Source
This commit is contained in:
HFO4
2020-02-27 15:34:16 +08:00
parent c4eada2723
commit 2f1f619f2f
5 changed files with 101 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
pkg/filesystem/driver/local/handller_test.go
View File

@@ -120,7 +120,9 @@ func TestHandler_Thumb(t *testing.T) {
func TestHandler_Source(t *testing.T) {
asserts := assert.New(t)
handler := Driver{}
handler := Driver{
Policy: &model.Policy{},
}
ctx := context.Background()
auth.General = auth.HMACAuth{SecretKey: []byte("test")}
@@ -150,6 +152,42 @@ func TestHandler_Source(t *testing.T) {
asserts.Error(err)
asserts.Empty(sourceURL)
}
// 设定了CDN
{
handler.Policy.BaseURL = "https://cqu.edu.cn"
file := model.File{
Model: gorm.Model{
ID: 1,
},
Name: "test.jpg",
}
ctx := context.WithValue(ctx, fsctx.FileModelCtx, file)
baseURL, err := url.Parse("https://cloudreve.org")
asserts.NoError(err)
sourceURL, err := handler.Source(ctx, "", *baseURL, 0, false, 0)
asserts.NoError(err)
asserts.NotEmpty(sourceURL)
asserts.Contains(sourceURL, "sign=")
asserts.Contains(sourceURL, "https://cqu.edu.cn")
}
// 设定了CDN，解析失败
{
handler.Policy.BaseURL = string(0x7f)
file := model.File{
Model: gorm.Model{
ID: 1,
},
Name: "test.jpg",
}
ctx := context.WithValue(ctx, fsctx.FileModelCtx, file)
baseURL, err := url.Parse("https://cloudreve.org")
asserts.NoError(err)
sourceURL, err := handler.Source(ctx, "", *baseURL, 0, false, 0)
asserts.Error(err)
asserts.Empty(sourceURL)
}
}
func TestHandler_GetDownloadURL(t *testing.T) {

30
pkg/filesystem/driver/oss/handler_test.go
View File

@@ -39,6 +39,23 @@ func TestDriver_InitOSSClient(t *testing.T) {
}
}
func TestDriver_CORS(t *testing.T) {
asserts := assert.New(t)
handler := Driver{
Policy: &model.Policy{
AccessKey: "ak",
SecretKey: "sk",
BucketName: "test",
Server: "test.com",
},
}
// 失败
{
asserts.Error(handler.CORS())
}
}
func TestDriver_Token(t *testing.T) {
asserts := assert.New(t)
handler := Driver{
@@ -149,7 +166,18 @@ func TestDriver_Source(t *testing.T) {
asserts.NoError(err)
query := resURL.Query()
asserts.Empty(query.Get("Signature"))
asserts.Empty(query.Get("Expires"))
}
// 正常 指定了CDN域名
{
handler.Policy.BaseURL = "https://cqu.edu.cn"
res, err := handler.Source(context.Background(), "/123", url.URL{}, 10, false, 0)
asserts.NoError(err)
resURL, err := url.Parse(res)
asserts.NoError(err)
query := resURL.Query()
asserts.Empty(query.Get("Signature"))
asserts.Contains(resURL.String(), handler.Policy.BaseURL)
}
}

31
pkg/filesystem/driver/remote/handler_test.go
View File

@@ -84,6 +84,37 @@ func TestHandler_Source(t *testing.T) {
asserts.Contains(res, "api/v3/slave/download/0")
}
// 成功 自定义CDN
{
handler := Driver{
Policy: &model.Policy{Server: "/", BaseURL: "https://cqu.edu.cn"},
AuthInstance: auth.HMACAuth{},
}
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")
asserts.Contains(res, "https://cqu.edu.cn")
}
// 解析失败 自定义CDN
{
handler := Driver{
Policy: &model.Policy{Server: "/", BaseURL: string(0x7f)},
AuthInstance: auth.HMACAuth{},
}
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.Error(err)
asserts.Empty(res)
}
// 成功 预览
{
handler := Driver{