mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-27 17:25:16 +00:00
35 lines
828 B
Go
35 lines
828 B
Go
|
package ftx
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
|
||
|
|
"github.com/c9s/bbgo/pkg/util"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test_toErrorResponse(t *testing.T) {
|
||
|
|
r, err := util.NewResponse(&http.Response{
|
||
|
|
Header: http.Header{},
|
||
|
|
StatusCode: 200,
|
||
|
|
Body: ioutil.NopCloser(bytes.NewReader([]byte(`{"Success": true}`))),
|
||
|
|
})
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
_, err = toErrorResponse(r)
|
||
|
|
assert.EqualError(t, err, "unexpected response content type ")
|
||
|
|
r.Header.Set("content-type", "text/json")
|
||
|
|
|
||
|
|
_, err = toErrorResponse(r)
|
||
|
|
assert.EqualError(t, err, "response.Success should be false")
|
||
|
|
|
||
|
|
r.Body = []byte(`{"error":"Not logged in","Success":false}`)
|
||
|
|
errResp, err := toErrorResponse(r)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.False(t, errResp.IsSuccess)
|
||
|
|
assert.Equal(t, "Not logged in", errResp.ErrorString)
|
||
|
|
}
|