fixup! Kucoin: Add Subscription batching

gbjk · Jul 12, 2024 · 4fe7234 · 4fe7234
1 parent b0133ff
commit 4fe7234
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/exchanges/subscription/fixtures_test.go b/exchanges/subscription/fixtures_test.go
@@ -29,7 +29,10 @@ func (m *mockEx) GetPairFormat(_ asset.Item, _ bool) (currency.PairFormat, error
 	return currency.PairFormat{Uppercase: true}, m.errFormat
 }
 
-func (m *mockEx) GetSubscriptionTemplate(_ *Subscription) (*template.Template, error) {
+func (m *mockEx) GetSubscriptionTemplate(s *Subscription) (*template.Template, error) {
+	if s.Channel == "nil" {
+		return nil, nil
+	}
 	return template.New(m.tpl).
 		Funcs(template.FuncMap{
 			"assetName": func(a asset.Item) string {

diff --git a/exchanges/subscription/template_test.go b/exchanges/subscription/template_test.go
@@ -113,4 +113,7 @@ func TestExpandTemplates(t *testing.T) {
 	require.Len(t, got, 1, "Must get back the one sub")
 	assert.Equal(t, "already happy", l[0].QualifiedChannel, "Should get back the one sub")
 	assert.NotSame(t, got, l, "Should get back a different actual list")
+
+	_, err = List{{Channel: "nil"}}.ExpandTemplates(e)
+	assert.ErrorIs(t, err, errInvalidTemplate, "Should get correct error on nil template")
 }