test: Increase coverage of MongoengineConnectionField.get_query.

graphql-python · Mar 19, 2018 · 2a322e5 · 2a322e5
1 parent 41978ae
commit 2a322e5
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 2 deletions.
diff --git a/examples/flask_mongoengine/app.py b/examples/flask_mongoengine/app.py
@@ -32,7 +32,6 @@
         tasks {
           edges {
             node {
-              id,
               name,
               deadline
             }

diff --git a/graphene_mongo/tests/models.py b/graphene_mongo/tests/models.py
@@ -65,3 +65,5 @@ class Player(Document):
     opponent = ReferenceField('Player')
     players = ListField(ReferenceField('Player'))
     articles = ListField(ReferenceField('Article'))
+    embedded_list_articles = EmbeddedDocumentListField(EmbeddedArticle)
+
diff --git a/graphene_mongo/tests/test_relay_query.py b/graphene_mongo/tests/test_relay_query.py
@@ -382,6 +382,13 @@ class Query(graphene.ObjectType):
                                     firstName
                                 }
                             }
+                        },
+                        embeddedListArticles {
+                            edges {
+                                node {
+                                    headline
+                                }
+                            }
                         }
                     }
                 }
@@ -402,6 +409,9 @@ class Query(graphene.ObjectType):
                                     }
                                 }
                             ]
+                        },
+                        'embeddedListArticles': {
+                            'edges': []
                         }
                     }
                 },
@@ -416,7 +426,11 @@ class Query(graphene.ObjectType):
                                     }
                                 }
                             ]
+                        },
+                        'embeddedListArticles': {
+                             'edges': []
                         }
+
                     }
                 },
                 {
@@ -435,6 +449,9 @@ class Query(graphene.ObjectType):
                                     }
                                 }
                             ]
+                        },
+                        'embeddedListArticles': {
+                            'edges': []
                         }
                     }
                 }
@@ -446,7 +463,6 @@ class Query(graphene.ObjectType):
     assert not result.errors
     assert json.dumps(result.data, sort_keys=True) == json.dumps(expected, sort_keys=True)
 
-
 # TODO:
 def test_should_paging():
     pass