Infrastructure/Cloudreve
0
Fork 0
Code Issues Pull requests Activity

Test: remote handler put

Browse source
This commit is contained in:
HFO4 2020-01-13 11:03:51 +08:00
parent 37e78cb39b
commit 57d2ad9e5f
4 changed files with 142 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
pkg/filesystem/remote/handler.go
View file

@ -35,8 +35,6 @@ func (handler Handler) getAPIUrl(scope string, routes ...string) string {
var controller *url.URL
switch scope {
case "upload":
controller, _ = url.Parse("/api/v3/slave/upload")
case "delete":
controller, _ = url.Parse("/api/v3/slave/delete")
case "thumb":
@ -53,7 +51,6 @@ func (handler Handler) getAPIUrl(scope string, routes ...string) string {
}
// Get 获取文件内容
// TODO 测试
func (handler Handler) Get(ctx context.Context, path string) (response.RSCloser, error) {
// 尝试获取速度限制 TODO 是否需要在这里限制？
speedLimit := 0
@ -109,7 +106,7 @@ func (handler Handler) Put(ctx context.Context, file io.ReadCloser, dst string,
// 上传文件
resp, err := handler.Client.Request(
"POST",
handler.getAPIUrl("upload"),
handler.Policy.GetUploadURL(),
file,
request.WithHeader(map[string][]string{
"Authorization": {credential.Token},

101
pkg/filesystem/remote/handler_test.go
View file

@ -251,3 +251,104 @@ func TestHandler_Get(t *testing.T) {
asserts.Error(err)
}
}
func TestHandler_Put(t *testing.T) {
asserts := assert.New(t)
handler := Handler{
Policy: &model.Policy{
Type: "remote",
SecretKey: "test",
Server: "http://test.com",
},
AuthInstance: auth.HMACAuth{},
}
ctx := context.Background()
asserts.NoError(cache.Set("setting_upload_credential_timeout", "3600", 0))
// 成功
{
ctx = context.WithValue(ctx, fsctx.UserCtx, model.User{})
clientMock := ClientMock{}
clientMock.On(
"Request",
"POST",
"http://test.com/api/v3/slave/upload",
testMock.Anything,
testMock.Anything,
).Return(&request.Response{
Err: nil,
Response: &http.Response{
StatusCode: 200,
Body: ioutil.NopCloser(strings.NewReader(`{"code":0}`)),
},
})
handler.Client = clientMock
err := handler.Put(ctx, ioutil.NopCloser(strings.NewReader("test input file")), "/", 15)
clientMock.AssertExpectations(t)
asserts.NoError(err)
}
// 请求失败
{
ctx = context.WithValue(ctx, fsctx.UserCtx, model.User{})
clientMock := ClientMock{}
clientMock.On(
"Request",
"POST",
"http://test.com/api/v3/slave/upload",
testMock.Anything,
testMock.Anything,
).Return(&request.Response{
Err: nil,
Response: &http.Response{
StatusCode: 404,
Body: ioutil.NopCloser(strings.NewReader(`{"code":0}`)),
},
})
handler.Client = clientMock
err := handler.Put(ctx, ioutil.NopCloser(strings.NewReader("test input file")), "/", 15)
clientMock.AssertExpectations(t)
asserts.Error(err)
}
// 返回错误
{
ctx = context.WithValue(ctx, fsctx.UserCtx, model.User{})
clientMock := ClientMock{}
clientMock.On(
"Request",
"POST",
"http://test.com/api/v3/slave/upload",
testMock.Anything,
testMock.Anything,
).Return(&request.Response{
Err: nil,
Response: &http.Response{
StatusCode: 200,
Body: ioutil.NopCloser(strings.NewReader(`{"code":1}`)),
},
})
handler.Client = clientMock
err := handler.Put(ctx, ioutil.NopCloser(strings.NewReader("test input file")), "/", 15)
clientMock.AssertExpectations(t)
asserts.Error(err)
}
}
func TestHandler_Thumb(t *testing.T) {
asserts := assert.New(t)
handler := Handler{
Policy: &model.Policy{
Type: "remote",
SecretKey: "test",
Server: "http://test.com",
},
AuthInstance: auth.HMACAuth{},
}
ctx := context.Background()
asserts.NoError(cache.Set("setting_slave_api_timeout", "60", 0))
resp, err := handler.Thumb(ctx, "/1.txt")
asserts.NoError(err)
asserts.True(resp.Redirect)
}

2
pkg/request/request.go
View file

@ -90,7 +90,6 @@ func WithHeader(header http.Header) Option {
}
// WithContentLength 设置请求大小
// TODO 测试
func WithContentLength(s int64) Option {
return optionFunc(func(o *options) {
o.contentLength = s
@ -170,7 +169,6 @@ func (resp *Response) CheckHTTPResponse(status int) *Response {
}
// DecodeResponse 尝试解析为serializer.Response，并对状态码进行检查
// TODO 测试
func (resp *Response) DecodeResponse() (*serializer.Response, error) {
if resp.Err != nil {
return nil, resp.Err

44
pkg/request/request_test.go
View file

@ -37,12 +37,11 @@ func TestWithHeader(t *testing.T) {
asserts.Equal(http.Header{"Origin": []string{"123"}}, options.header)
}
func TestWithCredential(t *testing.T) {
func TestWithContentLength(t *testing.T) {
asserts := assert.New(t)
options := newDefaultOption()
WithCredential(auth.HMACAuth{SecretKey: []byte("123")}, 10).apply(options)
asserts.Equal(auth.HMACAuth{SecretKey: []byte("123")}, options.sign)
asserts.EqualValues(10, options.signTTL)
WithContentLength(10).apply(options)
asserts.EqualValues(10, options.contentLength)
}
func TestWithContext(t *testing.T) {
@ -175,3 +174,40 @@ func TestResponse_GetRSCloser(t *testing.T) {
}
}
func TestResponse_DecodeResponse(t *testing.T) {
asserts := assert.New(t)
// 直接返回错误
{
resp := Response{Err: errors.New("error")}
response, err := resp.DecodeResponse()
asserts.Error(err)
asserts.Nil(response)
}
// 无法解析响应
{
resp := Response{
Response: &http.Response{
Body: ioutil.NopCloser(strings.NewReader("test")),
},
}
response, err := resp.DecodeResponse()
asserts.Error(err)
asserts.Nil(response)
}
// 成功
{
resp := Response{
Response: &http.Response{
Body: ioutil.NopCloser(strings.NewReader("{\"code\":0}")),
},
}
response, err := resp.DecodeResponse()
asserts.NoError(err)
asserts.NotNil(response)
asserts.Equal(0, response.Code)
}
}