graphql unit test

rsocket · Oct 8, 2023 · fc9660f · fc9660f
1 parent ca476c6
commit fc9660f
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/tests/test_integrations/test_graphql.py b/tests/test_integrations/test_graphql.py
@@ -51,3 +51,28 @@ def handler_factory():
         assert len(responses) == 10
         assert responses[0] == {'greetings': {'message': 'Hello world 0'}}
         assert responses[9] == {'greetings': {'message': 'Hello world 9'}}
+
+
+async def test_graphql_get_schema(lazy_pipe, graphql_schema):
+    expected_schema = {'__schema': {
+        'types': [{'name': 'Query'}, {'name': 'String'}, {'name': 'Subscription'}, {'name': 'Greeting'},
+                  {'name': 'Mutation'}, {'name': 'Boolean'}, {'name': '__Schema'}, {'name': '__Type'},
+                  {'name': '__TypeKind'}, {'name': '__Field'}, {'name': '__InputValue'}, {'name': '__EnumValue'},
+                  {'name': '__Directive'}, {'name': '__DirectiveLocation'}]}}
+
+    def handler_factory():
+        return RoutingRequestHandler(graphql_handler(graphql_schema, 'graphql'))
+
+    async with lazy_pipe(
+            client_arguments={'metadata_encoding': WellKnownMimeTypes.MESSAGE_RSOCKET_COMPOSITE_METADATA},
+            server_arguments={'handler_factory': handler_factory}) as (server, client):
+        graphql = Client(
+            schema=graphql_schema,
+            transport=RSocketTransport(client),
+        )
+
+        response = await graphql.execute_async(
+            gql("""{__schema { types { name  }  } }"""),
+            get_execution_result=True)
+
+        assert response.data == expected_schema