Kraken: Add TestEnforceStandardChannelNames

exchanges/kraken/kraken_test.go

@@ -1706,3 +1706,16 @@ func TestErrorResponse(t *testing.T) {
 		})
 	}
 }
+
+func TestEnforceStandardChannelNames(t *testing.T) {
+	for _, n := range []string{
+		krakenWsSpread, krakenWsTicker, subscription.TickerChannel, subscription.OrderbookChannel, subscription.CandlesChannel,
+		subscription.AllTradesChannel, subscription.MyTradesChannel, subscription.MyOrdersChannel,
+	} {
+		assert.NoError(t, enforceStandardChannelNames(&subscription.Subscription{Channel: n}), "Standard channel names and bespoke names should not error")
+	}
+	for _, n := range []string{krakenWsOrderbook, krakenWsOHLC, krakenWsTrade, krakenWsOwnTrades, krakenWsOpenOrders, krakenWsOrderbook + "-5"} {
+		err := enforceStandardChannelNames(&subscription.Subscription{Channel: n})
+		assert.ErrorIsf(t, err, subscription.ErrPrivateChannelName, "Private channel names should not be allowed for %s", n)
+	}
+}