Use proper model

This commit is contained in:
Jonathan Kim 2020-04-12 15:47:38 +01:00
parent 3483428f70
commit fa405b3bc0

View File

@ -198,10 +198,10 @@ class ModelFormMutationTests(TestCase):
class PetMutation(DjangoModelFormMutation): class PetMutation(DjangoModelFormMutation):
class Meta: class Meta:
form_class = PetForm form_class = PetForm
model = Film model = Pet
return_field_name = "animal" return_field_name = "animal"
self.assertEqual(PetMutation._meta.model, Film) self.assertEqual(PetMutation._meta.model, Pet)
self.assertEqual(PetMutation._meta.return_field_name, "animal") self.assertEqual(PetMutation._meta.return_field_name, "animal")
self.assertIn("animal", PetMutation._meta.fields) self.assertIn("animal", PetMutation._meta.fields)