Test for subscriptions with dynamic GraphQL clients

integration-tests/smallrye-graphql-client/src/main/java/io/quarkus/io/smallrye/graphql/client/LuckyNumbersResource.java

@@ -6,6 +6,9 @@
 import org.eclipse.microprofile.graphql.Mutation;
 import org.eclipse.microprofile.graphql.Query;
 
+import io.smallrye.graphql.api.Subscription;
+import io.smallrye.mutiny.Multi;
+
 @GraphQLApi
 @ApplicationScoped
 public class LuckyNumbersResource {
@@ -23,4 +26,9 @@ public Integer setLuckyNumber(Integer newLuckyNumber) {
         return luckyNumber;
     }
 
+    @Subscription
+    public Multi<Integer> primeNumbers() {
+        return Multi.createFrom().items(2, 3, 5, 7, 11, 13);
+    }
+
 }

integration-tests/smallrye-graphql-client/src/test/java/io/quarkus/it/smallrye/graphql/client/DynamicClientTest.java

@@ -6,8 +6,11 @@
 import static io.smallrye.graphql.client.core.Field.field;
 import static io.smallrye.graphql.client.core.Operation.operation;
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertFalse;
 
 import java.net.URL;
+import java.util.List;
+import java.util.stream.Collectors;
 
 import org.junit.jupiter.api.Test;
 
@@ -44,4 +47,23 @@ public void testDynamicClient() throws Exception {
             assertEquals(15, response.getData().getInt("get"), response.toString());
         }
     }
+
+    @Test
+    public void testDynamicClientSubscription() throws Exception {
+        try (DynamicGraphQLClient client = DynamicGraphQLClientBuilder.newBuilder().url(url.toString() + "/graphql").build()) {
+            Document op = document(
+                    operation(OperationType.SUBSCRIPTION,
+                            field("primeNumbers")));
+            List<Integer> expectedNumbers = List.of(2, 3, 5, 7, 11, 13);
+            List<Response> responses = client.subscription(op)
+                    .subscribe()
+                    .asStream()
+                    .collect(Collectors.toList());
+            for (int i = 0; i < expectedNumbers.size(); i++) {
+                assertEquals(expectedNumbers.get(i), responses.get(i).getData().getInt("primeNumbers"));
+                assertFalse(responses.get(i).hasError());
+            }
+        }
+    }
+
 }

integration-tests/smallrye-graphql-client/src/test/java/io/quarkus/it/smallrye/graphql/client/TypesafeClientTest.java

@@ -1,6 +1,5 @@
 package io.quarkus.it.smallrye.graphql.client;
 
-import static org.junit.jupiter.api.Assertions.*;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 
 import java.net.URL;
@@ -20,7 +19,6 @@ public class TypesafeClientTest {
 
     @Test
     public void testTypesafeClient() throws Exception {
-
         LuckyNumbersClientApi client = TypesafeGraphQLClientBuilder.newBuilder()
                 .endpoint(url.toString() + "/graphql")
                 .build(LuckyNumbersClientApi.class);