chore(api): Add capability to API Fields model create fixture to cater for many to many field

ref: #904
This commit is contained in:
2025-07-29 13:42:58 +09:30
parent 09c1acbb1f
commit 83ebf12978

View File

@ -92,10 +92,39 @@ class APIFieldsTestCases:
with django_db_blocker.unblock():
kwargs_many_to_many = {}
kwargs = {}
for key, value in model_kwargs.items():
field = model._meta.get_field(key)
if isinstance(field, models.ManyToManyField):
kwargs_many_to_many.update({
key: value
})
else:
kwargs.update({
key: value
})
item = model.objects.create(
**model_kwargs
**kwargs
)
for key, value in kwargs_many_to_many.items():
field = getattr(item, 'target_team')
for entry in value:
field.add(entry)
request.cls.item = item
yield item
@ -198,7 +227,7 @@ class APIFieldsTestCases:
else:
assert(
type( api_data['value'] ) is param_expected
type( api_data.get('value', 'is empty') ) is param_expected
or type( api_data_two.get('value', 'is empty') ) is param_expected
)