mirror of
https://github.com/graphql-python/graphene-django.git
synced 2024-09-21 19:39:09 +03:00
76 lines
2.0 KiB
Python
76 lines
2.0 KiB
Python
from django import forms
|
|
from django.test import TestCase
|
|
from py.test import raises
|
|
|
|
from graphene_django.tests.models import Pet, Film
|
|
from ..mutation import FormMutation, ModelFormMutation
|
|
|
|
|
|
class MyForm(forms.Form):
|
|
text = forms.CharField()
|
|
|
|
|
|
class PetForm(forms.ModelForm):
|
|
|
|
class Meta:
|
|
model = Pet
|
|
fields = ('name',)
|
|
|
|
|
|
def test_needs_form_class():
|
|
with raises(Exception) as exc:
|
|
class MyMutation(FormMutation):
|
|
pass
|
|
|
|
assert exc.value.args[0] == 'form_class is required for FormMutation'
|
|
|
|
|
|
def test_has_output_fields():
|
|
class MyMutation(FormMutation):
|
|
class Meta:
|
|
form_class = MyForm
|
|
|
|
assert 'errors' in MyMutation._meta.fields
|
|
|
|
|
|
def test_has_input_fields():
|
|
class MyMutation(FormMutation):
|
|
class Meta:
|
|
form_class = MyForm
|
|
|
|
assert 'text' in MyMutation.Input._meta.fields
|
|
|
|
|
|
class ModelFormMutationTests(TestCase):
|
|
|
|
def test_default_meta_fields(self):
|
|
class PetMutation(ModelFormMutation):
|
|
class Meta:
|
|
form_class = PetForm
|
|
|
|
self.assertEqual(PetMutation._meta.model, Pet)
|
|
self.assertEqual(PetMutation._meta.return_field_name, 'pet')
|
|
self.assertIn('pet', PetMutation._meta.fields)
|
|
|
|
def test_custom_return_field_name(self):
|
|
class PetMutation(ModelFormMutation):
|
|
class Meta:
|
|
form_class = PetForm
|
|
model = Film
|
|
return_field_name = 'animal'
|
|
|
|
self.assertEqual(PetMutation._meta.model, Film)
|
|
self.assertEqual(PetMutation._meta.return_field_name, 'animal')
|
|
self.assertIn('animal', PetMutation._meta.fields)
|
|
|
|
def test_model_form_mutation_mutate(self):
|
|
class PetMutation(ModelFormMutation):
|
|
class Meta:
|
|
form_class = PetForm
|
|
|
|
PetMutation.mutate_and_get_payload(None, None, name='Fluffy')
|
|
|
|
self.assertEqual(Pet.objects.count(), 1)
|
|
pet = Pet.objects.get()
|
|
self.assertEqual(pet.name, 'Fluffy')
|