test infra

Author: HaraldNordgren

internal/integration/generated.go

@@ -244,7 +244,7 @@ func TestSubscriptionConnectionParams(t *testing.T) {
 
 func TestSubscriptionClose(t *testing.T) {
 	_ = `# @genqlient
-	subscription count { count }`
+	subscription countClose { countClose }`
 
 	ctx := context.Background()
 	server := server.RunServer()
@@ -271,7 +271,7 @@ func TestSubscriptionClose(t *testing.T) {
 			_, err := wsClient.Start(ctx)
 			require.NoError(t, err)
 
-			_, subscriptionID, err := count(ctx, wsClient)
+			_, subscriptionID, err := countClose(ctx, wsClient)
 			require.NoError(t, err)
 
 			if tc.unsub {

internal/integration/integration_test.go

@@ -20,6 +20,7 @@ type Mutation {
 type Subscription {
   count: Int!
   countAuthorized: Int!
+  countClose: Int!
 }
 
 type User implements Being & Lucky {

internal/integration/schema.graphql

@@ -182,6 +182,10 @@ func (s *subscriptionResolver) CountAuthorized(ctx context.Context) (<-chan int,
 	return s.Count(ctx)
 }
 
+func (s *subscriptionResolver) CountClose(ctx context.Context) (<-chan int, error) {
+	return s.Count(ctx)
+}
+
 const AuthKey = "authToken"
 
 type (