mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 15:09:42 +00:00
b281759573
* gateio: Add websocket orderbook update manager * RM println * glorious: nit * Adds delivery futures update processing as well * Change to const value for delivery * Drop check out of order, can reinstate if required. * Adds in validation methods to ensure config changes are correct when expanding templates and return errors with correct info if not. * fix some things and add in todo when this gets updated * fix spelling * linter: fix * gk: initial nits * gk: nits shift to template only verification with funcmap, rm interface for single sub checking. * rm unused error * linter: fix * update to const frequency * gk: wrap with panic and single invocation in template, change name * gk: nits to check across stored subs with incoming subs * linter: fix * updates names, makes things slightly more efficient and adds tests * linter: fix * gk: sexc patch v2 * glorious: nits * gk: nits * Update exchanges/subscription/template.go Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com> * gk: nits * linter: make peace with linter regulations * glorious: Add TODO for future template integration * glorious: commentary nits * fix name * give me a break, have a kit kat * revert whoops * update wording on comment * revert secondary call to expand templates and update tests * misc lint: fix * Add spot orderbook update interval for 20ms, expand tests, piggy back limit/level off loaded subscription. Thanks to @thrasher- * linter/spell: fix * ai nits: drop go routine on mtx RUnlock * Update exchanges/gateio/ws_ob_update_manager.go Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com> * gk: revert to 100ms from 20ms waiting for config upgrade patch * test: fix * cranktakular: nits * strings quoted in fmt call * thrasher-: nits * Update exchanges/gateio/gateio_test.go Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com> * Update exchanges/gateio/gateio_test.go Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com> * gk: nits --------- Co-authored-by: Ryan O'Hara-Reid <ryan.oharareid@thrasher.io> Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com>
237 lines
8.6 KiB
Go
237 lines
8.6 KiB
Go
package gateio
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/thrasher-corp/gocryptotrader/common"
|
|
"github.com/thrasher-corp/gocryptotrader/currency"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/asset"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/order"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/sharedtestvalues"
|
|
testexch "github.com/thrasher-corp/gocryptotrader/internal/testing/exchange"
|
|
)
|
|
|
|
func TestWebsocketLogin(t *testing.T) {
|
|
t.Parallel()
|
|
err := g.websocketLogin(t.Context(), nil, "")
|
|
require.ErrorIs(t, err, common.ErrNilPointer)
|
|
|
|
sharedtestvalues.SkipTestIfCredentialsUnset(t, g, canManipulateRealOrders)
|
|
|
|
g := newExchangeWithWebsocket(t, asset.Spot) //nolint:govet // Intentional shadow to avoid future copy/paste mistakes
|
|
|
|
c, err := g.Websocket.GetConnection(asset.Spot)
|
|
require.NoError(t, err)
|
|
|
|
err = g.websocketLogin(t.Context(), c, "")
|
|
require.ErrorIs(t, err, errChannelEmpty)
|
|
|
|
err = g.websocketLogin(t.Context(), c, "spot.login")
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestWebsocketSpotSubmitOrder(t *testing.T) {
|
|
t.Parallel()
|
|
_, err := g.WebsocketSpotSubmitOrder(t.Context(), &CreateOrderRequest{})
|
|
require.ErrorIs(t, err, currency.ErrCurrencyPairEmpty)
|
|
out := &CreateOrderRequest{CurrencyPair: currency.NewPair(currency.NewCode("GT"), currency.USDT).Format(currency.PairFormat{Uppercase: true, Delimiter: "_"})}
|
|
_, err = g.WebsocketSpotSubmitOrder(t.Context(), out)
|
|
require.ErrorIs(t, err, order.ErrSideIsInvalid)
|
|
out.Side = strings.ToLower(order.Sell.String())
|
|
_, err = g.WebsocketSpotSubmitOrder(t.Context(), out)
|
|
require.ErrorIs(t, err, errInvalidAmount)
|
|
out.Amount = 1
|
|
out.Type = "limit"
|
|
_, err = g.WebsocketSpotSubmitOrder(t.Context(), out)
|
|
require.ErrorIs(t, err, errInvalidPrice)
|
|
out.Price = 100
|
|
|
|
sharedtestvalues.SkipTestIfCredentialsUnset(t, g, canManipulateRealOrders)
|
|
|
|
g := newExchangeWithWebsocket(t, asset.Spot) //nolint:govet // Intentional shadow to avoid future copy/paste mistakes
|
|
|
|
got, err := g.WebsocketSpotSubmitOrder(t.Context(), out)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, got)
|
|
}
|
|
|
|
func TestWebsocketSpotSubmitOrders(t *testing.T) {
|
|
t.Parallel()
|
|
_, err := g.WebsocketSpotSubmitOrders(t.Context())
|
|
require.ErrorIs(t, err, errOrdersEmpty)
|
|
out := &CreateOrderRequest{}
|
|
_, err = g.WebsocketSpotSubmitOrders(t.Context(), out)
|
|
require.ErrorIs(t, err, currency.ErrCurrencyPairEmpty)
|
|
out.CurrencyPair = currency.NewBTCUSDT()
|
|
_, err = g.WebsocketSpotSubmitOrders(t.Context(), out)
|
|
require.ErrorIs(t, err, order.ErrSideIsInvalid)
|
|
out.Side = strings.ToLower(order.Buy.String())
|
|
_, err = g.WebsocketSpotSubmitOrders(t.Context(), out)
|
|
require.ErrorIs(t, err, errInvalidAmount)
|
|
out.Amount = 0.0003
|
|
out.Type = "limit"
|
|
_, err = g.WebsocketSpotSubmitOrders(t.Context(), out)
|
|
require.ErrorIs(t, err, errInvalidPrice)
|
|
out.Price = 20000
|
|
|
|
sharedtestvalues.SkipTestIfCredentialsUnset(t, g, canManipulateRealOrders)
|
|
|
|
g := newExchangeWithWebsocket(t, asset.Spot) //nolint:govet // Intentional shadow to avoid future copy/paste mistakes
|
|
|
|
// test single order
|
|
got, err := g.WebsocketSpotSubmitOrders(t.Context(), out)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, got)
|
|
|
|
// test batch orders
|
|
got, err = g.WebsocketSpotSubmitOrders(t.Context(), out, out)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, got)
|
|
}
|
|
|
|
func TestWebsocketSpotCancelOrder(t *testing.T) {
|
|
t.Parallel()
|
|
_, err := g.WebsocketSpotCancelOrder(t.Context(), "", currency.EMPTYPAIR, "")
|
|
require.ErrorIs(t, err, order.ErrOrderIDNotSet)
|
|
_, err = g.WebsocketSpotCancelOrder(t.Context(), "1337", currency.EMPTYPAIR, "")
|
|
require.ErrorIs(t, err, currency.ErrCurrencyPairEmpty)
|
|
|
|
sharedtestvalues.SkipTestIfCredentialsUnset(t, g, canManipulateRealOrders)
|
|
|
|
g := newExchangeWithWebsocket(t, asset.Spot) //nolint:govet // Intentional shadow to avoid future copy/paste mistakes
|
|
|
|
got, err := g.WebsocketSpotCancelOrder(t.Context(), "644913098758", BTCUSDT, "")
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, got)
|
|
}
|
|
|
|
func TestWebsocketSpotCancelAllOrdersByIDs(t *testing.T) {
|
|
t.Parallel()
|
|
_, err := g.WebsocketSpotCancelAllOrdersByIDs(t.Context(), []WebsocketOrderBatchRequest{})
|
|
require.ErrorIs(t, err, errNoOrdersToCancel)
|
|
out := WebsocketOrderBatchRequest{}
|
|
_, err = g.WebsocketSpotCancelAllOrdersByIDs(t.Context(), []WebsocketOrderBatchRequest{out})
|
|
require.ErrorIs(t, err, order.ErrOrderIDNotSet)
|
|
out.OrderID = "1337"
|
|
_, err = g.WebsocketSpotCancelAllOrdersByIDs(t.Context(), []WebsocketOrderBatchRequest{out})
|
|
require.ErrorIs(t, err, currency.ErrCurrencyPairEmpty)
|
|
|
|
out.Pair = BTCUSDT
|
|
|
|
sharedtestvalues.SkipTestIfCredentialsUnset(t, g, canManipulateRealOrders)
|
|
|
|
g := newExchangeWithWebsocket(t, asset.Spot) //nolint:govet // Intentional shadow to avoid future copy/paste mistakes
|
|
|
|
out.OrderID = "644913101755"
|
|
got, err := g.WebsocketSpotCancelAllOrdersByIDs(t.Context(), []WebsocketOrderBatchRequest{out})
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, got)
|
|
}
|
|
|
|
func TestWebsocketSpotCancelAllOrdersByPair(t *testing.T) {
|
|
t.Parallel()
|
|
_, err := g.WebsocketSpotCancelAllOrdersByPair(t.Context(), currency.NewPairWithDelimiter("LTC", "USDT", "_"), 0, "")
|
|
require.ErrorIs(t, err, order.ErrSideIsInvalid)
|
|
|
|
sharedtestvalues.SkipTestIfCredentialsUnset(t, g, canManipulateRealOrders)
|
|
|
|
g := newExchangeWithWebsocket(t, asset.Spot) //nolint:govet // Intentional shadow to avoid future copy/paste mistakes
|
|
|
|
got, err := g.WebsocketSpotCancelAllOrdersByPair(t.Context(), currency.EMPTYPAIR, order.Buy, "")
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, got)
|
|
}
|
|
|
|
func TestWebsocketSpotAmendOrder(t *testing.T) {
|
|
t.Parallel()
|
|
_, err := g.WebsocketSpotAmendOrder(t.Context(), nil)
|
|
require.ErrorIs(t, err, common.ErrNilPointer)
|
|
|
|
amend := &WebsocketAmendOrder{}
|
|
_, err = g.WebsocketSpotAmendOrder(t.Context(), amend)
|
|
require.ErrorIs(t, err, order.ErrOrderIDNotSet)
|
|
|
|
amend.OrderID = "1337"
|
|
_, err = g.WebsocketSpotAmendOrder(t.Context(), amend)
|
|
require.ErrorIs(t, err, currency.ErrCurrencyPairEmpty)
|
|
|
|
amend.Pair = BTCUSDT
|
|
|
|
_, err = g.WebsocketSpotAmendOrder(t.Context(), amend)
|
|
require.ErrorIs(t, err, errInvalidAmount)
|
|
|
|
amend.Amount = "0.0004"
|
|
|
|
sharedtestvalues.SkipTestIfCredentialsUnset(t, g, canManipulateRealOrders)
|
|
|
|
g := newExchangeWithWebsocket(t, asset.Spot) //nolint:govet // Intentional shadow to avoid future copy/paste mistakes
|
|
|
|
amend.OrderID = "645029162673"
|
|
got, err := g.WebsocketSpotAmendOrder(t.Context(), amend)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, got)
|
|
}
|
|
|
|
func TestWebsocketSpotGetOrderStatus(t *testing.T) {
|
|
t.Parallel()
|
|
_, err := g.WebsocketSpotGetOrderStatus(t.Context(), "", currency.EMPTYPAIR, "")
|
|
require.ErrorIs(t, err, order.ErrOrderIDNotSet)
|
|
|
|
_, err = g.WebsocketSpotGetOrderStatus(t.Context(), "1337", currency.EMPTYPAIR, "")
|
|
require.ErrorIs(t, err, currency.ErrCurrencyPairEmpty)
|
|
|
|
sharedtestvalues.SkipTestIfCredentialsUnset(t, g, canManipulateRealOrders)
|
|
|
|
testexch.UpdatePairsOnce(t, g)
|
|
g := newExchangeWithWebsocket(t, asset.Spot) //nolint:govet // Intentional shadow to avoid future copy/paste mistakes
|
|
|
|
got, err := g.WebsocketSpotGetOrderStatus(t.Context(), "644999650452", BTCUSDT, "")
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, got)
|
|
}
|
|
|
|
// getWebsocketInstance returns a websocket instance copy for testing.
|
|
// This restricts the pairs to a single pair per asset type to reduce test time.
|
|
func newExchangeWithWebsocket(t *testing.T, a asset.Item) *Gateio {
|
|
t.Helper()
|
|
if apiKey == "" || apiSecret == "" {
|
|
t.Skip()
|
|
}
|
|
g := new(Gateio) //nolint:govet // Intentional shadow to avoid future copy/paste mistakes
|
|
require.NoError(t, testexch.Setup(g), "Test instance Setup must not error")
|
|
testexch.UpdatePairsOnce(t, g)
|
|
g.API.AuthenticatedSupport = true
|
|
g.API.AuthenticatedWebsocketSupport = true
|
|
g.SetCredentials(apiKey, apiSecret, "", "", "", "")
|
|
g.Websocket.SetCanUseAuthenticatedEndpoints(true)
|
|
switch a {
|
|
case asset.Spot:
|
|
avail, err := g.GetAvailablePairs(a)
|
|
require.NoError(t, err)
|
|
if len(avail) > 1 { // reduce pairs to 1 to speed up tests
|
|
avail = avail[:1]
|
|
}
|
|
require.NoError(t, g.SetPairs(avail, a, true))
|
|
case asset.Futures:
|
|
avail, err := g.GetAvailablePairs(a)
|
|
require.NoError(t, err)
|
|
usdtPairs, err := avail.GetPairsByQuote(currency.USDT) // Get USDT margin pairs
|
|
require.NoError(t, err)
|
|
btcPairs, err := avail.GetPairsByQuote(currency.USD) // Get BTC margin pairs
|
|
require.NoError(t, err)
|
|
// below makes sure there is both a USDT and BTC pair available
|
|
// so that allows two connections to be made.
|
|
avail[0] = usdtPairs[0]
|
|
avail[1] = btcPairs[0]
|
|
avail = avail[:2]
|
|
require.NoError(t, g.SetPairs(avail, a, true))
|
|
default:
|
|
require.NoError(t, g.CurrencyPairs.SetAssetEnabled(a, false))
|
|
}
|
|
|
|
require.NoError(t, g.Websocket.Connect())
|
|
return g
|
|
}
|