From ca83173af01cb4f35e71782a481790ab82f5d627 Mon Sep 17 00:00:00 2001 From: Rustam Ganeyev Date: Thu, 31 Dec 2020 11:38:32 +0000 Subject: [PATCH] applied formatting & linting --- .../rest_framework/tests/test_mutation.py | 27 +++++++++---------- 1 file changed, 13 insertions(+), 14 deletions(-) diff --git a/graphene_django/rest_framework/tests/test_mutation.py b/graphene_django/rest_framework/tests/test_mutation.py index 72adb99..c8e19ec 100644 --- a/graphene_django/rest_framework/tests/test_mutation.py +++ b/graphene_django/rest_framework/tests/test_mutation.py @@ -97,25 +97,24 @@ def test_exclude_fields(): assert "cool_name" in MyMutation.Input._meta.fields assert "created" not in MyMutation.Input._meta.fields -def test_model_serializer_required_fields(): - class MyMutation(SerializerMutation): - class Meta: - serializer_class = MyModelSerializer - assert "cool_name" in MyMutation.Input._meta.fields - assert MyMutation.Input._meta.fields['cool_name'].type == NonNull(String) - -''' -The same test as previous, but with 'cool_name' in `optional_fields` -''' -def test_model_serializer_optional_fields(): +def test_model_serializer_required_fields(): class MyMutation(SerializerMutation): class Meta: serializer_class = MyModelSerializer - optional_fields = ('cool_name') - assert "cool_name" in MyMutation.Input._meta.fields - assert MyMutation.Input._meta.fields['cool_name'].type == String + assert "cool_name" in MyMutation.Input._meta.fields + assert MyMutation.Input._meta.fields["cool_name"].type == NonNull(String) + + +def test_model_serializer_optional_fields(): + class MyMutation(SerializerMutation): + class Meta: + serializer_class = MyModelSerializer + optional_fields = "cool_name" + + assert "cool_name" in MyMutation.Input._meta.fields + assert MyMutation.Input._meta.fields["cool_name"].type == String def test_write_only_field():