From 20e505c3abcfa4991bb62d8da96ec2d3b5e4f37b Mon Sep 17 00:00:00 2001 From: Rustam Ganeyev Date: Thu, 31 Dec 2020 11:35:29 +0000 Subject: [PATCH] added tests --- .../rest_framework/tests/test_mutation.py | 22 ++++++++++++++++++- 1 file changed, 21 insertions(+), 1 deletion(-) diff --git a/graphene_django/rest_framework/tests/test_mutation.py b/graphene_django/rest_framework/tests/test_mutation.py index ffbc4b5..72adb99 100644 --- a/graphene_django/rest_framework/tests/test_mutation.py +++ b/graphene_django/rest_framework/tests/test_mutation.py @@ -3,7 +3,7 @@ import datetime from py.test import raises from rest_framework import serializers -from graphene import Field, ResolveInfo +from graphene import Field, ResolveInfo, NonNull, String from graphene.types.inputobjecttype import InputObjectType from ...types import DjangoObjectType @@ -97,6 +97,26 @@ 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(): + 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(): class WriteOnlyFieldModelSerializer(serializers.ModelSerializer):