78 lines
4.2 KiB
Python
78 lines
4.2 KiB
Python
from django.test import TestCase
|
|
|
|
from dcim.models import Device, DeviceRole, DeviceType, Manufacturer, Site
|
|
from virtualization.models import Cluster, ClusterType, VirtualMachine
|
|
from netbox_slm.models import SoftwareProduct, SoftwareProductVersion, SoftwareProductInstallation, SoftwareLicense
|
|
|
|
|
|
class ModelTestCase(TestCase):
|
|
def setUp(self):
|
|
self.p_name = "test product"
|
|
self.v_name = "test version"
|
|
self.l_name = "test license"
|
|
|
|
manufacturer = Manufacturer.objects.create(name="test manufacturer")
|
|
device_type = DeviceType.objects.create(model="test device type", manufacturer=manufacturer)
|
|
device_role = DeviceRole.objects.create(name="test device role")
|
|
site = Site.objects.create(name="test site")
|
|
self.device = Device.objects.create(name="test device", device_type=device_type, role=device_role, site=site)
|
|
self.vm = VirtualMachine.objects.create(name="test VM")
|
|
cluster_type = ClusterType.objects.create(name="test cluster type")
|
|
self.cluster = Cluster.objects.create(name="test cluster", type=cluster_type)
|
|
self.test_url = "https://github.com/ICTU/netbox_slm"
|
|
|
|
self.software_product = SoftwareProduct.objects.create(name=self.p_name)
|
|
self.software_product_version = SoftwareProductVersion.objects.create(
|
|
name=self.v_name, software_product=self.software_product
|
|
)
|
|
self.software_product_installation = SoftwareProductInstallation.objects.create(
|
|
virtualmachine=self.vm, software_product=self.software_product, version=self.software_product_version
|
|
)
|
|
self.software_license = SoftwareLicense.objects.create(
|
|
name=self.l_name,
|
|
software_product=self.software_product,
|
|
version=self.software_product_version,
|
|
installation=self.software_product_installation,
|
|
stored_location_url=self.test_url,
|
|
)
|
|
|
|
def test_model_name(self):
|
|
self.assertEqual(self.p_name, str(self.software_product))
|
|
self.assertEqual(self.v_name, str(self.software_product_version))
|
|
self.assertTrue(str(self.software_product_installation)[0].isdigit()) # starts with PK
|
|
self.assertEqual(self.l_name, str(self.software_license))
|
|
|
|
def test_absolute_url(self):
|
|
self.assertEqual("/plugins/slm/software-products/1/", self.software_product.get_absolute_url())
|
|
self.assertEqual("/plugins/slm/versions/1/", self.software_product_version.get_absolute_url())
|
|
self.assertEqual("/plugins/slm/installations/1/", self.software_product_installation.get_absolute_url())
|
|
self.assertEqual("/plugins/slm/licenses/1/", self.software_license.get_absolute_url())
|
|
|
|
def test_get_installation_count(self):
|
|
installation_ss = '<a href="/plugins/slm/installations/?q={}">1</a>'
|
|
self.assertEqual(installation_ss.format(self.p_name), self.software_product.get_installation_count())
|
|
self.assertEqual(installation_ss.format(self.v_name), self.software_product_version.get_installation_count())
|
|
|
|
def test_product_installation_methods(self):
|
|
self.assertEqual("virtualmachine", self.software_product_installation.render_type())
|
|
self.assertEqual(self.vm, self.software_product_installation.platform)
|
|
|
|
self.software_product_installation.virtualmachine = None
|
|
self.software_product_installation.device = self.device
|
|
self.software_product_installation.save()
|
|
self.assertEqual("device", self.software_product_installation.render_type())
|
|
self.assertEqual(self.device, self.software_product_installation.platform)
|
|
|
|
self.software_product_installation.device = None
|
|
self.software_product_installation.cluster = self.cluster
|
|
self.software_product_installation.save()
|
|
self.assertEqual("cluster", self.software_product_installation.render_type())
|
|
self.assertEqual(self.cluster, self.software_product_installation.platform)
|
|
|
|
def test_software_license_stored_location_txt(self):
|
|
self.assertEqual("Link", self.software_license.stored_location_txt)
|
|
|
|
self.software_license.stored_location = "GitHub"
|
|
self.software_license.save()
|
|
self.assertEqual("GitHub", self.software_license.stored_location_txt)
|