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