From 17e9bd3331029d7b09de9f0c91baf6e687279db0 Mon Sep 17 00:00:00 2001 From: Jacob Foster Date: Wed, 19 Jul 2017 14:19:42 -0500 Subject: [PATCH] Fix existing test with ast mock --- graphene_django/tests/test_types.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/graphene_django/tests/test_types.py b/graphene_django/tests/test_types.py index 0ae12c0..1172cc9 100644 --- a/graphene_django/tests/test_types.py +++ b/graphene_django/tests/test_types.py @@ -1,4 +1,4 @@ -from mock import patch +from mock import Mock, patch from graphene import Interface, ObjectType, Schema from graphene.relay import Node @@ -38,7 +38,11 @@ def test_django_interface(): @patch('graphene_django.tests.models.Article.objects.get', return_value=Article(id=1)) def test_django_get_node(get): - article = Article.get_node(1, None, None) + ast_mock = Mock() + ast_mock.selection_set.selections = [] + info_mock = Mock(field_asts=[ast_mock]) + + article = Article.get_node(1, None, info_mock) get.assert_called_with(pk=1) assert article.id == 1