lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-09-20 08:11:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #1558 from c9s/edwin/bitget/test-query-kline

Browse Source 
TEST: [bitget] add test for kline and ticker
This commit is contained in:
bailantaotao 2024-03-05 20:44:44 +08:00 committed by GitHub
commit b13f0f5085
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 403 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
pkg/exchange/bitget/bitgetapi/v2/testdata/get_k_line_request.json vendored Normal file
View File

@ -0,0 +1,27 @@
{
"code":"00000",
"msg":"success",
"requestTime":1709629803685,
"data":[
[
"1709352000000",
"62308.42",
"62308.43",
"61760",
"62014.17",
"987.377637",
"61283110.57046518",
"61283110.57046518"
],
[
"1709366400000",
"62014.17",
"62122.8",
"61648.26",
"61825.64",
"1271.183413",
"78680550.55539777",
"78680550.55539777"
]
]
}

25
pkg/exchange/bitget/bitgetapi/v2/testdata/get_ticker_request.json vendored Normal file
View File

@ -0,0 +1,25 @@
{
"code":"00000",
"msg":"success",
"requestTime":1709626632870,
"data":[
{
"open":"64654.54",
"symbol":"BTCUSDT",
"high24h":"68686.93",
"low24h":"64583.42",
"lastPr":"66554.03",
"quoteVolume":"1963568540.741927",
"baseVolume":"29439.351448",
"usdtVolume":"1963568540.74192678",
"ts":"1709626631127",
"bidPr":"66554",
"askPr":"66554.07",
"bidSz":"0.000237",
"askSz":"0.08228",
"openUtc":"68245.32",
"changeUtc24h":"-0.02478",
"change24h":"0.00153"
}
]
}

43
pkg/exchange/bitget/bitgetapi/v2/testdata/get_tickers_request.json vendored Normal file
View File

@ -0,0 +1,43 @@
{
"code":"00000",
"msg":"success",
"requestTime":1709626632870,
"data":[
{
"open":"64654.54",
"symbol":"BTCUSDT",
"high24h":"68686.93",
"low24h":"64583.42",
"lastPr":"66554.03",
"quoteVolume":"1963568540.741927",
"baseVolume":"29439.351448",
"usdtVolume":"1963568540.74192678",
"ts":"1709626631127",
"bidPr":"66554",
"askPr":"66554.07",
"bidSz":"0.000237",
"askSz":"0.08228",
"openUtc":"68245.32",
"changeUtc24h":"-0.02478",
"change24h":"0.00153"
},
{
"open":"3506.6",
"symbol":"ETHUSDT",
"high24h":"3740",
"low24h":"3461.17",
"lastPr":"3686.95",
"quoteVolume":"873897317.8273",
"baseVolume":"243220.866",
"usdtVolume":"873897317.827294",
"ts":"1709626631726",
"bidPr":"3686.94",
"askPr":"3686.98",
"bidSz":"1.0046",
"askSz":"1.0015",
"openUtc":"3627.67",
"changeUtc24h":"0.01634",
"change24h":"0.03572"
}
]
}

0
pkg/exchange/bitget/bitgetapi/v2/testdata/get_symbols_request_error.json → pkg/exchange/bitget/bitgetapi/v2/testdata/request_error.json vendored
View File

309
pkg/exchange/bitget/exchange_test.go
View File

