MrBlake
01d40ddecc
Add tenant group and tenant assignments across the model, forms, filters, API, tables, imports, and tests. Replace the Docker-based setup with Git/Pip installation documentation for NetBox 4.6.5.
67 lines
3.1 KiB
Python
67 lines
3.1 KiB
Python
from django.test import TestCase
|
|
|
|
from dcim.models import Device, DeviceRole, DeviceType, Manufacturer, Site
|
|
from netbox_slm.models import SoftwareProduct, SoftwareProductVersion, SoftwareProductInstallation, SoftwareLicense
|
|
from tenancy.models import Tenant, TenantGroup
|
|
from virtualization.models import Cluster, ClusterType, VirtualMachine
|
|
|
|
|
|
class SlmBaseTestCase(TestCase):
|
|
p_name: str = "test product"
|
|
v_name: str = "test version"
|
|
l_name: str = "test license"
|
|
test_url: str = "https://github.com/ICTU/netbox_slm"
|
|
vm: VirtualMachine
|
|
software_product: SoftwareProduct
|
|
software_product_version: SoftwareProductVersion
|
|
software_product_installation: SoftwareProductInstallation
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super().setUpClass()
|
|
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")
|
|
cls.device = Device.objects.create(name="test device", device_type=device_type, role=device_role, site=site)
|
|
cls.vm = VirtualMachine.objects.create(name="test VM")
|
|
cluster_type = ClusterType.objects.create(name="test cluster type")
|
|
cls.cluster = Cluster.objects.create(name="test cluster", type=cluster_type)
|
|
cls.tenant_group = TenantGroup.objects.create(name="test tenant group", slug="test-tenant-group")
|
|
cls.tenant = Tenant.objects.create(name="test tenant", slug="test-tenant", group=cls.tenant_group)
|
|
|
|
cls.software_product = SoftwareProduct.objects.create(name=cls.p_name)
|
|
cls.software_product_version = SoftwareProductVersion.objects.create(
|
|
name=cls.v_name, software_product=cls.software_product
|
|
)
|
|
cls.software_product_installation = SoftwareProductInstallation.objects.create(
|
|
virtualmachine=cls.vm, software_product=cls.software_product, version=cls.software_product_version
|
|
)
|
|
cls.software_license = SoftwareLicense.objects.create(
|
|
name=cls.l_name,
|
|
software_product=cls.software_product,
|
|
version=cls.software_product_version,
|
|
installation=cls.software_product_installation,
|
|
stored_location_url=cls.test_url,
|
|
tenant_group=cls.tenant_group,
|
|
tenant=cls.tenant,
|
|
)
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
SoftwareLicense.objects.all().delete()
|
|
SoftwareProductInstallation.objects.all().delete()
|
|
SoftwareProductVersion.objects.all().delete()
|
|
SoftwareProduct.objects.all().delete()
|
|
Tenant.objects.all().delete()
|
|
TenantGroup.objects.all().delete()
|
|
Cluster.objects.all().delete()
|
|
ClusterType.objects.all().delete()
|
|
VirtualMachine.objects.all().delete()
|
|
Device.objects.all().delete()
|
|
Site.objects.all().delete()
|
|
DeviceRole.objects.all().delete()
|
|
DeviceType.objects.all().delete()
|
|
Manufacturer.objects.all().delete()
|
|
super().tearDownClass()
|