mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-20 15:10:10 +00:00
bea16af380
* golangci-lint: Enable usetesting and unused linters * tests: Improve assertions in various test cases for clarity and accuracy * tests: Enhance error assertions in TestExecuteStrategyFromFile for improved clarity * tests: Update assertions for improved clarity and accuracy * tests: Replace assert with require for task count checks * config/versions/v7: Replace context.Background() with t.Context() * Bithumb: Centralise and consoliate testPair, relax UpdateTickers check with some glorious Doom Eternal music * Bithumb: Use UpdatePairsOnce and update remaining pair string * Bithumb: Add UpdatePairsOnce to TestUpdateTickers
83 lines
2.6 KiB
Go
83 lines
2.6 KiB
Go
package binance
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/request"
|
|
)
|
|
|
|
func TestRateLimit_Limit(t *testing.T) {
|
|
t.Parallel()
|
|
symbol := "BTC-USDT"
|
|
|
|
testTable := map[string]struct {
|
|
Expected request.EndpointLimit
|
|
Limit request.EndpointLimit
|
|
Deadline time.Time
|
|
}{
|
|
"Open Orders": {Expected: spotOpenOrdersSpecificRate, Limit: openOrdersLimit(symbol)},
|
|
"Orderbook Depth 5": {Expected: spotOrderbookDepth100Rate, Limit: orderbookLimit(5)},
|
|
"Orderbook Depth 10": {Expected: spotOrderbookDepth100Rate, Limit: orderbookLimit(10)},
|
|
"Orderbook Depth 20": {Expected: spotOrderbookDepth100Rate, Limit: orderbookLimit(20)},
|
|
"Orderbook Depth 50": {Expected: spotOrderbookDepth100Rate, Limit: orderbookLimit(50)},
|
|
"Orderbook Depth 100": {Expected: spotOrderbookDepth100Rate, Limit: orderbookLimit(100)},
|
|
"Orderbook Depth 500": {Expected: spotOrderbookDepth500Rate, Limit: orderbookLimit(500)},
|
|
"Orderbook Depth 1000": {Expected: spotOrderbookDepth1000Rate, Limit: orderbookLimit(1000)},
|
|
"Orderbook Depth 5000": {Expected: spotOrderbookDepth5000Rate, Limit: orderbookLimit(5000)},
|
|
"Exceeds deadline": {Expected: spotOrderbookDepth5000Rate, Limit: orderbookLimit(5000), Deadline: time.Now().Add(time.Nanosecond)},
|
|
}
|
|
|
|
rl, err := request.New("rateLimitTest", &http.Client{}, request.WithLimiter(GetRateLimits()))
|
|
require.NoError(t, err)
|
|
|
|
for name, tt := range testTable {
|
|
t.Run(name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
exp, got := tt.Expected, tt.Limit
|
|
if exp != got {
|
|
t.Fatalf("incorrect limit applied.\nexp: %v\ngot: %v", exp, got)
|
|
}
|
|
|
|
ctx := t.Context()
|
|
if !tt.Deadline.IsZero() {
|
|
var cancel context.CancelFunc
|
|
ctx, cancel = context.WithDeadline(ctx, tt.Deadline)
|
|
defer cancel()
|
|
}
|
|
|
|
if err := rl.InitiateRateLimit(ctx, tt.Limit); err != nil && !errors.Is(err, context.DeadlineExceeded) {
|
|
t.Fatalf("error applying rate limit: %v", err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestRateLimit_LimitStatic(t *testing.T) {
|
|
t.Parallel()
|
|
testTable := map[string]request.EndpointLimit{
|
|
"Default": spotDefaultRate,
|
|
"Historical Trades": spotHistoricalTradesRate,
|
|
"All Price Changes": spotTickerAllRate,
|
|
"All Orders": spotAllOrdersRate,
|
|
}
|
|
|
|
rl, err := request.New("rateLimitTest2", http.DefaultClient, request.WithLimiter(GetRateLimits()))
|
|
require.NoError(t, err)
|
|
|
|
for name, tt := range testTable {
|
|
t.Run(name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
if err := rl.InitiateRateLimit(t.Context(), tt); err != nil {
|
|
t.Fatalf("error applying rate limit: %v", err)
|
|
}
|
|
})
|
|
}
|
|
}
|