refactor(itam): Serializer Unit Test Suite re-written to Pytest for model DeviceType

ref: #920 #919
This commit is contained in:
2025-07-31 13:25:50 +09:30
parent 9b3340f44c
commit 5e1ebc949e
6 changed files with 89 additions and 64 deletions

View File

@ -1,63 +0,0 @@
import pytest
from django.test import TestCase
from rest_framework.exceptions import ValidationError
from access.models.tenant import Tenant as Organization
from centurion.tests.abstract.mock_view import MockView, User
from itam.serializers.device_type import DeviceType, DeviceTypeModelSerializer
class DeviceTypeValidationAPI(
TestCase,
):
model = DeviceType
@classmethod
def setUpTestData(self):
"""Setup Test
1. Create an org
2. Create an item
"""
organization = Organization.objects.create(name='test_org')
self.user = User.objects.create_user(username="test_user_view", password="password")
self.mock_view = MockView( user = self.user )
self.organization = organization
self.item = self.model.objects.create(
organization=organization,
name = 'valid-hostname',
)
def test_serializer_validation_no_name(self):
"""Serializer Validation Check
Ensure that if creating and no name is provided a validation error occurs
"""
with pytest.raises(ValidationError) as err:
serializer = DeviceTypeModelSerializer(
context = {
'request': self.mock_view.request,
'view': self.mock_view,
},
data={
"organization": self.organization.id,
})
serializer.is_valid(raise_exception = True)
assert err.value.get_codes()['name'][0] == 'required'

View File

@ -17,3 +17,9 @@ def model_kwargs(request, kwargs_devicetype):
if hasattr(request.cls, 'kwargs_create_item'):
del request.cls.kwargs_create_item
@pytest.fixture( scope = 'class')
def model_serializer(serializer_devicetype):
yield serializer_devicetype

View File

@ -0,0 +1,63 @@
import pytest
from rest_framework.exceptions import ValidationError
from api.tests.unit.test_unit_serializer import (
SerializerTestCases
)
from centurion.tests.abstract.mock_view import MockView
@pytest.mark.model_devicetype
class DeviceTypeSerializerTestCases(
SerializerTestCases
):
def test_serializer_validation_no_name(self,
kwargs_api_create, model, model_serializer, request_user,
):
"""Serializer Validation Check
Ensure that if creating and no name is provided a validation error occurs
"""
mock_view = MockView(
user = request_user,
model = model,
action = 'create',
)
kwargs = kwargs_api_create.copy()
del kwargs['name']
with pytest.raises(ValidationError) as err:
serializer = model_serializer['model'](
context = {
'request': mock_view.request,
'view': mock_view,
},
data = kwargs,
)
serializer.is_valid(raise_exception = True)
assert err.value.get_codes()['name'][0] == 'required'
class DeviceTypeSerializerInheritedCases(
DeviceTypeSerializerTestCases
):
pass
@pytest.mark.module_itam
class DeviceTypeSerializerPyTest(
DeviceTypeSerializerTestCases
):
pass

View File

@ -125,6 +125,7 @@ from .model_devicesoftware import (
from .model_devicetype import (
kwargs_devicetype,
model_devicetype,
serializer_devicetype,
)
from .model_employee import (

View File

@ -1,5 +1,6 @@
import datetime
import pytest
import random
from itam.models.device import DeviceModel
from itam.serializers.device_model import (
@ -25,7 +26,9 @@ def kwargs_devicemodel(kwargs_centurionmodel, django_db_blocker,
with django_db_blocker.unblock():
manufacturer = model_manufacturer.objects.create( **kwargs_manufacturer )
kwargs = kwargs_manufacturer.copy()
kwargs['name'] = 'dm_' + str( random.randint(1, 99999) )
manufacturer = model_manufacturer.objects.create( **kwargs )
kwargs = {
**kwargs_centurionmodel.copy(),

View File

@ -2,6 +2,11 @@ import datetime
import pytest
from itam.models.device import DeviceType
from itam.serializers.device_type import (
DeviceTypeBaseSerializer,
DeviceTypeModelSerializer,
DeviceTypeViewSerializer,
)
@ -23,3 +28,13 @@ def kwargs_devicetype(kwargs_centurionmodel):
}
yield kwargs.copy()
@pytest.fixture( scope = 'class')
def serializer_devicetype():
yield {
'base': DeviceTypeBaseSerializer,
'model': DeviceTypeModelSerializer,
'view': DeviceTypeViewSerializer
}