added tests

This commit is contained in:
Rustam Ganeyev 2020-12-31 11:35:29 +00:00
parent 4453f7a34d
commit 20e505c3ab

View File

@ -3,7 +3,7 @@ import datetime
from py.test import raises from py.test import raises
from rest_framework import serializers from rest_framework import serializers
from graphene import Field, ResolveInfo from graphene import Field, ResolveInfo, NonNull, String
from graphene.types.inputobjecttype import InputObjectType from graphene.types.inputobjecttype import InputObjectType
from ...types import DjangoObjectType from ...types import DjangoObjectType
@ -97,6 +97,26 @@ def test_exclude_fields():
assert "cool_name" in MyMutation.Input._meta.fields assert "cool_name" in MyMutation.Input._meta.fields
assert "created" not 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(): def test_write_only_field():
class WriteOnlyFieldModelSerializer(serializers.ModelSerializer): class WriteOnlyFieldModelSerializer(serializers.ModelSerializer):