Subscriptions: Add List.Authenticated

thrasher-corp · Aug 19, 2024 · 1a5c59e · 1a5c59e
1 parent 17c2ef2
commit 1a5c59e
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/exchanges/subscription/list.go b/exchanges/subscription/list.go
@@ -55,6 +55,17 @@ func (l List) Clone() List {
 	return n
 }
 
+// Authenticated returns only Authenticated subscriptions
+func (l List) Authenticated() List {
+	a := List{}
+	for _, s := range l {
+		if s.Authenticated {
+			a = append(a, s)
+		}
+	}
+	return a
+}
+
 // QualifiedChannels returns a sorted list of all the qualified Channels in the list
 func (l List) QualifiedChannels() []string {
 	c := make([]string, len(l))

diff --git a/exchanges/subscription/list_test.go b/exchanges/subscription/list_test.go
@@ -106,3 +106,9 @@ func TestListClone(t *testing.T) {
 	l[0].Interval = kline.OneHour
 	assert.NotEqual(t, n[0], l[0], "Subscriptions should be cloned")
 }
+
+func TestListAuthenticated(t *testing.T) {
+	t.Parallel()
+	assert.NotEmpty(t, 1, List{{}, {Authenticated: true}}.Authenticated(), "Authenticated should return correct length")
+	assert.Empty(t, List{{}, {}}.Authenticated(), "Authenticated should return empty")
+}