Skip to content

Commit 6bd75f5

Browse files
committed
feat: Take care empty EmbeddedListField.
1 parent 8665d61 commit 6bd75f5

File tree

2 files changed

+9
-5
lines changed

2 files changed

+9
-5
lines changed

graphene_mongo/fields.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,10 @@ def fields(self):
8686

8787
@classmethod
8888
def get_query(cls, model, info, **args):
89+
90+
if not callable(getattr(model, 'objects', None)):
91+
return []
92+
8993
objs = model.objects()
9094

9195
if args:

graphene_mongo/tests/fixtures.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,9 @@ def setup_fixtures():
1717
article2.save()
1818

1919
Reporter.drop_collection()
20-
reporter = Reporter(id='1', first_name='Allen', last_name='Iverson',
20+
reporter1 = Reporter(id='1', first_name='Allen', last_name='Iverson',
2121
email='[email protected]', awards=['2010-mvp'])
22-
reporter.articles = [article1, article2]
22+
reporter1.articles = [article1, article2]
2323
embedded_article1 = EmbeddedArticle(
2424
headline='Real',
2525
editor=editor1
@@ -28,9 +28,9 @@ def setup_fixtures():
2828
headline='World',
2929
editor=editor2
3030
)
31-
reporter.embedded_articles = [embedded_article1, embedded_article2]
32-
reporter.embedded_list_articles = [embedded_article2, embedded_article1]
33-
reporter.save()
31+
reporter1.embedded_articles = [embedded_article1, embedded_article2]
32+
reporter1.embedded_list_articles = [embedded_article2, embedded_article1]
33+
reporter1.save()
3434

3535
Player.drop_collection()
3636
player1 = Player(first_name='Michael', last_name='Jordan')

0 commit comments

Comments
 (0)