mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 23:16:45 +00:00
c601575c66
* Subscriptions: Add List.AssetPairs * Subscriptions: Add Template and QualifiedChannel These fields separate the concept of what the channel is from the qualified resource name * Subscriptions: Add List.SetStates() * Subscriptions: Add List.QualifiedChannels * Subscriptions: Rename testsubs.EqualLists * Binance: Switch to ExpandTemplates * Binance: Update ConfigTest format * Subscriptions: Test Coverage improvements * Subscriptions: Reenterant List.ExpandTemplates * Subscriptions: Move templates from subscriptions to exchanges * Binance: Inline subscription template and improvements
94 lines
3.1 KiB
Go
94 lines
3.1 KiB
Go
package subscription
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/thrasher-corp/gocryptotrader/currency"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/asset"
|
|
)
|
|
|
|
// TestListStrings exercises List.Strings()
|
|
func TestListStrings(t *testing.T) {
|
|
t.Parallel()
|
|
l := List{
|
|
&Subscription{
|
|
Channel: TickerChannel,
|
|
Asset: asset.Spot,
|
|
Pairs: currency.Pairs{ethusdcPair, btcusdtPair},
|
|
},
|
|
&Subscription{
|
|
Channel: OrderbookChannel,
|
|
Pairs: currency.Pairs{ethusdcPair},
|
|
},
|
|
}
|
|
exp := []string{"orderbook ETH/USDC", "ticker spot ETH/USDC,BTC/USDT"}
|
|
assert.ElementsMatch(t, exp, l.Strings(), "String must return correct sorted list")
|
|
}
|
|
|
|
// TestQualifiedChannels exercises List.QualifiedChannels()
|
|
func TestQualifiedChannels(t *testing.T) {
|
|
t.Parallel()
|
|
l := List{
|
|
&Subscription{
|
|
QualifiedChannel: "ticker-btc",
|
|
},
|
|
&Subscription{
|
|
QualifiedChannel: "candles-btc",
|
|
},
|
|
}
|
|
exp := []string{"ticker-btc", "candles-btc"}
|
|
assert.ElementsMatch(t, exp, l.QualifiedChannels(), "QualifiedChannels should return correct sorted list")
|
|
}
|
|
|
|
// TestListGroupPairs exercises List.GroupPairs()
|
|
func TestListGroupPairs(t *testing.T) {
|
|
t.Parallel()
|
|
l := List{
|
|
{Asset: asset.Spot, Channel: TickerChannel, Pairs: currency.Pairs{ethusdcPair, btcusdtPair}},
|
|
}
|
|
for _, c := range []string{TickerChannel, OrderbookChannel} {
|
|
for _, p := range []currency.Pair{ethusdcPair, btcusdtPair} {
|
|
l = append(l, &Subscription{
|
|
Channel: c,
|
|
Asset: asset.Spot,
|
|
Pairs: currency.Pairs{p},
|
|
})
|
|
}
|
|
}
|
|
n := l.GroupPairs()
|
|
assert.Len(t, l, 5, "Orig list should not be changed")
|
|
assert.Len(t, n, 2, "New list should be grouped")
|
|
exp := []string{"ticker spot ETH/USDC,BTC/USDT", "orderbook spot ETH/USDC,BTC/USDT"}
|
|
assert.ElementsMatch(t, exp, n.Strings(), "String must return correct sorted list")
|
|
}
|
|
|
|
// TestListSetStates exercises List.SetState()
|
|
func TestListSetStates(t *testing.T) {
|
|
t.Parallel()
|
|
l := List{{Channel: TickerChannel}, {Channel: OrderbookChannel}}
|
|
assert.NoError(t, l.SetStates(SubscribingState), "SetStates should not error")
|
|
assert.Equal(t, SubscribingState, l[1].State(), "SetStates should set State correctly")
|
|
|
|
require.NoError(t, l[0].SetState(SubscribedState), "Individual SetState must not error")
|
|
err := l.SetStates(SubscribedState)
|
|
assert.ErrorIs(t, ErrInStateAlready, err, "SetStates should error when duplicate state")
|
|
assert.Equal(t, SubscribedState, l[1].State(), "SetStates should set State correctly after the error")
|
|
}
|
|
|
|
// TestAssetPairs exercises AssetPairs error handling
|
|
// All other code is covered under TestExpandTemplates
|
|
func TestAssetPairs(t *testing.T) {
|
|
t.Parallel()
|
|
expErr := errors.New("Krypton is gone")
|
|
for _, a := range []asset.Item{asset.Spot, asset.All} {
|
|
l := &List{{Channel: CandlesChannel, Asset: a}}
|
|
_, err := l.assetPairs(&mockEx{errPairs: expErr})
|
|
assert.ErrorIs(t, err, expErr, "Should error correctly on GetEnabledPairs")
|
|
_, err = l.assetPairs(&mockEx{errFormat: expErr})
|
|
assert.ErrorIs(t, err, expErr, "Should error correctly on GetPairFormat")
|
|
}
|
|
}
|