test: Add test_should_filter_by_reference_field

Author: abawchen

graphene_mongo/fields.py

@@ -11,6 +11,9 @@
 from graphene.types.argument import to_arguments
 
 
+from .utils import get_model_reference_fields
+
+
 # noqa
 class MongoengineListField(Field):
 
@@ -60,7 +63,8 @@ def model(self):
     @property
     def args(self):
         return to_arguments(
-            self._base_args or OrderedDict(), dict(self.field_args.items() + self.reference_args.items())
+            self._base_args or OrderedDict(),
+            dict(self.field_args.items() + self.reference_args.items())
         )
 
     @args.setter
@@ -103,8 +107,15 @@ def get_query(cls, model, info, **args):
             return []
 
         objs = model.objects()
-
         if args:
+            reference_fields = get_model_reference_fields(model)
+            for arg_name, arg in args.items():
+                if arg_name in reference_fields:
+                    reference_model = model._fields[arg_name]
+                    pk = from_global_id(args.pop(arg_name))[-1]
+                    reference_obj = reference_model.document_type_obj.objects(pk=pk).get()
+                    args[arg_name] = reference_obj
+
             first = args.pop('first', None)
             last = args.pop('last', None)
             id = args.pop('id', None)
@@ -132,6 +143,7 @@ def get_query(cls, model, info, **args):
                 # https://github.com/graphql-python/graphene-mongo/issues/20
                 objs = objs[-(last+1):]
 
+        print(objs)
         return objs
 
     # noqa

graphene_mongo/tests/test_relay_query.py

@@ -295,6 +295,46 @@ class Query(graphene.ObjectType):
     assert result.data == expected
 
 
+def test_should_filter_by_reference_field():
+
+    class Query(graphene.ObjectType):
+        node = Node.Field()
+        articles = MongoengineConnectionField(ArticleNode)
+
+    query = '''
+        query ArticlesQuery {
+            articles(editor: "RWRpdG9yTm9kZTox") {
+                edges {
+                    node {
+                        headline,
+                        editor {
+                            firstName
+                        }
+                    }
+                }
+            }
+        }
+    '''
+    expected = {
+        'articles': {
+            'edges': [
+                {
+                    'node': {
+                        'headline': 'Hello',
+                        'editor': {
+                            'firstName': 'Penny'
+                        }
+                    }
+                }
+            ]
+        }
+    }
+    schema = graphene.Schema(query=Query)
+    result = schema.execute(query)
+    assert not result.errors
+    assert result.data == expected
+
+
 def test_should_filter_through_inheritance():
 
     class Query(graphene.ObjectType):

graphene_mongo/utils.py

@@ -5,8 +5,7 @@
 
 
 def get_model_fields(model, excluding=None):
-    if excluding is None:
-        excluding = []
+    excluding = excluding or []
     attributes = dict()
     for attr_name, attr in model._fields.items():
         if attr_name in excluding:
@@ -15,6 +14,17 @@ def get_model_fields(model, excluding=None):
     return OrderedDict(sorted(attributes.items()))
 
 
+def get_model_reference_fields(model, excluding=None):
+    excluding = excluding or []
+    attributes = dict()
+    for attr_name, attr in model._fields.items():
+        if attr_name in excluding \
+                or not isinstance(attr, mongoengine.fields.ReferenceField):
+            continue
+        attributes[attr_name] = attr
+    return attributes
+
+
 def is_valid_mongoengine_model(model):
     return inspect.isclass(model) and (
         issubclass(model, mongoengine.Document) or issubclass(model, mongoengine.EmbeddedDocument)