From bfe72a3203f5f242bb13d9bdbae9ef91592536da Mon Sep 17 00:00:00 2001 From: Hanusz Leszek Date: Sun, 12 Jul 2020 12:13:35 +0200 Subject: [PATCH] import_transports_directly_from_gql --- gql/__init__.py | 11 ++++++++++- tests/test_transport.py | 8 ++++++++ 2 files changed, 18 insertions(+), 1 deletion(-) diff --git a/gql/__init__.py b/gql/__init__.py index 571c7371..7c21c1c8 100644 --- a/gql/__init__.py +++ b/gql/__init__.py @@ -1,4 +1,13 @@ from .client import Client from .gql import gql +from .transport.aiohttp import AIOHTTPTransport +from .transport.requests import RequestsHTTPTransport +from .transport.websockets import WebsocketsTransport -__all__ = ["gql", "Client"] +__all__ = [ + "gql", + "AIOHTTPTransport", + "Client", + "RequestsHTTPTransport", + "WebsocketsTransport", +] diff --git a/tests/test_transport.py b/tests/test_transport.py index b7048249..4e40d595 100644 --- a/tests/test_transport.py +++ b/tests/test_transport.py @@ -119,3 +119,11 @@ def test_named_query(client): with use_cassette("queries"): result = client.execute(query, operation_name="Planet2") assert result == expected + + +def test_import_transports_directly_from_gql(): + from gql import AIOHTTPTransport, RequestsHTTPTransport, WebsocketsTransport + + assert AIOHTTPTransport + assert RequestsHTTPTransport + assert WebsocketsTransport