Kraken: Fix test instance locally scoped

gbjk · Feb 12, 2024 · 5bbffe3 · 5bbffe3
1 parent 924edb3
commit 5bbffe3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/exchanges/kraken/kraken_test.go b/exchanges/kraken/kraken_test.go
@@ -50,7 +50,7 @@ var btcusdPair = currency.NewPairWithDelimiter("XBT", "USD", "/")
 
 func TestMain(m *testing.M) {
 	var err error
-	k := new(Kraken)
+	k = new(Kraken)
 	if err = testexch.TestInstance(k); err != nil {
 		log.Fatal(err)
 	}
@@ -1186,7 +1186,6 @@ func TestWsSubscribe(t *testing.T) {
 
 	k := new(Kraken)
 	require.NoError(t, testexch.TestInstance(k), "TestInstance must not error")
-	k.Features.Subscriptions = []*subscription.Subscription{}
 	testexch.SetupWs(t, k)
 
 	err := k.Subscribe([]subscription.Subscription{{Channel: krakenWsTicker, Pairs: currency.Pairs{currency.NewPairWithDelimiter("XBT", "USD", "/")}}})

diff --git a/internal/testing/exchange/exchange.go b/internal/testing/exchange/exchange.go
@@ -160,6 +160,9 @@ func SetupWs(tb testing.TB, e exchange.IBotExchange) {
 		return
 	}
 
+	// Disable any default subscriptions, which aren't relevant for tests
+	b.Features.Subscriptions = []*subscription.Subscription{}
+
 	err = w.Connect()
 	require.NoError(tb, err, "WsConnect should not error")