feat(core): Add supprt to model create test for manytomany field
ref: #783
This commit is contained in:
@ -245,10 +245,44 @@ class CenturionAbstractModelInheritedCases(
|
||||
|
||||
model.context['user'] = user
|
||||
|
||||
kwargs = {}
|
||||
|
||||
many_field = {}
|
||||
|
||||
for field, value in self.kwargs_create_item.items():
|
||||
|
||||
if isinstance(getattr(model, field).field, models.ManyToManyField):
|
||||
|
||||
if field in many_field:
|
||||
|
||||
many_field[field] += [ value ]
|
||||
|
||||
else:
|
||||
|
||||
many_field.update({
|
||||
field: [
|
||||
value
|
||||
]
|
||||
})
|
||||
|
||||
continue
|
||||
|
||||
kwargs.update({
|
||||
field: value
|
||||
})
|
||||
|
||||
|
||||
model_object = model.objects.create(
|
||||
**self.kwargs_create_item
|
||||
**kwargs
|
||||
)
|
||||
|
||||
for field, values in many_field.items():
|
||||
|
||||
for value in values:
|
||||
|
||||
getattr(model_object, field).add( value )
|
||||
|
||||
|
||||
model.context['user'] = default_val
|
||||
|
||||
assert type(model_object.id) is int
|
||||
|
Reference in New Issue
Block a user