use GetSubscriptions in tests

hasura · Mar 4, 2024 · 39f351b · 39f351b
1 parent cb8e215
commit 39f351b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/subscriptions_transport_ws_test.go b/subscriptions_transport_ws_test.go
@@ -204,7 +204,7 @@ func TestSubscription_closeThenRun(t *testing.T) {
 	bulkSubscribe()
 
 	go func() {
-		length := subscriptionClient.getContext().GetSubscriptionsLength(nil)
+		length := len(subscriptionClient.GetSubscriptions())
 		if length != 2 {
 			t.Errorf("unexpected subscription client. got: %d, want: 2", length)
 			return
@@ -220,7 +220,7 @@ func TestSubscription_closeThenRun(t *testing.T) {
 	}()
 
 	time.Sleep(3 * time.Second)
-	length := subscriptionClient.getContext().GetSubscriptionsLength(nil)
+	length := len(subscriptionClient.GetSubscriptions())
 	if length != 2 {
 		t.Fatalf("unexpected subscription client after restart. got: %d, want: 2, subscriptions: %+v", length, subscriptionClient.context.subscriptions)
 	}