@@ -36,6 +36,12 @@ def clean_age(self):
36
36
return age
37
37
38
38
39
+ class FilmDetailsForm (forms .ModelForm ):
40
+ class Meta :
41
+ model = FilmDetails
42
+ fields = "__all__"
43
+
44
+
39
45
class PetType (DjangoObjectType ):
40
46
class Meta :
41
47
model = Pet
@@ -192,22 +198,22 @@ class Meta:
192
198
self .assertNotIn ("id" , PetMutation .Input ._meta .fields )
193
199
194
200
def test_return_field_name_is_camelcased (self ):
195
- class PetMutation (DjangoModelFormMutation ):
201
+ class FilmDetailsMutation (DjangoModelFormMutation ):
196
202
class Meta :
197
- form_class = PetForm
203
+ form_class = FilmDetailsForm
198
204
model = FilmDetails
199
205
200
- self .assertEqual (PetMutation ._meta .model , FilmDetails )
201
- self .assertEqual (PetMutation ._meta .return_field_name , "filmDetails" )
206
+ self .assertEqual (FilmDetailsMutation ._meta .model , FilmDetails )
207
+ self .assertEqual (FilmDetailsMutation ._meta .return_field_name , "filmDetails" )
202
208
203
209
def test_custom_return_field_name (self ):
204
210
class PetMutation (DjangoModelFormMutation ):
205
211
class Meta :
206
212
form_class = PetForm
207
- model = Film
213
+ model = Pet
208
214
return_field_name = "animal"
209
215
210
- self .assertEqual (PetMutation ._meta .model , Film )
216
+ self .assertEqual (PetMutation ._meta .model , Pet )
211
217
self .assertEqual (PetMutation ._meta .return_field_name , "animal" )
212
218
self .assertIn ("animal" , PetMutation ._meta .fields )
213
219
0 commit comments