Better way to support Python 3.10 for ensure_future (#277)

Author: leszekhanusz

gql/client.py

@@ -220,11 +220,9 @@ def subscribe(
                 # Note: we need to create a task here in order to be able to close
                 # the async generator properly on python 3.8
                 # See https://bugs.python.org/issue38559
-                with warnings.catch_warnings():
-                    warnings.filterwarnings(
-                        "ignore", message="There is no current event loop"
-                    )
-                    generator_task = asyncio.ensure_future(async_generator.__anext__())
+                generator_task = asyncio.ensure_future(
+                    async_generator.__anext__(), loop=loop
+                )
                 result = loop.run_until_complete(generator_task)
                 yield result
 

tests/test_websocket_subscription.py

@@ -521,6 +521,8 @@ def test_websocket_subscription_sync_graceful_shutdown(server, subscription_str)
     count = 10
     subscription = gql(subscription_str.format(count=count))
 
+    interrupt_task = None
+
     with pytest.raises(KeyboardInterrupt):
         for result in client.subscribe(subscription):
 
@@ -536,14 +538,17 @@ def test_websocket_subscription_sync_graceful_shutdown(server, subscription_str)
                     warnings.filterwarnings(
                         "ignore", message="There is no current event loop"
                     )
-                    asyncio.ensure_future(
+                    interrupt_task = asyncio.ensure_future(
                         client.session._generator.athrow(KeyboardInterrupt)
                     )
 
             count -= 1
 
     assert count == 4
 
+    # Catch interrupt_task exception to remove warning
+    interrupt_task.exception()
+
     # Check that the server received a connection_terminate message last
     assert logged_messages.pop() == '{"type": "connection_terminate"}'