Adding new test to have 100% code coverage now that aiohttp transport supports subscriptions

Author: leszekhanusz

tests/test_httpx_async.py

@@ -1447,3 +1447,62 @@ async def handler(request):
         pi = result["pi"]
 
         assert pi == Decimal("3.141592653589793238462643383279502884197")
+
+
+@pytest.mark.aiohttp
+@pytest.mark.asyncio
+async def test_httpx_subscribe_not_supported_cli(aiohttp_server):
+    """Test that the CLI falls back to execute when subscribe is not supported."""
+    from aiohttp import web
+
+    from gql.transport.httpx import HTTPXAsyncTransport
+
+    async def handler(request):
+        return web.Response(text=query1_server_answer, content_type="application/json")
+
+    app = web.Application()
+    app.router.add_route("POST", "/", handler)
+    server = await aiohttp_server(app)
+
+    url = str(server.make_url("/"))
+
+    transport = HTTPXAsyncTransport(url=url)
+
+    async with Client(transport=transport) as _:
+
+        # Define arguments for the CLI
+        # We use the query "query getContinents..."
+        import io
+        import sys
+        from io import StringIO
+
+        from gql import cli
+
+        test_args = ["gql-cli", url, "--transport", "httpx"]
+
+        # Mock sys.stdin to provide the query
+        sys.stdin = io.StringIO(query1_str)
+
+        # Capture stdout
+        captured_output = StringIO()
+        original_stdout = sys.stdout
+        sys.stdout = captured_output
+
+        try:
+            # We need to mock sys.argv as well because cli.get_parser() uses
+            # usage from sys.argv[0] sometimes,
+            # but mainly passing args to parse_args is cleaner.
+            parser = cli.get_parser()
+            parsed_args = parser.parse_args(test_args[1:])  # skip prog name
+
+            exit_code = await cli.main(parsed_args)
+            assert exit_code == 0
+
+        except SystemExit:
+            pass
+        finally:
+            sys.stdout = original_stdout
+            sys.stdin = sys.__stdin__  # Restore stdin
+
+        output = captured_output.getvalue()
+        assert "Africa" in output