Bugfix

- allow SQLALchemyConnectionField.connection for all SQLAlchemyBase types

Author: gbunkoczi

graphene_sqlalchemy/fields.py

@@ -26,14 +26,14 @@
 class SQLAlchemyConnectionField(ConnectionField):
     @property
     def type(self):
-        from .types import SQLAlchemyObjectType
+        from .types import SQLAlchemyBase
 
         type_ = super(ConnectionField, self).type
         nullable_type = get_nullable_type(type_)
         if issubclass(nullable_type, Connection):
             return type_
-        assert issubclass(nullable_type, SQLAlchemyObjectType), (
-            "SQLALchemyConnectionField only accepts SQLAlchemyObjectType types, not {}"
+        assert issubclass(nullable_type, SQLAlchemyBase), (
+            "SQLALchemyConnectionField only accepts SQLAlchemyBase types, not {}"
         ).format(nullable_type.__name__)
         assert nullable_type.connection, "The type {} doesn't have a connection".format(
             nullable_type.__name__

graphene_sqlalchemy/tests/test_fields.py

@@ -51,7 +51,7 @@ def resolver(_obj, _info):
 
 
 def test_type_assert_sqlalchemy_object_type():
-    with pytest.raises(AssertionError, match="only accepts SQLAlchemyObjectType"):
+    with pytest.raises(AssertionError, match="only accepts SQLAlchemyBase types"):
         SQLAlchemyConnectionField(ObjectType).type