Add connection

Author: jnak

graphene_sqlalchemy/tests/test_types.py

@@ -4,6 +4,7 @@
 
 from graphene import (Dynamic, Field, GlobalID, Int, List, Node, NonNull,
                       ObjectType, Schema, String)
+from graphene.relay import Connection
 
 from ..converter import convert_sqlalchemy_composite
 from ..fields import (SQLAlchemyConnectionField,
@@ -46,6 +47,15 @@ class Meta:
     assert reporter == reporter_node
 
 
+def test_connection():
+    class ReporterType(SQLAlchemyObjectType):
+        class Meta:
+            model = Reporter
+            interfaces = (Node,)
+
+    assert issubclass(ReporterType.connection, Connection)
+
+
 def test_sqlalchemy_default_fields():
     @convert_sqlalchemy_composite.register(CompositeFullName)
     def convert_composite_class(composite, registry):

graphene_sqlalchemy/types.py

@@ -325,6 +325,8 @@ def __init_subclass_with_meta__(
         _meta.connection = connection
         _meta.id = id or "id"
 
+        cls.connection = connection  # Public way to get the connection
+
         super(SQLAlchemyObjectType, cls).__init_subclass_with_meta__(
             _meta=_meta, interfaces=interfaces, **options
         )