package request import ( "bytes" "encoding/json" "errors" "github.com/HFO4/cloudreve/pkg/serializer" "github.com/stretchr/testify/assert" testMock "github.com/stretchr/testify/mock" "io/ioutil" "net/http" "strings" "testing" ) func TestRemoteCallback(t *testing.T) { asserts := assert.New(t) // 回调成功 { clientMock := ClientMock{} mockResp, _ := json.Marshal(serializer.Response{Code: 0}) clientMock.On( "Request", "POST", "http://test/test/url", testMock.Anything, testMock.Anything, ).Return(Response{ Err: nil, Response: &http.Response{ StatusCode: 200, Body: ioutil.NopCloser(bytes.NewReader(mockResp)), }, }) GeneralClient = clientMock resp := RemoteCallback("http://test/test/url", serializer.RemoteUploadCallback{ SourceName: "source", }) asserts.NoError(resp) clientMock.AssertExpectations(t) } // 服务端返回业务错误 { clientMock := ClientMock{} mockResp, _ := json.Marshal(serializer.Response{Code: 401}) clientMock.On( "Request", "POST", "http://test/test/url", testMock.Anything, testMock.Anything, ).Return(Response{ Err: nil, Response: &http.Response{ StatusCode: 200, Body: ioutil.NopCloser(bytes.NewReader(mockResp)), }, }) GeneralClient = clientMock resp := RemoteCallback("http://test/test/url", serializer.RemoteUploadCallback{ SourceName: "source", }) asserts.EqualValues(401, resp.(serializer.AppError).Code) clientMock.AssertExpectations(t) } // 无法解析回调响应 { clientMock := ClientMock{} clientMock.On( "Request", "POST", "http://test/test/url", testMock.Anything, testMock.Anything, ).Return(Response{ Err: nil, Response: &http.Response{ StatusCode: 200, Body: ioutil.NopCloser(strings.NewReader("mockResp")), }, }) GeneralClient = clientMock resp := RemoteCallback("http://test/test/url", serializer.RemoteUploadCallback{ SourceName: "source", }) asserts.Error(resp) clientMock.AssertExpectations(t) } // HTTP状态码非200 { clientMock := ClientMock{} clientMock.On( "Request", "POST", "http://test/test/url", testMock.Anything, testMock.Anything, ).Return(Response{ Err: nil, Response: &http.Response{ StatusCode: 404, Body: ioutil.NopCloser(strings.NewReader("mockResp")), }, }) GeneralClient = clientMock resp := RemoteCallback("http://test/test/url", serializer.RemoteUploadCallback{ SourceName: "source", }) asserts.Error(resp) clientMock.AssertExpectations(t) } // 无法发起回调 { clientMock := ClientMock{} clientMock.On( "Request", "POST", "http://test/test/url", testMock.Anything, testMock.Anything, ).Return(Response{ Err: errors.New("error"), }) GeneralClient = clientMock resp := RemoteCallback("http://test/test/url", serializer.RemoteUploadCallback{ SourceName: "source", }) asserts.Error(resp) clientMock.AssertExpectations(t) } }