chore(access): Update common viewset test suite mock request

ref: #847 #840 #841 #842 #843 #845
This commit is contained in:
2025-06-21 15:33:02 +09:30
parent 1ba7661f07
commit 7b22541002
3 changed files with 50 additions and 7 deletions

View File

@ -93,6 +93,22 @@ class EntityModelTestCases(
assert model.kb_model_name == 'entity'
def test_method_get_url_kwargs(self, mocker, model_instance, settings):
"""Test Class Method
Ensure method `get_url_kwargs` returns the correct value.
"""
url = model_instance.get_url_kwargs()
assert model_instance.get_url_kwargs() == {
'model_name': model_instance._meta.model_name,
'pk': model_instance.id
}
class EntityModelInheritedCases(
EntityModelTestCases,
@ -113,3 +129,16 @@ class EntityModelPyTest(
"""
assert model_instance.get_related_model() is None
def test_method_get_url_kwargs(self, mocker, model_instance, settings):
"""Test Class Method
Ensure method `get_url_kwargs` returns the correct value.
"""
url = model_instance.get_url_kwargs()
assert model_instance.get_url_kwargs() == {
'pk': model_instance.id
}

View File

@ -4,7 +4,7 @@ from django.test import Client, TestCase
from rest_framework.reverse import reverse
from access.models.entity import Entity
from access.viewsets.entity import (
NoDocsViewSet,
ViewSet,
@ -21,6 +21,8 @@ class ViewsetTestCases(
kwargs = None
model = Entity
viewset = None
route_name = None

View File

@ -69,7 +69,7 @@ class MockRequest:
user: User = None
def __init__(self, user: User, organization: Organization, viewset):
def __init__(self, user: User, organization: Organization, viewset, model = None):
self.user = user
@ -77,11 +77,15 @@ class MockRequest:
viewset = viewset()
if model is None:
model = viewset.model
view_permission = Permission.objects.get(
codename = 'view_' + viewset.model._meta.model_name,
codename = 'view_' + model._meta.model_name,
content_type = ContentType.objects.get(
app_label = viewset.model._meta.app_label,
model = viewset.model._meta.model_name,
app_label = model._meta.app_label,
model = model._meta.model_name,
)
)
@ -927,6 +931,12 @@ class ModelViewSetBaseCases(
"""
view_set = self.viewset()
view_set.request = MockRequest(
user = self.view_user,
model = getattr(self, 'model',None),
viewset = self.viewset,
organization = self.organization
)
assert view_set.model is not None
@ -982,8 +992,9 @@ class ModelViewSetBaseCases(
view_set.request = MockRequest(
user = self.view_user,
model = getattr(self, 'model',None),
organization = self.organization,
viewset = self.viewset
viewset = self.viewset,
)
view_set.request.headers = {}
@ -1014,8 +1025,9 @@ class ModelViewSetBaseCases(
view_set.request = MockRequest(
user = self.view_user,
model = getattr(self, 'model',None),
organization = self.organization,
viewset = self.viewset
viewset = self.viewset,
)
view_set.request.headers = {}