test(itam): Add missing unit tests for software installs viewset

ref: #512 #513 #518
This commit is contained in:
2025-02-06 20:03:49 +09:30
parent d608a3d138
commit c54c54d821

View File

@ -0,0 +1,75 @@
import pytest
from django.contrib.auth.models import User
from django.test import Client, TestCase
from rest_framework.reverse import reverse
from access.models import Organization
from api.tests.abstract.viewsets import ViewSetModel
from itam.models.software import Software
from itam.viewsets.device_software import ViewSet
class ViewsetCommon(
ViewSetModel,
):
viewset = ViewSet
route_name = 'v2:_api_v2_software_installs'
@classmethod
def setUpTestData(self):
"""Setup Test
1. Create an organization
3. create super user
"""
organization = Organization.objects.create(name='test_org')
self.organization = organization
self.view_user = User.objects.create_user(username="test_view_user", password="password", is_superuser=True)
self.kwargs = {
'software_id': Software.objects.create(
organization = organization,
name = 'software'
).id
}
class SoftwareInstallsViewsetList(
ViewsetCommon,
TestCase,
):
@classmethod
def setUpTestData(self):
"""Setup Test
1. make list request
"""
super().setUpTestData()
client = Client()
url = reverse(
self.route_name + '-list',
kwargs = self.kwargs
)
client.force_login(self.view_user)
self.http_options_response_list = client.options(url)