mirror of
https://github.com/graphql-python/graphene-django.git
synced 2024-11-15 06:08:59 +03:00
136 lines
3.7 KiB
Python
136 lines
3.7 KiB
Python
import datetime
|
|
|
|
from graphene import Field
|
|
from graphene.types.inputobjecttype import InputObjectType
|
|
from py.test import raises
|
|
from py.test import mark
|
|
from rest_framework import serializers
|
|
|
|
from ...types import DjangoObjectType
|
|
from ..models import MyFakeModel
|
|
from ..mutation import SerializerMutation
|
|
|
|
|
|
class MyModelSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = MyFakeModel
|
|
fields = '__all__'
|
|
|
|
|
|
class MySerializer(serializers.Serializer):
|
|
text = serializers.CharField()
|
|
model = MyModelSerializer()
|
|
|
|
def create(self, validated_data):
|
|
return validated_data
|
|
|
|
|
|
def test_needs_serializer_class():
|
|
with raises(Exception) as exc:
|
|
class MyMutation(SerializerMutation):
|
|
pass
|
|
|
|
assert str(exc.value) == 'serializer_class is required for the SerializerMutation'
|
|
|
|
|
|
def test_has_fields():
|
|
class MyMutation(SerializerMutation):
|
|
class Meta:
|
|
serializer_class = MySerializer
|
|
|
|
assert 'text' in MyMutation._meta.fields
|
|
assert 'model' in MyMutation._meta.fields
|
|
assert 'errors' in MyMutation._meta.fields
|
|
|
|
|
|
def test_has_input_fields():
|
|
class MyMutation(SerializerMutation):
|
|
class Meta:
|
|
serializer_class = MySerializer
|
|
|
|
assert 'text' in MyMutation.Input._meta.fields
|
|
assert 'model' in MyMutation.Input._meta.fields
|
|
|
|
|
|
def test_exclude_fields():
|
|
class MyMutation(SerializerMutation):
|
|
class Meta:
|
|
serializer_class = MyModelSerializer
|
|
exclude_fields = ['created']
|
|
|
|
assert 'cool_name' in MyMutation._meta.fields
|
|
assert 'created' not in MyMutation._meta.fields
|
|
assert 'errors' in MyMutation._meta.fields
|
|
assert 'cool_name' in MyMutation.Input._meta.fields
|
|
assert 'created' not in MyMutation.Input._meta.fields
|
|
|
|
|
|
def test_nested_model():
|
|
|
|
class MyFakeModelGrapheneType(DjangoObjectType):
|
|
class Meta:
|
|
model = MyFakeModel
|
|
|
|
class MyMutation(SerializerMutation):
|
|
class Meta:
|
|
serializer_class = MySerializer
|
|
|
|
model_field = MyMutation._meta.fields['model']
|
|
assert isinstance(model_field, Field)
|
|
assert model_field.type == MyFakeModelGrapheneType
|
|
|
|
model_input = MyMutation.Input._meta.fields['model']
|
|
model_input_type = model_input._type.of_type
|
|
assert issubclass(model_input_type, InputObjectType)
|
|
assert 'cool_name' in model_input_type._meta.fields
|
|
assert 'created' in model_input_type._meta.fields
|
|
|
|
|
|
def test_mutate_and_get_payload_success():
|
|
|
|
class MyMutation(SerializerMutation):
|
|
class Meta:
|
|
serializer_class = MySerializer
|
|
|
|
result = MyMutation.mutate_and_get_payload(None, None, **{
|
|
'text': 'value',
|
|
'model': {
|
|
'cool_name': 'other_value'
|
|
}
|
|
})
|
|
assert result.errors is None
|
|
|
|
|
|
@mark.django_db
|
|
def test_model_mutate_and_get_payload_success():
|
|
class MyMutation(SerializerMutation):
|
|
class Meta:
|
|
serializer_class = MyModelSerializer
|
|
|
|
result = MyMutation.mutate_and_get_payload(None, None, **{
|
|
'cool_name': 'Narf',
|
|
})
|
|
assert result.errors is None
|
|
assert result.cool_name == 'Narf'
|
|
assert isinstance(result.created, datetime.datetime)
|
|
|
|
def test_mutate_and_get_payload_error():
|
|
|
|
class MyMutation(SerializerMutation):
|
|
class Meta:
|
|
serializer_class = MySerializer
|
|
|
|
# missing required fields
|
|
result = MyMutation.mutate_and_get_payload(None, None, **{})
|
|
assert len(result.errors) > 0
|
|
|
|
def test_model_mutate_and_get_payload_error():
|
|
|
|
class MyMutation(SerializerMutation):
|
|
class Meta:
|
|
serializer_class = MyModelSerializer
|
|
|
|
# missing required fields
|
|
result = MyMutation.mutate_and_get_payload(None, None, **{})
|
|
assert len(result.errors) > 0
|