Fix existing test with ast mock

This commit is contained in:
Jacob Foster 2017-07-19 14:19:42 -05:00
parent 2d2c5ee889
commit 17e9bd3331

View File

@ -1,4 +1,4 @@
from mock import patch from mock import Mock, patch
from graphene import Interface, ObjectType, Schema from graphene import Interface, ObjectType, Schema
from graphene.relay import Node 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)) @patch('graphene_django.tests.models.Article.objects.get', return_value=Article(id=1))
def test_django_get_node(get): 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) get.assert_called_with(pk=1)
assert article.id == 1 assert article.id == 1