chore(access): Update common viewset test suite mock request
ref: #847 #840 #841 #842 #843 #845
This commit is contained in:
@ -93,6 +93,22 @@ class EntityModelTestCases(
|
|||||||
assert model.kb_model_name == 'entity'
|
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(
|
class EntityModelInheritedCases(
|
||||||
EntityModelTestCases,
|
EntityModelTestCases,
|
||||||
@ -113,3 +129,16 @@ class EntityModelPyTest(
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
assert model_instance.get_related_model() is None
|
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
|
||||||
|
}
|
||||||
|
@ -4,7 +4,7 @@ from django.test import Client, TestCase
|
|||||||
|
|
||||||
from rest_framework.reverse import reverse
|
from rest_framework.reverse import reverse
|
||||||
|
|
||||||
|
from access.models.entity import Entity
|
||||||
from access.viewsets.entity import (
|
from access.viewsets.entity import (
|
||||||
NoDocsViewSet,
|
NoDocsViewSet,
|
||||||
ViewSet,
|
ViewSet,
|
||||||
@ -21,6 +21,8 @@ class ViewsetTestCases(
|
|||||||
|
|
||||||
kwargs = None
|
kwargs = None
|
||||||
|
|
||||||
|
model = Entity
|
||||||
|
|
||||||
viewset = None
|
viewset = None
|
||||||
|
|
||||||
route_name = None
|
route_name = None
|
||||||
|
@ -69,7 +69,7 @@ class MockRequest:
|
|||||||
|
|
||||||
user: User = None
|
user: User = None
|
||||||
|
|
||||||
def __init__(self, user: User, organization: Organization, viewset):
|
def __init__(self, user: User, organization: Organization, viewset, model = None):
|
||||||
|
|
||||||
self.user = user
|
self.user = user
|
||||||
|
|
||||||
@ -77,11 +77,15 @@ class MockRequest:
|
|||||||
|
|
||||||
viewset = viewset()
|
viewset = viewset()
|
||||||
|
|
||||||
|
if model is None:
|
||||||
|
|
||||||
|
model = viewset.model
|
||||||
|
|
||||||
view_permission = Permission.objects.get(
|
view_permission = Permission.objects.get(
|
||||||
codename = 'view_' + viewset.model._meta.model_name,
|
codename = 'view_' + model._meta.model_name,
|
||||||
content_type = ContentType.objects.get(
|
content_type = ContentType.objects.get(
|
||||||
app_label = viewset.model._meta.app_label,
|
app_label = model._meta.app_label,
|
||||||
model = viewset.model._meta.model_name,
|
model = model._meta.model_name,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -927,6 +931,12 @@ class ModelViewSetBaseCases(
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
view_set = self.viewset()
|
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
|
assert view_set.model is not None
|
||||||
|
|
||||||
@ -982,8 +992,9 @@ class ModelViewSetBaseCases(
|
|||||||
|
|
||||||
view_set.request = MockRequest(
|
view_set.request = MockRequest(
|
||||||
user = self.view_user,
|
user = self.view_user,
|
||||||
|
model = getattr(self, 'model',None),
|
||||||
organization = self.organization,
|
organization = self.organization,
|
||||||
viewset = self.viewset
|
viewset = self.viewset,
|
||||||
)
|
)
|
||||||
|
|
||||||
view_set.request.headers = {}
|
view_set.request.headers = {}
|
||||||
@ -1014,8 +1025,9 @@ class ModelViewSetBaseCases(
|
|||||||
|
|
||||||
view_set.request = MockRequest(
|
view_set.request = MockRequest(
|
||||||
user = self.view_user,
|
user = self.view_user,
|
||||||
|
model = getattr(self, 'model',None),
|
||||||
organization = self.organization,
|
organization = self.organization,
|
||||||
viewset = self.viewset
|
viewset = self.viewset,
|
||||||
)
|
)
|
||||||
|
|
||||||
view_set.request.headers = {}
|
view_set.request.headers = {}
|
||||||
|
Reference in New Issue
Block a user