diff --git a/graphene_django/fields.py b/graphene_django/fields.py index 6383258..35df1e1 100644 --- a/graphene_django/fields.py +++ b/graphene_django/fields.py @@ -1,4 +1,4 @@ -from functools import partial +from functools import partial, reduce from django.db.models.query import QuerySet from graphene.types import Field, List diff --git a/graphene_django/forms/tests/test_mutation.py b/graphene_django/forms/tests/test_mutation.py index 329a7c1..5d28259 100644 --- a/graphene_django/forms/tests/test_mutation.py +++ b/graphene_django/forms/tests/test_mutation.py @@ -133,15 +133,14 @@ class ModelFormMutationTests(TestCase): # A pet was not created self.assertEqual(Pet.objects.count(), 0) - - fields_w_error = [e.field for e in result.errors] + fields_w_error = {e.field: e.messages for e in result.errors} self.assertEqual(len(result.errors), 3) self.assertIn("testCamel", fields_w_error) - self.assertEqual(result.errors[0].messages, ["Enter a whole number."]) + self.assertEqual(fields_w_error['testCamel'], ["Enter a whole number."]) self.assertIn("name", fields_w_error) - self.assertEqual(result.errors[1].messages, ["This field is required."]) + self.assertEqual(fields_w_error['name'], ["This field is required."]) self.assertIn("age", fields_w_error) - self.assertEqual(result.errors[2].messages, ["This field is required."]) + self.assertEqual(fields_w_error['age'], ["This field is required."]) class FormMutationTests(TestCase):