Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
hexbee
Cloudreve
提交
ca9f44c0
C
Cloudreve
项目概览
hexbee
/
Cloudreve
通知
2
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
C
Cloudreve
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
ca9f44c0
编写于
12月 29, 2019
作者:
H
HFO4
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Test: SlaveAfterUpload hook
上级
64342fa8
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
60 addition
and
7 deletion
+60
-7
pkg/filesystem/hooks_test.go
pkg/filesystem/hooks_test.go
+53
-0
pkg/request/callback.go
pkg/request/callback.go
+1
-1
pkg/request/callback_test.go
pkg/request/callback_test.go
+5
-5
pkg/request/request.go
pkg/request/request.go
+1
-1
未找到文件。
pkg/filesystem/hooks_test.go
浏览文件 @
ca9f44c0
...
...
@@ -9,11 +9,16 @@ import (
"github.com/HFO4/cloudreve/pkg/conf"
"github.com/HFO4/cloudreve/pkg/filesystem/fsctx"
"github.com/HFO4/cloudreve/pkg/filesystem/local"
"github.com/HFO4/cloudreve/pkg/request"
"github.com/HFO4/cloudreve/pkg/serializer"
"github.com/jinzhu/gorm"
"github.com/stretchr/testify/assert"
testMock
"github.com/stretchr/testify/mock"
"io"
"io/ioutil"
"net/http"
"os"
"strings"
"testing"
)
...
...
@@ -540,3 +545,51 @@ func TestHookSlaveUploadValidate(t *testing.T) {
}
}
type
ClientMock
struct
{
testMock
.
Mock
}
func
(
m
ClientMock
)
Request
(
method
,
target
string
,
body
io
.
Reader
,
opts
...
request
.
Option
)
request
.
Response
{
args
:=
m
.
Called
(
method
,
target
,
body
,
opts
)
return
args
.
Get
(
0
)
.
(
request
.
Response
)
}
func
TestSlaveAfterUpload
(
t
*
testing
.
T
)
{
asserts
:=
assert
.
New
(
t
)
conf
.
SystemConfig
.
Mode
=
"slave"
fs
,
err
:=
NewAnonymousFileSystem
()
conf
.
SystemConfig
.
Mode
=
"master"
asserts
.
NoError
(
err
)
// 成功
{
clientMock
:=
ClientMock
{}
clientMock
.
On
(
"Request"
,
"POST"
,
"http://test/callbakc"
,
testMock
.
Anything
,
testMock
.
Anything
,
)
.
Return
(
request
.
Response
{
Err
:
nil
,
Response
:
&
http
.
Response
{
StatusCode
:
200
,
Body
:
ioutil
.
NopCloser
(
strings
.
NewReader
(
`{"code":0}`
)),
},
})
request
.
GeneralClient
=
clientMock
ctx
:=
context
.
WithValue
(
context
.
Background
(),
fsctx
.
FileHeaderCtx
,
local
.
FileStream
{
Size
:
10
,
VirtualPath
:
"/my"
,
Name
:
"test.txt"
,
})
ctx
=
context
.
WithValue
(
ctx
,
fsctx
.
UploadPolicyCtx
,
serializer
.
UploadPolicy
{
CallbackURL
:
"http://test/callbakc"
,
})
ctx
=
context
.
WithValue
(
ctx
,
fsctx
.
SavePathCtx
,
"/not_exist"
)
err
:=
SlaveAfterUpload
(
ctx
,
fs
)
clientMock
.
AssertExpectations
(
t
)
asserts
.
NoError
(
err
)
}
}
pkg/request/callback.go
浏览文件 @
ca9f44c0
...
...
@@ -19,7 +19,7 @@ func RemoteCallback(url string, body serializer.UploadCallback) error {
return
serializer
.
NewError
(
serializer
.
CodeCallbackError
,
"无法编码回调正文"
,
err
)
}
resp
:=
g
eneralClient
.
Request
(
resp
:=
G
eneralClient
.
Request
(
"POST"
,
url
,
bytes
.
NewReader
(
callbackBody
),
...
...
pkg/request/callback_test.go
浏览文件 @
ca9f44c0
...
...
@@ -33,7 +33,7 @@ func TestRemoteCallback(t *testing.T) {
Body
:
ioutil
.
NopCloser
(
bytes
.
NewReader
(
mockResp
)),
},
})
g
eneralClient
=
clientMock
G
eneralClient
=
clientMock
resp
:=
RemoteCallback
(
"http://test/test/url"
,
serializer
.
UploadCallback
{
SourceName
:
"source"
,
})
...
...
@@ -58,7 +58,7 @@ func TestRemoteCallback(t *testing.T) {
Body
:
ioutil
.
NopCloser
(
bytes
.
NewReader
(
mockResp
)),
},
})
g
eneralClient
=
clientMock
G
eneralClient
=
clientMock
resp
:=
RemoteCallback
(
"http://test/test/url"
,
serializer
.
UploadCallback
{
SourceName
:
"source"
,
})
...
...
@@ -82,7 +82,7 @@ func TestRemoteCallback(t *testing.T) {
Body
:
ioutil
.
NopCloser
(
strings
.
NewReader
(
"mockResp"
)),
},
})
g
eneralClient
=
clientMock
G
eneralClient
=
clientMock
resp
:=
RemoteCallback
(
"http://test/test/url"
,
serializer
.
UploadCallback
{
SourceName
:
"source"
,
})
...
...
@@ -106,7 +106,7 @@ func TestRemoteCallback(t *testing.T) {
Body
:
ioutil
.
NopCloser
(
strings
.
NewReader
(
"mockResp"
)),
},
})
g
eneralClient
=
clientMock
G
eneralClient
=
clientMock
resp
:=
RemoteCallback
(
"http://test/test/url"
,
serializer
.
UploadCallback
{
SourceName
:
"source"
,
})
...
...
@@ -126,7 +126,7 @@ func TestRemoteCallback(t *testing.T) {
)
.
Return
(
Response
{
Err
:
errors
.
New
(
"error"
),
})
g
eneralClient
=
clientMock
G
eneralClient
=
clientMock
resp
:=
RemoteCallback
(
"http://test/test/url"
,
serializer
.
UploadCallback
{
SourceName
:
"source"
,
})
...
...
pkg/request/request.go
浏览文件 @
ca9f44c0
...
...
@@ -7,7 +7,7 @@ import (
"time"
)
var
g
eneralClient
Client
=
HTTPClient
{}
var
G
eneralClient
Client
=
HTTPClient
{}
// Response 请求的响应或错误信息
type
Response
struct
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录