Test mutation

Author: Grant McConnaughey

graphene_django/forms/tests/test_mutation.py

@@ -1,4 +1,5 @@
 from django import forms
+from django.test import TestCase
 from py.test import raises
 
 from graphene_django.tests.models import Pet
@@ -40,10 +41,23 @@ class Meta:
     assert 'text' in MyMutation.Input._meta.fields
 
 
-def test_model_form():
-    class PetMutation(ModelFormMutation):
-        class Meta:
-            form_class = PetForm
+class ModelFormMutationTests(TestCase):
+
+    def test_model_form_mutation(self):
+        class PetMutation(ModelFormMutation):
+            class Meta:
+                form_class = PetForm
+
+        self.assertEqual(PetMutation.model, Pet)
+        self.assertEqual(PetMutation.return_field_name, 'pet')
+
+    def test_model_form_mutation_mutate(self):
+        class PetMutation(ModelFormMutation):
+            class Meta:
+                form_class = PetForm
+
+        PetMutation.mutate(None, {'input': {'name': 'Fluffy'}}, None, None)
 
-    assert PetMutation.model == Pet
-    assert PetMutation.return_field_name == 'pet'
+        self.assertEqual(Pet.objects.count(), 1)
+        pet = Pet.objects.get()
+        self.assertEqual(pet.name, 'Fluffy')