You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
cloudreve/pkg/request/request_test.go

239 lines
5.0 KiB

package request
import (
"context"
"errors"
"github.com/cloudreve/Cloudreve/v3/pkg/cache"
"io"
"io/ioutil"
"net/http"
"strings"
"testing"
"time"
"github.com/cloudreve/Cloudreve/v3/pkg/auth"
"github.com/stretchr/testify/assert"
testMock "github.com/stretchr/testify/mock"
)
type ClientMock struct {
testMock.Mock
}
func (m ClientMock) Request(method, target string, body io.Reader, opts ...Option) *Response {
args := m.Called(method, target, body, opts)
return args.Get(0).(*Response)
}
func TestWithTimeout(t *testing.T) {
asserts := assert.New(t)
options := newDefaultOption()
WithTimeout(time.Duration(5) * time.Second).apply(options)
asserts.Equal(time.Duration(5)*time.Second, options.timeout)
}
func TestWithHeader(t *testing.T) {
asserts := assert.New(t)
options := newDefaultOption()
WithHeader(map[string][]string{"Origin": []string{"123"}}).apply(options)
asserts.Equal(http.Header{"Origin": []string{"123"}}, options.header)
}
func TestWithContentLength(t *testing.T) {
asserts := assert.New(t)
options := newDefaultOption()
WithContentLength(10).apply(options)
asserts.EqualValues(10, options.contentLength)
}
func TestWithContext(t *testing.T) {
asserts := assert.New(t)
options := newDefaultOption()
WithContext(context.Background()).apply(options)
asserts.NotNil(options.ctx)
}
func TestHTTPClient_Request(t *testing.T) {
asserts := assert.New(t)
client := NewClient(WithSlaveMeta("test"))
// 正常
{
resp := client.Request(
"POST",
"/test",
strings.NewReader(""),
WithContentLength(0),
WithEndpoint("http://cloudreveisnotexist.com"),
WithTimeout(time.Duration(1)*time.Microsecond),
WithCredential(auth.HMACAuth{SecretKey: []byte("123")}, 10),
WithoutHeader([]string{"origin", "origin"}),
)
asserts.Error(resp.Err)
asserts.Nil(resp.Response)
}
// 正常 带有ctx
{
resp := client.Request(
"GET",
"http://cloudreveisnotexist.com",
strings.NewReader(""),
WithTimeout(time.Duration(1)*time.Microsecond),
WithCredential(auth.HMACAuth{SecretKey: []byte("123")}, 10),
WithContext(context.Background()),
)
asserts.Error(resp.Err)
asserts.Nil(resp.Response)
}
}
func TestResponse_GetResponse(t *testing.T) {
asserts := assert.New(t)
// 直接返回错误
{
resp := Response{
Err: errors.New("error"),
}
content, err := resp.GetResponse()
asserts.Empty(content)
asserts.Error(err)
}
// 正常
{
resp := Response{
Response: &http.Response{Body: ioutil.NopCloser(strings.NewReader("123"))},
}
content, err := resp.GetResponse()
asserts.Equal("123", content)
asserts.NoError(err)
}
}
func TestResponse_CheckHTTPResponse(t *testing.T) {
asserts := assert.New(t)
// 直接返回错误
{
resp := Response{
Err: errors.New("error"),
}
res := resp.CheckHTTPResponse(200)
asserts.Error(res.Err)
}
// 404错误
{
resp := Response{
Response: &http.Response{StatusCode: 404},
}
res := resp.CheckHTTPResponse(200)
asserts.Error(res.Err)
}
// 通过
{
resp := Response{
Response: &http.Response{StatusCode: 200},
}
res := resp.CheckHTTPResponse(200)
asserts.NoError(res.Err)
}
}
func TestResponse_GetRSCloser(t *testing.T) {
asserts := assert.New(t)
// 直接返回错误
{
resp := Response{
Err: errors.New("error"),
}
res, err := resp.GetRSCloser()
asserts.Error(err)
asserts.Nil(res)
}
// 正常
{
resp := Response{
Response: &http.Response{ContentLength: 3, Body: ioutil.NopCloser(strings.NewReader("123"))},
}
res, err := resp.GetRSCloser()
asserts.NoError(err)
content, err := ioutil.ReadAll(res)
asserts.NoError(err)
asserts.Equal("123", string(content))
offset, err := res.Seek(0, 0)
asserts.NoError(err)
asserts.Equal(int64(0), offset)
offset, err = res.Seek(0, 2)
asserts.NoError(err)
asserts.Equal(int64(3), offset)
_, err = res.Seek(1, 2)
asserts.Error(err)
asserts.NoError(res.Close())
}
}
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)
}
}
func TestNopRSCloser_SetFirstFakeChunk(t *testing.T) {
asserts := assert.New(t)
rsc := NopRSCloser{
status: &rscStatus{},
}
rsc.SetFirstFakeChunk()
asserts.True(rsc.status.IgnoreFirst)
rsc.SetContentLength(20)
asserts.EqualValues(20, rsc.status.Size)
}
func TestBlackHole(t *testing.T) {
a := assert.New(t)
cache.Set("setting_reset_after_upload_failed", "true", 0)
a.NotPanics(func() {
BlackHole(strings.NewReader("TestBlackHole"))
})
}