mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 23:16:45 +00:00
fc0f262c42
* set limiter to first level mock data list and updated unit tests * address nested slices length limit * minor fix recording file and update unit tests * minor updates on unit tests * re-record mock files and minor fix on the unit tests ti adapt the mock data change * improve http recording limit value and fix issues with mock data in binance * added MockDataSliceLimit in request items and resolve minor unit test issues * resolve missed conflict * rename mock variables, resolve unit test issues, and other updates * minor fix to CheckJSON and update unit tests * minor unit test fix * further optimization on mock CheckJSON method, unit tests, and re-record poloniex * common and recording unit tests fix * minor linter issues fix * unit tests format fix * fix miscellaneous error * unit tests fix and minor docs update * re-record and reduce mock file size * indentation fix * minor assertion test fix * reverted log.Printf line in live testing * rename variables * update NewVCRServer unit test * replace string comparison with *net.OpError check * restructur net error test * exchanges/mock: Remove redundant error assertion message in TestNewVCRServer --------- Co-authored-by: Adrian Gallagher <adrian.gallagher@thrasher.io>
93 lines
2.7 KiB
Go
93 lines
2.7 KiB
Go
package mock
|
|
|
|
import (
|
|
"bytes"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/thrasher-corp/gocryptotrader/common"
|
|
"github.com/thrasher-corp/gocryptotrader/encoding/json"
|
|
)
|
|
|
|
type responsePayload struct {
|
|
Price float64 `json:"price"`
|
|
Amount float64 `json:"amount"`
|
|
Currency string `json:"currency"`
|
|
}
|
|
|
|
const (
|
|
queryString = "currency=btc&command=getprice"
|
|
testFile = "test.json"
|
|
)
|
|
|
|
func TestNewVCRServer(t *testing.T) {
|
|
_, _, err := NewVCRServer("")
|
|
assert.ErrorIs(t, err, errJSONMockFilePathRequired)
|
|
|
|
// Set up mock data
|
|
test1 := VCRMock{}
|
|
test1.Routes = make(map[string]map[string][]HTTPResponse)
|
|
test1.Routes["/test"] = make(map[string][]HTTPResponse)
|
|
|
|
rp, err := json.Marshal(responsePayload{
|
|
Price: 8000.0,
|
|
Amount: 1,
|
|
Currency: "bitcoin",
|
|
})
|
|
require.NoError(t, err, "Marshal must not error")
|
|
|
|
testValue := HTTPResponse{Data: rp, QueryString: queryString, BodyParams: queryString}
|
|
test1.Routes["/test"][http.MethodGet] = []HTTPResponse{testValue}
|
|
|
|
payload, err := json.Marshal(test1)
|
|
require.NoError(t, err, "Marshal must not error")
|
|
|
|
err = os.WriteFile(testFile, payload, os.ModePerm)
|
|
require.NoError(t, err, "WriteFile must not error")
|
|
|
|
deets, client, err := NewVCRServer(testFile)
|
|
assert.NoError(t, err, "NewVCRServer should not error")
|
|
|
|
err = common.SetHTTPClient(client) // Set common package global HTTP Client
|
|
require.NoError(t, err)
|
|
|
|
_, err = common.SendHTTPRequest(t.Context(),
|
|
http.MethodGet,
|
|
"http://localhost:300/somethingElse?"+queryString,
|
|
nil,
|
|
bytes.NewBufferString(""), true)
|
|
var netErr *net.OpError
|
|
assert.ErrorAs(t, err, &netErr, "SendHTTPRequest should return a net.OpError for an invalid host")
|
|
|
|
// Expected good outcome
|
|
r, err := common.SendHTTPRequest(t.Context(),
|
|
http.MethodGet,
|
|
deets,
|
|
nil,
|
|
bytes.NewBufferString(""), true)
|
|
assert.NoError(t, err, "SendHTTPRequest should not error")
|
|
assert.Contains(t, string(r), "404 page not found", "SendHTTPRequest return should only contain 404")
|
|
|
|
r, err = common.SendHTTPRequest(t.Context(),
|
|
http.MethodGet,
|
|
deets+"/test?"+queryString,
|
|
nil,
|
|
bytes.NewBufferString(""), true)
|
|
assert.NoError(t, err, "SendHTTPRequest should not error")
|
|
|
|
var res responsePayload
|
|
err = json.Unmarshal(r, &res)
|
|
assert.NoError(t, err, "Unmarshal should not error")
|
|
assert.Equalf(t, 8000.0, res.Price, "response error expected 8000 but received: %f", res.Price)
|
|
assert.Equalf(t, 1.0, res.Amount, "response error expected 1 but received: %f", res.Amount)
|
|
assert.Equalf(t, "bitcoin", res.Currency, "response error expected \"bitcoin\" but received: %s", res.Currency)
|
|
|
|
// clean up test.json file
|
|
err = os.Remove(testFile)
|
|
require.NoError(t, err, "Remove testFile must not error")
|
|
}
|