bbgo/pkg/exchange/binance/parse_test.go

415 lines
11 KiB
Go
Raw Permalink Normal View History

package binance
import (
"regexp"
"testing"
"time"
"github.com/stretchr/testify/assert"
"git.qtrade.icu/lychiyu/bbgo/pkg/fixedpoint"
"git.qtrade.icu/lychiyu/bbgo/pkg/types"
)
var jsCommentTrimmer = regexp.MustCompile("(?m)//.*$")
func TestMarginResponseParsing(t *testing.T) {
type testcase struct {
input string
}
var testcases = []testcase{
{
input: `{
"e": "executionReport",
"E": 1608545107403,
"s": "BNBUSDT",
"c": "ios_0de017ca7ceb4102b3baa664feb46e65",
"S": "BUY",
"o": "MARKET",
"f": "GTC",
"q": "0.50000000",
"p": "0.00000000",
"P": "0.00000000",
"F": "0.00000000",
"g": -1,
"C": "",
"x": "NEW",
"X": "NEW",
"r": "NONE",
"i": 1147335332,
"l": "0.00000000",
"z": "0.00000000",
"L": "0.00000000",
"n": "0",
"N": null,
"T": 1608545107401,
"t": -1,
"I": 2387303818,
"w": true,
"m": false,
"M": false,
"O": 1608545107401,
"Z": "0.00000000",
"Y": "0.00000000",
"Q": "0.00000000"
}`,
},
{
input: `{
"e": "executionReport",
"E": 1608545107403,
"s": "BNBUSDT",
"c": "ios_0de017ca7ceb4102b3baa664feb46e65",
"S": "BUY",
"o": "MARKET",
"f": "GTC",
"q": "0.50000000",
"p": "0.00000000",
"P": "0.00000000",
"F": "0.00000000",
"g": -1,
"C": "",
"x": "TRADE",
"X": "FILLED",
"r": "NONE",
"i": 1147335332,
"l": "0.50000000",
"z": "0.50000000",
"L": "33.85710000",
"n": "0.00037500",
"N": "BNB",
"T": 1608545107401,
"t": 98414801,
"I": 2387303819,
"w": false,
"m": false,
"M": true,
"O": 1608545107401,
"Z": "16.92855000",
"Y": "16.92855000",
"Q": "0.00000000"
}`,
},
{
input: `{
"e": "outboundAccountInfo",
"E": 1608545107403,
"m": 9,
"t": 10,
"b": 0,
"s": 0,
"T": true,
"W": true,
"D": true,
"u": 1608545107401,
"B": [
{
"a": "BNB",
"f": "89.99345616",
"l": "12.00000000"
},
{
"a": "USDT",
"f": "0.16410063",
"l": "0.00000000"
}
],
"P": []
}`,
},
{
input: `{
"e":"outboundAccountPosition",
"E":1608545107403,
"u":1608545107401,
"B":[{"a":"BNB","f":"89.99345616","l":"12.00000000"},{"a":"USDT","f":"0.16410063","l":"0.00000000"}]
}`,
},
}
for _, testcase := range testcases {
payload := testcase.input
payload = jsCommentTrimmer.ReplaceAllLiteralString(payload, "")
event, err := parseWebSocketEvent([]byte(payload))
assert.NoError(t, err)
assert.NotNil(t, event)
}
}
func TestParseOrderUpdate(t *testing.T) {
payload := `{
"e": "executionReport", // Event type
"E": 1499405658658, // Event time
"s": "ETHBTC", // Symbol
"c": "mUvoqJxFIILMdfAW5iGSOW", // Client order ID
"S": "BUY", // Side
"o": "LIMIT", // Order type
"f": "GTC", // Time in force
"q": "1.00000000", // Order quantity
"p": "0.10264410", // Order price
"P": "0.222", // Stop price
"F": "0.00000000", // Iceberg quantity
"g": -1, // OrderListId
"C": null, // Original client order ID; This is the ID of the order being canceled
"x": "NEW", // Current execution type
"X": "NEW", // Current order status
"r": "NONE", // Order reject reason; will be an error code.
"i": 4293153, // Order ID
"l": "0.00000000", // Last executed quantity
"z": "0.00000000", // Cumulative filled quantity
"L": "0.00000001", // Last executed price
"n": "0", // Commission amount
"N": null, // Commission asset
"T": 1499405658657, // Transaction time
"t": -1, // Trade ID
"I": 8641984, // Ignore
"w": true, // Is the order on the book?
"m": false, // Is this trade the maker side?
"M": true, // Ignore
"O": 1499405658657, // Order creation time
"Z": "0.1", // Cumulative quote asset transacted quantity
"Y": "0.00000000", // Last quote asset transacted quantity (i.e. lastPrice * lastQty)
"Q": "2.0" // Quote Order Quantity
}`
payload = jsCommentTrimmer.ReplaceAllLiteralString(payload, "")
event, err := parseWebSocketEvent([]byte(payload))
assert.NoError(t, err)
assert.NotNil(t, event)
executionReport, ok := event.(*ExecutionReportEvent)
assert.True(t, ok)
assert.NotNil(t, executionReport)
assert.Equal(t, executionReport.Symbol, "ETHBTC")
assert.Equal(t, executionReport.Side, "BUY")
assert.Equal(t, executionReport.ClientOrderID, "mUvoqJxFIILMdfAW5iGSOW")
assert.Equal(t, executionReport.OriginalClientOrderID, "")
assert.Equal(t, executionReport.OrderType, "LIMIT")
assert.Equal(t, executionReport.OrderCreationTime, int64(1499405658657))
assert.Equal(t, executionReport.TimeInForce, "GTC")
assert.Equal(t, executionReport.IcebergQuantity, fixedpoint.MustNewFromString("0.00000000"))
assert.Equal(t, executionReport.OrderQuantity, fixedpoint.MustNewFromString("1.00000000"))
assert.Equal(t, executionReport.QuoteOrderQuantity, fixedpoint.MustNewFromString("2.0"))
assert.Equal(t, executionReport.OrderPrice, fixedpoint.MustNewFromString("0.10264410"))
assert.Equal(t, executionReport.StopPrice, fixedpoint.MustNewFromString("0.222"))
assert.Equal(t, executionReport.IsOnBook, true)
assert.Equal(t, executionReport.IsMaker, false)
assert.Equal(t, executionReport.Ignore, true)
assert.Equal(t, executionReport.CommissionAmount, fixedpoint.MustNewFromString("0"))
assert.Equal(t, executionReport.CommissionAsset, "")
assert.Equal(t, executionReport.CurrentExecutionType, "NEW")
assert.Equal(t, executionReport.CurrentOrderStatus, "NEW")
assert.Equal(t, executionReport.OrderID, int64(4293153))
assert.Equal(t, executionReport.Ignored, int64(8641984))
assert.Equal(t, executionReport.TradeID, int64(-1))
assert.Equal(t, executionReport.TransactionTime, int64(1499405658657))
assert.Equal(t, executionReport.LastExecutedQuantity, fixedpoint.MustNewFromString("0.00000000"))
assert.Equal(t, executionReport.LastExecutedPrice, fixedpoint.MustNewFromString("0.00000001"))
assert.Equal(t, executionReport.CumulativeFilledQuantity, fixedpoint.MustNewFromString("0.00000000"))
assert.Equal(t, executionReport.CumulativeQuoteAssetTransactedQuantity, fixedpoint.MustNewFromString("0.1"))
assert.Equal(t, executionReport.LastQuoteAssetTransactedQuantity, fixedpoint.MustNewFromString("0.00000000"))
orderUpdate, err := executionReport.Order()
assert.NoError(t, err)
assert.NotNil(t, orderUpdate)
}
func TestFuturesResponseParsing(t *testing.T) {
type testcase struct {
input string
}
var testcases = []testcase{
{
input: `{
"e": "ORDER_TRADE_UPDATE",
"T": 1639933384755,
"E": 1639933384763,
"o": {
"s": "BTCUSDT",
"c": "x-NSUYEBKMe60cf610-f5c7-49a4-9c1",
"S": "SELL",
"o": "MARKET",
"f": "GTC",
"q": "0.001",
"p": "0",
"ap": "0",
"sp": "0",
"x": "NEW",
"X": "NEW",
"i": 38541728873,
"l": "0",
"z": "0",
"L": "0",
"T": 1639933384755,
"t": 0,
"b": "0",
"a": "0",
"m": false,
"R": false,
"wt": "CONTRACT_PRICE",
"ot": "MARKET",
"ps": "BOTH",
"cp": false,
"rp": "0",
"pP": false,
"si": 0,
"ss": 0
}
}`,
},
{
input: `{
"e": "ACCOUNT_UPDATE",
"T": 1639933384755,
"E": 1639933384763,
"a": {
"B": [
{
"a": "USDT",
"wb": "86.94966888",
"cw": "86.94966888",
"bc": "0"
}
],
"P": [
{
"s": "BTCUSDT",
"pa": "-0.001",
"ep": "47202.40000",
"cr": "7.78107001",
"up": "-0.00233523",
"mt": "cross",
"iw": "0",
"ps": "BOTH",
"ma": "USDT"
}
],
"m": "ORDER"
}
}`,
},
{
input: `{
"e": "ORDER_TRADE_UPDATE",
"T": 1639933384755,
"E": 1639933384763,
"o": {
"s": "BTCUSDT",
"c": "x-NSUYEBKMe60cf610-f5c7-49a4-9c1",
"S": "SELL",
"o": "MARKET",
"f": "GTC",
"q": "0.001",
"p": "0",
"ap": "47202.40000",
"sp": "0",
"x": "TRADE",
"X": "FILLED",
"i": 38541728873,
"l": "0.001",
"z": "0.001",
"L": "47202.40",
"n": "0.01888095",
"N": "USDT",
"T": 1639933384755,
"t": 1741505949,
"b": "0",
"a": "0",
"m": false,
"R": false,
"wt": "CONTRACT_PRICE",
"ot": "MARKET",
"ps": "BOTH",
"cp": false,
"rp": "0",
"pP": false,
"si": 0,
"ss": 0
}
}`,
},
}
for _, testcase := range testcases {
payload := testcase.input
payload = jsCommentTrimmer.ReplaceAllLiteralString(payload, "")
event, err := parseWebSocketEvent([]byte(payload))
assert.NoError(t, err)
assert.NotNil(t, event)
}
}
func TestParseOrderFuturesUpdate(t *testing.T) {
payload := `{
"e": "ORDER_TRADE_UPDATE",
"T": 1639933384755,
"E": 1639933384763,
"o": {
"s": "BTCUSDT",
"c": "x-NSUYEBKMe60cf610-f5c7-49a4-9c1",
"S": "SELL",
"o": "MARKET",
"f": "GTC",
"q": "0.001",
"p": "0",
"ap": "47202.40000",
"sp": "0",
"x": "TRADE",
"X": "FILLED",
"i": 38541728873,
"l": "0.001",
"z": "0.001",
"L": "47202.40",
"n": "0.01888095",
"N": "USDT",
"T": 1639933384755,
"t": 1741505949,
"b": "0",
"a": "0",
"m": false,
"R": false,
"wt": "CONTRACT_PRICE",
"ot": "MARKET",
"ps": "BOTH",
"cp": false,
"rp": "0",
"pP": false,
"si": 0,
"ss": 0
}
}`
payload = jsCommentTrimmer.ReplaceAllLiteralString(payload, "")
event, err := parseWebSocketEvent([]byte(payload))
assert.NoError(t, err)
assert.NotNil(t, event)
orderTradeEvent, ok := event.(*OrderTradeUpdateEvent)
assert.True(t, ok)
assert.NotNil(t, orderTradeEvent)
assert.Equal(t, "BTCUSDT", orderTradeEvent.OrderTrade.Symbol)
assert.Equal(t, "SELL", orderTradeEvent.OrderTrade.Side)
assert.Equal(t, "x-NSUYEBKMe60cf610-f5c7-49a4-9c1", orderTradeEvent.OrderTrade.ClientOrderID)
assert.Equal(t, "MARKET", orderTradeEvent.OrderTrade.OrderType)
assert.Equal(t, types.NewMillisecondTimestampFromInt(1639933384763), orderTradeEvent.Time)
assert.Equal(t, types.MillisecondTimestamp(time.UnixMilli(1639933384755)), orderTradeEvent.OrderTrade.OrderTradeTime)
assert.Equal(t, fixedpoint.MustNewFromString("0.001"), orderTradeEvent.OrderTrade.OriginalQuantity)
assert.Equal(t, fixedpoint.MustNewFromString("0.001"), orderTradeEvent.OrderTrade.OrderLastFilledQuantity)
assert.Equal(t, fixedpoint.MustNewFromString("0.001"), orderTradeEvent.OrderTrade.OrderFilledAccumulatedQuantity)
assert.Equal(t, "TRADE", orderTradeEvent.OrderTrade.CurrentExecutionType)
assert.Equal(t, "FILLED", orderTradeEvent.OrderTrade.CurrentOrderStatus)
assert.Equal(t, fixedpoint.MustNewFromString("47202.40"), orderTradeEvent.OrderTrade.LastFilledPrice)
assert.Equal(t, int64(38541728873), orderTradeEvent.OrderTrade.OrderId)
assert.Equal(t, int64(1741505949), orderTradeEvent.OrderTrade.TradeId)
orderUpdate, err := orderTradeEvent.OrderFutures()
assert.NoError(t, err)
assert.NotNil(t, orderUpdate)
}