chore(basae): Add capability to Functional Model Test Suite create fixture to cater for many to many field

ref: #904
This commit is contained in:
2025-07-29 14:36:28 +09:30
parent 3248d5201d
commit cc481f94e9

View File

@ -1,5 +1,7 @@
import pytest
from django.db import models
@pytest.mark.models
@ -23,25 +25,66 @@ class ModelTestCases:
"""
@pytest.fixture( scope = 'function' )
def created_model(self, django_db_blocker, model, model_kwargs):
@pytest.fixture( scope = 'function')
def created_model(self, request, django_db_blocker,
model, model_kwargs, mocker, model_user, kwargs_user
):
model_object = None
if not model._meta.abstract:
with django_db_blocker.unblock():
model_object = model.objects.create(
**model_kwargs
)
yield model_object
item = None
with django_db_blocker.unblock():
if model_object:
model_object.delete()
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
})
context_user = mocker.patch.object(
model, 'context'
)
user = model_user.objects.create( **kwargs_user )
context_user.__getitem__.side_effect = {
'logger': None,
'user': user
}.__getitem__
item = model.objects.create(
**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
with django_db_blocker.unblock():
item.delete()
user.delete()