mirror of
https://github.com/c9s/bbgo.git
synced 2024-09-21 08:41:10 +00:00
40 lines
1.2 KiB
Go
40 lines
1.2 KiB
Go
package ftx
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_rawResponse_toSubscribedResp(t *testing.T) {
|
|
input := `{"type": "subscribed", "channel": "orderbook", "market": "BTC/USDT"}`
|
|
var m rawResponse
|
|
assert.NoError(t, json.Unmarshal([]byte(input), &m))
|
|
r := m.toSubscribedResp()
|
|
assert.Equal(t, subscribedRespType, r.Type)
|
|
assert.Equal(t, orderbook, r.Channel)
|
|
assert.Equal(t, "BTC/USDT", r.Market)
|
|
}
|
|
|
|
func Test_rawResponse_toSnapshotResp(t *testing.T) {
|
|
f, err := ioutil.ReadFile("./orderbook_snapshot.json")
|
|
assert.NoError(t, err)
|
|
var m rawResponse
|
|
assert.NoError(t, json.Unmarshal(f, &m))
|
|
r, err := m.toSnapshotResp()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, partialRespType, r.Type)
|
|
assert.Equal(t, orderbook, r.Channel)
|
|
assert.Equal(t, "BTC/USDT", r.Market)
|
|
assert.Equal(t, int64(1614520368), r.Time.Unix())
|
|
assert.Equal(t, int64(2150525410), r.Checksum)
|
|
assert.Len(t, r.Bids, 100)
|
|
assert.Equal(t, []float64{44555.0, 3.3968}, r.Bids[0])
|
|
assert.Equal(t, []float64{44554.0, 0.0561}, r.Bids[1])
|
|
assert.Len(t, r.Asks, 100)
|
|
assert.Equal(t, []float64{44574.0, 0.4591}, r.Asks[0])
|
|
assert.Equal(t, []float64{44579.0, 0.15}, r.Asks[1])
|
|
}
|