From 20e505c3abcfa4991bb62d8da96ec2d3b5e4f37b Mon Sep 17 00:00:00 2001
From: Rustam Ganeyev <rganeyev@gmail.com>
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):