wip: Can support contains query at this moment

Author: abawchen

graphene_mongo/fields.py

@@ -65,6 +65,7 @@ def args(self, args):
 
     def _field_args(self, items):
         def is_filterable(k):
+            print(k)
             if not hasattr(self.model, k):
                 return False
             if isinstance(getattr(self.model, k), property):
@@ -73,14 +74,15 @@ def is_filterable(k):
                 converted = convert_mongoengine_field(getattr(self.model, k), self.registry)
             except MongoEngineConversionError:
                 return False
-            if isinstance(converted, (ConnectionField, Dynamic, List)):
+            if isinstance(converted, (ConnectionField, Dynamic)):
                 return False
             if callable(getattr(converted, 'type', None)) and isinstance(converted.type(),
                                                                          (PointFieldType, MultiPolygonFieldType)):
                 return False
             return True
 
         def get_type(v):
+            print(v)
             if isinstance(v.type, Structure):
                 return v.type.of_type()
             return v.type()
@@ -128,6 +130,7 @@ def get_queryset(self, model, info, **args):
                 return queryset_or_filters
             else:
                 args.update(queryset_or_filters)
+        print(args)
         return model.objects(**args)
 
     def default_resolver(self, _root, info, **args):

graphene_mongo/tests/models.py

@@ -65,10 +65,10 @@ class Reporter(Document):
     first_name = StringField(required=True)
     last_name = StringField(required=True)
     email = EmailField()
+    awards = ListField(StringField())
     articles = ListField(ReferenceField(Article))
     embedded_articles = ListField(EmbeddedDocumentField(EmbeddedArticle))
     embedded_list_articles = EmbeddedDocumentListField(EmbeddedArticle)
-    awards = ListField(StringField())
 
 
 class Player(Document):

graphene_mongo/tests/test_fields.py

@@ -1,9 +1,9 @@
 from ..fields import MongoengineConnectionField
-from .types import ArticleNode, PublisherNode, ErroneousModelNode
+from .types import ArticleNode, PublisherNode, ReporterNode, ErroneousModelNode
 from .setup import fixtures
 
 
-def test_field_args():
+def test_article_field_args():
     field = MongoengineConnectionField(ArticleNode)
 
     field_args = ['id', 'headline', 'pub_date']
@@ -17,6 +17,14 @@ def test_field_args():
     assert set(field.args) == set(args)
 
 
+def test_reporter_field_args():
+    field = MongoengineConnectionField(ReporterNode)
+
+    field_args = ['id', 'first_name', 'last_name', 'email', 'awards']
+    print(field.field_args['awards'])
+    assert set(field.field_args.keys()) == set(field_args)
+
+
 def test_field_args_with_property():
     field = MongoengineConnectionField(PublisherNode)
 

graphene_mongo/tests/test_relay_query.py

@@ -36,6 +36,7 @@ def resolve_reporter(self, *args, **kwargs):
                 firstName,
                 lastName,
                 email,
+                awards,
                 articles {
                     edges {
                         node {
@@ -65,6 +66,7 @@ def resolve_reporter(self, *args, **kwargs):
             'firstName': 'Allen',
             'lastName': 'Iverson',
             'email': 'ai@gmail.com',
+            'awards': ['2010-mvp'],
             'articles': {
                 'edges': [
                     {
@@ -337,24 +339,62 @@ class Query(graphene.ObjectType):
         expected, sort_keys=True)
 
 
-def test_should_get_node_by_id(fixtures):
+def test_should_filter_by_list_of_string(fixtures):
     # Notes: https://goo.gl/hMNRgs
     class Query(graphene.ObjectType):
-        reporter = Node.Field(ReporterNode)
         reporters = MongoengineConnectionField(ReporterNode)
 
     query = '''
         query ReportersQuery {
+            reporters (awards: "2010") {
+                edges {
+                    node {
+                        id,
+                        firstName,
+                        awards
+                    }
+                }
+            }
+        }
+    '''
+    expected = {
+        'reporters': {
+            'edges': [
+                {
+                    'node': {
+                        'id': 'UmVwb3J0ZXJOb2RlOjE=',
+                        'firstName': 'Allen',
+                        'awards': ['2010-mvp']
+                    }
+                }
+            ]
+        }
+    }
+    schema = graphene.Schema(query=Query)
+    result = schema.execute(query)
+    assert not result.errors
+    assert result.data == expected
+
+
+def test_should_filter_by_id(fixtures):
+    # Notes: https://goo.gl/hMNRgs
+    class Query(graphene.ObjectType):
+        reporter = Node.Field(ReporterNode)
+
+    query = '''
+        query ReporterQuery {
             reporter (id: "UmVwb3J0ZXJOb2RlOjE=") {
                 id,
-                firstName
+                firstName,
+                awards
             }
         }
     '''
     expected = {
         'reporter': {
             'id': 'UmVwb3J0ZXJOb2RlOjE=',
-            'firstName': 'Allen'
+            'firstName': 'Allen',
+            'awards': ['2010-mvp']
         }
     }
     schema = graphene.Schema(query=Query)