added additional tests for CustomSQLAlchemyObjectType

Author: dfee

graphene_sqlalchemy/tests/test_types.py

@@ -1,9 +1,10 @@
 
 from graphene import Field, Int, Interface, ObjectType
 from graphene.relay import Node, is_node
+import six
 
 from ..registry import Registry
-from ..types import SQLAlchemyObjectType
+from ..types import SQLAlchemyObjectType, SQLAlchemyObjectTypeMeta
 from .models import Article, Reporter
 
 registry = Registry()
@@ -74,3 +75,46 @@ def test_object_type():
     assert issubclass(Human, ObjectType)
     assert list(Human._meta.fields.keys()) == ['id', 'headline', 'reporter_id', 'reporter', 'pub_date']
     assert is_node(Human)
+
+
+
+# Test Custom SQLAlchemyObjectType Implementation
+class CustomSQLAlchemyObjectType(six.with_metaclass(SQLAlchemyObjectTypeMeta, ObjectType)):
+    @classmethod
+    def is_type_of(cls, root, context, info):
+        return SQLAlchemyObjectType.is_type_of(root, context, info)
+
+    @classmethod
+    def get_query(cls, context):
+        return SQLAlchemyObjectType.get_query(context)
+
+    @classmethod
+    def get_node(cls, id, context, info):
+        return SQLAlchemyObjectType.get_node(id, context, info)
+
+    @classmethod
+    def resolve_id(self, args, context, info):
+        graphene_type = info.parent_type.graphene_type
+        if is_node(graphene_type):
+            return self.__mapper__.primary_key_from_instance(self)[0]
+        return getattr(self, graphene_type._meta.id, None)
+
+
+class CustomCharacter(CustomSQLAlchemyObjectType):
+    '''Character description'''
+    class Meta:
+        model = Reporter
+        registry = registry
+
+def test_custom_objecttype_registered():
+    assert issubclass(CustomCharacter, ObjectType)
+    assert CustomCharacter._meta.model == Reporter
+    assert list(
+        CustomCharacter._meta.fields.keys()) == [
+        'id',
+        'first_name',
+        'last_name',
+        'email',
+        'pets',
+        'articles',
+        'favorite_article']