@ -5,7 +5,9 @@ import (
"math"
"net/http"
"os"
"strconv"
"testing"
"time"
"github.com/stretchr/testify/assert"
@ -74,7 +76,7 @@ func TestExchange_QueryMarkets(t *testing.T) {
transport := &httptesting.MockTransport{}
ex.client.HttpClient.Transport = transport
f, err := os.ReadFile("bitgetapi/v2/testdata/get_symbols_request_error.json")
f, err := os.ReadFile("bitgetapi/v2/testdata/request_error.json")
assert.NoError(t, err)
transport.GET("/api/v2/spot/public/symbols", func(req *http.Request) (*http.Response, error) {
@ -85,3 +87,308 @@ func TestExchange_QueryMarkets(t *testing.T) {
assert.ErrorContains(t, err, "Invalid IP")
})
}
func TestExchange_QueryTicker(t *testing.T) {
var (
assert = assert.New(t)
ex = New("key", "secret", "passphrase")
url = "/api/v2/spot/market/tickers"
)
t.Run("succeeds", func(t *testing.T) {
transport := &httptesting.MockTransport{}
ex.client.HttpClient.Transport = transport
f, err := os.ReadFile("bitgetapi/v2/testdata/get_ticker_request.json")
assert.NoError(err)
transport.GET(url, func(req *http.Request) (*http.Response, error) {
return httptesting.BuildResponseString(http.StatusOK, string(f)), nil
})
tickers, err := ex.QueryTicker(context.Background(), "BTCUSDT")
assert.NoError(err)
expTicker := &types.Ticker{
Time: types.NewMillisecondTimestampFromInt(1709626631127).Time(),
Volume: fixedpoint.MustNewFromString("29439.351448"),
Last: fixedpoint.MustNewFromString("66554.03"),
Open: fixedpoint.MustNewFromString("64654.54"),
High: fixedpoint.MustNewFromString("68686.93"),
Low: fixedpoint.MustNewFromString("64583.42"),
Buy: fixedpoint.MustNewFromString("66554"),
Sell: fixedpoint.MustNewFromString("66554.07"),
}
assert.Equal(expTicker, tickers)
})
t.Run("unexpected length", func(t *testing.T) {
transport := &httptesting.MockTransport{}
ex.client.HttpClient.Transport = transport
f, err := os.ReadFile("bitgetapi/v2/testdata/get_tickers_request.json")
assert.NoError(err)
transport.GET(url, func(req *http.Request) (*http.Response, error) {
return httptesting.BuildResponseString(http.StatusOK, string(f)), nil
})
_, err = ex.QueryTicker(context.Background(), "BTCUSDT")
assert.ErrorContains(err, "unexpected length of query")
})
t.Run("error", func(t *testing.T) {
transport := &httptesting.MockTransport{}
ex.client.HttpClient.Transport = transport
f, err := os.ReadFile("bitgetapi/v2/testdata/request_error.json")
assert.NoError(err)
transport.GET(url, func(req *http.Request) (*http.Response, error) {
return httptesting.BuildResponseString(http.StatusBadRequest, string(f)), nil
})
_, err = ex.QueryTicker(context.Background(), "BTCUSDT")
assert.ErrorContains(err, "Invalid IP")
})
}
func TestExchange_QueryTickers(t *testing.T) {
var (
assert = assert.New(t)
ex = New("key", "secret", "passphrase")
url = "/api/v2/spot/market/tickers"
expBtcSymbol = "BTCUSDT"
expBtcTicker = types.Ticker{
Time: types.NewMillisecondTimestampFromInt(1709626631127).Time(),
Volume: fixedpoint.MustNewFromString("29439.351448"),
Last: fixedpoint.MustNewFromString("66554.03"),
Open: fixedpoint.MustNewFromString("64654.54"),
High: fixedpoint.MustNewFromString("68686.93"),
Low: fixedpoint.MustNewFromString("64583.42"),
Buy: fixedpoint.MustNewFromString("66554"),
Sell: fixedpoint.MustNewFromString("66554.07"),
}
)
t.Run("succeeds", func(t *testing.T) {
transport := &httptesting.MockTransport{}
ex.client.HttpClient.Transport = transport
f, err := os.ReadFile("bitgetapi/v2/testdata/get_tickers_request.json")
assert.NoError(err)
transport.GET(url, func(req *http.Request) (*http.Response, error) {
return httptesting.BuildResponseString(http.StatusOK, string(f)), nil
})
tickers, err := ex.QueryTickers(context.Background())
assert.NoError(err)
expTickers := map[string]types.Ticker{
expBtcSymbol: expBtcTicker,
"ETHUSDT": {
Time: types.NewMillisecondTimestampFromInt(1709626631726).Time(),
Volume: fixedpoint.MustNewFromString("243220.866"),
Last: fixedpoint.MustNewFromString("3686.95"),
Open: fixedpoint.MustNewFromString("3506.6"),
High: fixedpoint.MustNewFromString("3740"),
Low: fixedpoint.MustNewFromString("3461.17"),
Buy: fixedpoint.MustNewFromString("3686.94"),
Sell: fixedpoint.MustNewFromString("3686.98"),
},
}
assert.Equal(expTickers, tickers)
})
t.Run("succeeds for query one markets", func(t *testing.T) {
transport := &httptesting.MockTransport{}
ex.client.HttpClient.Transport = transport
f, err := os.ReadFile("bitgetapi/v2/testdata/get_ticker_request.json")
assert.NoError(err)
transport.GET(url, func(req *http.Request) (*http.Response, error) {
assert.Contains(req.URL.Query(), "symbol")
assert.Equal(req.URL.Query()["symbol"], []string{expBtcSymbol})
return httptesting.BuildResponseString(http.StatusOK, string(f)), nil
})
tickers, err := ex.QueryTickers(context.Background(), expBtcSymbol)
assert.NoError(err)
expTickers := map[string]types.Ticker{
expBtcSymbol: expBtcTicker,
}
assert.Equal(expTickers, tickers)
})
t.Run("error", func(t *testing.T) {
transport := &httptesting.MockTransport{}
ex.client.HttpClient.Transport = transport
f, err := os.ReadFile("bitgetapi/v2/testdata/request_error.json")
assert.NoError(err)
transport.GET(url, func(req *http.Request) (*http.Response, error) {
return httptesting.BuildResponseString(http.StatusBadRequest, string(f)), nil
})
_, err = ex.QueryTicker(context.Background(), expBtcSymbol)
assert.ErrorContains(err, "Invalid IP")
})
}
func TestExchange_QueryKLines(t *testing.T) {
var (
assert = assert.New(t)
ex = New("key", "secret", "passphrase")
url = "/api/v2/spot/market/candles"
expBtcSymbol = "BTCUSDT"
interval = types.Interval4h
expBtcKlines = []types.KLine{
{
Exchange: types.ExchangeBitget,
Symbol: expBtcSymbol,
StartTime: types.Time(types.NewMillisecondTimestampFromInt(1709352000000).Time()),
EndTime: types.Time(types.NewMillisecondTimestampFromInt(1709352000000).Time().Add(interval.Duration() - time.Millisecond)),
Interval: interval,
Open: fixedpoint.MustNewFromString("62308.42"),
Close: fixedpoint.MustNewFromString("62014.17"),
High: fixedpoint.MustNewFromString("62308.43"),
Low: fixedpoint.MustNewFromString("61760"),
Volume: fixedpoint.MustNewFromString("987.377637"),
QuoteVolume: fixedpoint.MustNewFromString("61283110.57046518"),
Closed: false,
},
{
Exchange: types.ExchangeBitget,
Symbol: expBtcSymbol,
StartTime: types.Time(types.NewMillisecondTimestampFromInt(1709366400000).Time()),
EndTime: types.Time(types.NewMillisecondTimestampFromInt(1709366400000).Time().Add(interval.Duration() - time.Millisecond)),
Interval: interval,
Open: fixedpoint.MustNewFromString("62014.17"),
Close: fixedpoint.MustNewFromString("61825.64"),
High: fixedpoint.MustNewFromString("62122.8"),
Low: fixedpoint.MustNewFromString("61648.26"),
Volume: fixedpoint.MustNewFromString("1271.183413"),
QuoteVolume: fixedpoint.MustNewFromString("78680550.55539777"),
Closed: false,
},
}
)
t.Run("succeeds without time range", func(t *testing.T) {
transport := &httptesting.MockTransport{}
ex.client.HttpClient.Transport = transport
f, err := os.ReadFile("bitgetapi/v2/testdata/get_k_line_request.json")
assert.NoError(err)
transport.GET(url, func(req *http.Request) (*http.Response, error) {
query := req.URL.Query()
assert.Len(query, 3)
assert.Contains(query, "symbol")
assert.Contains(query, "granularity")
assert.Contains(query, "limit")
assert.Equal(query["symbol"], []string{expBtcSymbol})
assert.Equal(query["granularity"], []string{interval.String()})
assert.Equal(query["limit"], []string{strconv.Itoa(defaultKLineLimit)})
return httptesting.BuildResponseString(http.StatusOK, string(f)), nil
})
klines, err := ex.QueryKLines(context.Background(), expBtcSymbol, interval, types.KLineQueryOptions{})
assert.NoError(err)
assert.Equal(expBtcKlines, klines)
})
t.Run("succeeds with time range", func(t *testing.T) {
var (
transport = &httptesting.MockTransport{}
limit = 50
startTime = time.Now()
endTime = startTime.Add(8 * time.Hour)
startTimeMs = strconv.FormatInt(startTime.UnixNano()/int64(time.Millisecond), 10)
endTimeMs = strconv.FormatInt(endTime.UnixNano()/int64(time.Millisecond), 10)
)
ex.client.HttpClient.Transport = transport
f, err := os.ReadFile("bitgetapi/v2/testdata/get_k_line_request.json")
assert.NoError(err)
transport.GET(url, func(req *http.Request) (*http.Response, error) {
query := req.URL.Query()
assert.Len(query, 5)
assert.Contains(query, "symbol")
assert.Contains(query, "granularity")
assert.Contains(query, "limit")
assert.Contains(query, "startTime")
assert.Contains(query, "endTime")
assert.Equal(query["symbol"], []string{expBtcSymbol})
assert.Equal(query["granularity"], []string{interval.String()})
assert.Equal(query["limit"], []string{strconv.Itoa(limit)})
assert.Equal(query["startTime"], []string{startTimeMs})
assert.Equal(query["endTime"], []string{endTimeMs})
return httptesting.BuildResponseString(http.StatusOK, string(f)), nil
})
klines, err := ex.QueryKLines(context.Background(), expBtcSymbol, interval, types.KLineQueryOptions{
Limit: limit,
StartTime: &startTime,
EndTime: &endTime,
})
assert.NoError(err)
assert.Equal(expBtcKlines, klines)
})
t.Run("error", func(t *testing.T) {
transport := &httptesting.MockTransport{}
ex.client.HttpClient.Transport = transport
f, err := os.ReadFile("bitgetapi/v2/testdata/request_error.json")
assert.NoError(err)
transport.GET(url, func(req *http.Request) (*http.Response, error) {
return httptesting.BuildResponseString(http.StatusBadRequest, string(f)), nil
})
_, err = ex.QueryKLines(context.Background(), expBtcSymbol, interval, types.KLineQueryOptions{})
assert.ErrorContains(err, "Invalid IP")
})
t.Run("reach max duration", func(t *testing.T) {
transport := &httptesting.MockTransport{}
ex.client.HttpClient.Transport = transport
transport.GET(url, func(req *http.Request) (*http.Response, error) {
return httptesting.BuildResponse(http.StatusBadRequest, nil), nil
})
before31Days := time.Now().Add(-31 * 24 * time.Hour)
_, err := ex.QueryKLines(context.Background(), expBtcSymbol, types.Interval1m, types.KLineQueryOptions{
EndTime: &before31Days,
})
assert.ErrorContains(err, "are greater than max duration")
})
t.Run("end time before start time", func(t *testing.T) {
transport := &httptesting.MockTransport{}
ex.client.HttpClient.Transport = transport
transport.GET(url, func(req *http.Request) (*http.Response, error) {
return httptesting.BuildResponse(http.StatusBadRequest, nil), nil
})
startTime := time.Now()
endTime := startTime.Add(-time.Hour)
_, err := ex.QueryKLines(context.Background(), expBtcSymbol, types.Interval1m, types.KLineQueryOptions{
StartTime: &startTime,
EndTime: &endTime,
})
assert.ErrorContains(err, "before start time")
})
t.Run("unexpected duraiton", func(t *testing.T) {
_, err := ex.QueryKLines(context.Background(), expBtcSymbol, "87h", types.KLineQueryOptions{})
assert.ErrorContains(err, "not supported")
})
}