import_transports_directly_from_gql (#112)

Author: leszekhanusz

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",
+]

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