nautobot 2.0.0a2__py3-none-any.whl → 2.0.0b1__py3-none-any.whl

nautobot/virtualization/tests/test_filters.py

@@ -5,7 +5,7 @@ from nautobot.dcim.choices import InterfaceModeChoices
 from nautobot.dcim.models import Device, DeviceType, Location, LocationType, Manufacturer, Platform
 from nautobot.extras.models import Role, Status, Tag
 from nautobot.ipam.choices import ServiceProtocolChoices
-from nautobot.ipam.models import IPAddress, VLAN, Service
+from nautobot.ipam.models import IPAddress, VLAN, Service, Namespace, Prefix
 from nautobot.tenancy.models import Tenant
 from nautobot.virtualization.filters import (
     ClusterTypeFilterSet,
@@ -23,17 +23,16 @@ from nautobot.virtualization.models import (
 )
 
 
-class ClusterTypeTestCase(FilterTestCases.NameSlugFilterTestCase):
+class ClusterTypeTestCase(FilterTestCases.NameOnlyFilterTestCase):
     queryset = ClusterType.objects.all()
     filterset = ClusterTypeFilterSet
 
     @classmethod
     def setUpTestData(cls):
-
         cluster_types = (
-            ClusterType.objects.create(name="Cluster Type 1", slug="cluster-type-1", description="A"),
-            ClusterType.objects.create(name="Cluster Type 2", slug="cluster-type-2", description="B"),
-            ClusterType.objects.create(name="Cluster Type 3", slug="cluster-type-3", description="C"),
+            ClusterType.objects.create(name="Cluster Type 1", description="A"),
+            ClusterType.objects.create(name="Cluster Type 2", description="B"),
+            ClusterType.objects.create(name="Cluster Type 3", description="C"),
         )
 
         cls.clusters = [
@@ -58,22 +57,22 @@ class ClusterTypeTestCase(FilterTestCases.NameSlugFilterTestCase):
             self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
 
 
-class ClusterGroupTestCase(FilterTestCases.NameSlugFilterTestCase):
+class ClusterGroupTestCase(FilterTestCases.NameOnlyFilterTestCase):
     queryset = ClusterGroup.objects.all()
     filterset = ClusterGroupFilterSet
 
     @classmethod
     def setUpTestData(cls):
-
         cluster_groups = (
-            ClusterGroup.objects.create(name="Cluster Group 1", slug="cluster-group-1", description="A"),
-            ClusterGroup.objects.create(name="Cluster Group 2", slug="cluster-group-2", description="B"),
-            ClusterGroup.objects.create(name="Cluster Group 3", slug="cluster-group-3", description="C"),
+            ClusterGroup.objects.create(name="Cluster Group 1", description="A"),
+            ClusterGroup.objects.create(name="Cluster Group 2", description="B"),
+            ClusterGroup.objects.create(name="Cluster Group 3", description="C"),
         )
 
         cluster_types = (
-            ClusterType.objects.create(name="Cluster Type 1", slug="cluster-type-1"),
-            ClusterType.objects.create(name="Cluster Type 2", slug="cluster-type-2"),
+            ClusterType.objects.create(name="Cluster Type 1", description="A"),
+            ClusterType.objects.create(name="Cluster Type 2", description="B"),
+            ClusterType.objects.create(name="Cluster Type 3", description="C"),
         )
 
         cls.clusters = (
@@ -105,27 +104,29 @@ class ClusterTestCase(FilterTestCases.FilterTestCase, FilterTestCases.TenancyFil
 
     @classmethod
     def setUpTestData(cls):
-
         cluster_types = (
-            ClusterType.objects.create(name="Cluster Type 1", slug="cluster-type-1"),
-            ClusterType.objects.create(name="Cluster Type 2", slug="cluster-type-2"),
-            ClusterType.objects.create(name="Cluster Type 3", slug="cluster-type-3"),
+            ClusterType.objects.create(name="Cluster Type 1", description="A"),
+            ClusterType.objects.create(name="Cluster Type 2", description="B"),
+            ClusterType.objects.create(name="Cluster Type 3", description="C"),
         )
 
         cluster_groups = (
-            ClusterGroup.objects.create(name="Cluster Group 1", slug="cluster-group-1"),
-            ClusterGroup.objects.create(name="Cluster Group 2", slug="cluster-group-2"),
-            ClusterGroup.objects.create(name="Cluster Group 3", slug="cluster-group-3"),
+            ClusterGroup.objects.create(name="Cluster Group 1", description="A"),
+            ClusterGroup.objects.create(name="Cluster Group 2", description="B"),
+            ClusterGroup.objects.create(name="Cluster Group 3", description="C"),
         )
 
         location_type_1 = LocationType.objects.get(name="Campus")
         location_type_2 = LocationType.objects.get(name="Building")
+        location_status = Status.objects.get_for_model(Location).first()
         cls.locations = (
-            Location.objects.create(name="Location 1", location_type=location_type_1),
-            Location.objects.create(name="Location 2", location_type=location_type_2),
-            Location.objects.create(name="Location 3", location_type=location_type_2),
+            Location(name="Location 1", location_type=location_type_1, status=location_status),
+            Location(name="Location 2", location_type=location_type_2, status=location_status),
+            Location(name="Location 3", location_type=location_type_1, status=location_status),
         )
         cls.locations[1].parent = cls.locations[0]
+        for location in cls.locations:
+            location.validated_save()
 
         tenants = Tenant.objects.filter(tenant_group__isnull=False)[:3]
 
@@ -156,15 +157,24 @@ class ClusterTestCase(FilterTestCases.FilterTestCase, FilterTestCases.TenancyFil
             ),
         )
 
-        manufacturer = Manufacturer.objects.create(name="Manufacturer 1", slug="manufacturer-1")
+        manufacturer = Manufacturer.objects.first()
         devicetype = DeviceType.objects.create(manufacturer=manufacturer, model="Device Type", slug="device-type")
         devicerole = Role.objects.get_for_model(Device).first()
+        devicestatus = Status.objects.get_for_model(Device).first()
 
         cls.device = Device.objects.create(
-            name="Device 1", device_type=devicetype, role=devicerole, location=cls.locations[0], cluster=clusters[0]
+            name="Device 1",
+            device_type=devicetype,
+            role=devicerole,
+            status=devicestatus,
+            location=cls.locations[0],
+            cluster=clusters[0],
         )
 
-        cls.virtualmachine = VirtualMachine.objects.create(name="Virtual Machine 1", cluster=clusters[1])
+        vm_status = Status.objects.get_for_model(VirtualMachine).first()
+        cls.virtualmachine = VirtualMachine.objects.create(
+            name="Virtual Machine 1", cluster=clusters[1], status=vm_status
+        )
 
         clusters[0].tags.set(Tag.objects.get_for_model(Cluster))
         clusters[1].tags.set(Tag.objects.get_for_model(Cluster)[:3])
@@ -215,7 +225,7 @@ class ClusterTestCase(FilterTestCases.FilterTestCase, FilterTestCases.TenancyFil
         cluster_groups = list(ClusterGroup.objects.all()[:2])
         filter_params = [
             {"cluster_group_id": [cluster_groups[0].pk, cluster_groups[1].pk]},
-            {"cluster_group": [cluster_groups[0].pk, cluster_groups[1].slug]},
+            {"cluster_group": [cluster_groups[0].pk, cluster_groups[1].name]},
         ]
         for params in filter_params:
             self.assertQuerysetEqualAndNotEmpty(
@@ -227,7 +237,7 @@ class ClusterTestCase(FilterTestCases.FilterTestCase, FilterTestCases.TenancyFil
         cluster_types = list(ClusterType.objects.all()[:2])
         filter_params = [
             {"cluster_type_id": [cluster_types[0].pk, cluster_types[1].pk]},
-            {"cluster_type": [cluster_types[0].pk, cluster_types[1].slug]},
+            {"cluster_type": [cluster_types[0].pk, cluster_types[1].name]},
         ]
         for params in filter_params:
             self.assertQuerysetEqualAndNotEmpty(
@@ -248,27 +258,29 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
 
     @classmethod
     def setUpTestData(cls):
-
         cluster_types = (
-            ClusterType.objects.create(name="Cluster Type 1", slug="cluster-type-1"),
-            ClusterType.objects.create(name="Cluster Type 2", slug="cluster-type-2"),
-            ClusterType.objects.create(name="Cluster Type 3", slug="cluster-type-3"),
+            ClusterType.objects.create(name="Cluster Type 1", description="A"),
+            ClusterType.objects.create(name="Cluster Type 2", description="B"),
+            ClusterType.objects.create(name="Cluster Type 3", description="C"),
         )
 
         cluster_groups = (
-            ClusterGroup.objects.create(name="Cluster Group 1", slug="cluster-group-1"),
-            ClusterGroup.objects.create(name="Cluster Group 2", slug="cluster-group-2"),
-            ClusterGroup.objects.create(name="Cluster Group 3", slug="cluster-group-3"),
+            ClusterGroup.objects.create(name="Cluster Group 1", description="A"),
+            ClusterGroup.objects.create(name="Cluster Group 2", description="B"),
+            ClusterGroup.objects.create(name="Cluster Group 3", description="C"),
         )
 
         location_type_1 = LocationType.objects.get(name="Campus")
         location_type_2 = LocationType.objects.get(name="Building")
+        location_status = Status.objects.get_for_model(Location).first()
         cls.locations = (
-            Location.objects.create(name="Location 1", location_type=location_type_1),
-            Location.objects.create(name="Location 2", location_type=location_type_2),
-            Location.objects.create(name="Location 3", location_type=location_type_2),
+            Location(name="Location 1", location_type=location_type_1, status=location_status),
+            Location(name="Location 2", location_type=location_type_2, status=location_status),
+            Location(name="Location 3", location_type=location_type_1, status=location_status),
         )
         cls.locations[1].parent = cls.locations[0]
+        for location in cls.locations:
+            location.validated_save()
 
         clusters = (
             Cluster.objects.create(
@@ -291,11 +303,7 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
             ),
         )
 
-        platforms = (
-            Platform.objects.create(name="Platform 1", slug="platform-1"),
-            Platform.objects.create(name="Platform 2", slug="platform-2"),
-            Platform.objects.create(name="Platform 3", slug="platform-3"),
-        )
+        platforms = Platform.objects.all()[:3]
         cls.platforms = platforms
 
         roles = Role.objects.get_for_model(VirtualMachine)
@@ -303,8 +311,7 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
 
         tenants = Tenant.objects.filter(tenant_group__isnull=False)[:3]
 
-        statuses = Status.objects.get_for_model(VirtualMachine)
-        status_map = {s.slug: s for s in statuses.all()}
+        cls.statuses = Status.objects.get_for_model(VirtualMachine)
 
         vms = (
             VirtualMachine.objects.create(
@@ -313,7 +320,7 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
                 platform=platforms[0],
                 role=roles[0],
                 tenant=tenants[0],
-                status=status_map["active"],
+                status=cls.statuses[0],
                 vcpus=1,
                 memory=1,
                 disk=1,
@@ -326,7 +333,7 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
                 platform=platforms[1],
                 role=roles[1],
                 tenant=tenants[1],
-                status=status_map["staged"],
+                status=cls.statuses[2],
                 vcpus=2,
                 memory=2,
                 disk=2,
@@ -338,7 +345,7 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
                 platform=platforms[2],
                 role=roles[2],
                 tenant=tenants[2],
-                status=status_map["offline"],
+                status=cls.statuses[1],
                 vcpus=3,
                 memory=3,
                 disk=3,
@@ -350,7 +357,7 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
                 platform=platforms[2],
                 role=roles[2],
                 tenant=tenants[2],
-                status=status_map["offline"],
+                status=cls.statuses[1],
                 vcpus=3,
                 memory=3,
                 disk=3,
@@ -362,7 +369,7 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
                 platform=platforms[2],
                 role=roles[2],
                 tenant=tenants[2],
-                status=status_map["offline"],
+                status=cls.statuses[1],
                 vcpus=3,
                 memory=3,
                 disk=3,
@@ -374,7 +381,7 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
                 platform=platforms[2],
                 role=roles[2],
                 tenant=tenants[2],
-                status=status_map["offline"],
+                status=cls.statuses[1],
                 vcpus=3,
                 memory=3,
                 disk=3,
@@ -382,21 +389,25 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
             ),
         )
 
+        int_status = Status.objects.get_for_model(VMInterface).first()
         cls.interfaces = (
             VMInterface.objects.create(
                 virtual_machine=vms[0],
                 name="Interface 1",
                 mac_address="00-00-00-00-00-01",
+                status=int_status,
             ),
             VMInterface.objects.create(
                 virtual_machine=vms[1],
                 name="Interface 2",
                 mac_address="00-00-00-00-00-02",
+                status=int_status,
             ),
             VMInterface.objects.create(
                 virtual_machine=vms[2],
                 name="Interface 3",
                 mac_address="00-00-00-00-00-03",
+                status=int_status,
             ),
         )
 
@@ -415,10 +426,19 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
             ),
         )
 
+        cls.namespace = Namespace.objects.first()
+        cls.prefix_status = Status.objects.get_for_model(Prefix).first()
+        cls.ipadd_status = Status.objects.get_for_model(IPAddress).first()
+        cls.prefix4 = Prefix.objects.create(prefix="192.0.2.0/24", namespace=cls.namespace, status=cls.prefix_status)
+        cls.prefix6 = Prefix.objects.create(
+            prefix="fe80::8ef:3eff:fe4c:3895/24", namespace=cls.namespace, status=cls.prefix_status
+        )
         # Assign primary IPs for filtering
         cls.ipaddresses = (
-            IPAddress.objects.create(address="192.0.2.1/24"),
-            IPAddress.objects.create(address="fe80::8ef:3eff:fe4c:3895/24"),
+            IPAddress.objects.create(address="192.0.2.1/24", namespace=cls.namespace, status=cls.ipadd_status),
+            IPAddress.objects.create(
+                address="fe80::8ef:3eff:fe4c:3895/24", namespace=cls.namespace, status=cls.ipadd_status
+            ),
         )
         cls.interfaces[0].add_ip_addresses(cls.ipaddresses[0])
         cls.interfaces[1].add_ip_addresses(cls.ipaddresses[1])
@@ -472,14 +492,17 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
         self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
 
     def test_status(self):
-        params = {"status": ["active", "staged"]}
-        self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
+        params = {"status": [self.statuses[0].name, self.statuses[1].name]}
+        self.assertQuerysetEqualAndNotEmpty(
+            self.filterset(params, self.queryset).qs,
+            self.queryset.filter(status__in=[self.statuses[0], self.statuses[1]]),
+        )
 
     def test_cluster_group(self):
         groups = list(ClusterGroup.objects.all()[:2])
         filter_params = [
             {"cluster_group_id": [groups[0].pk, groups[1].pk]},
-            {"cluster_group": [groups[0].pk, groups[1].slug]},
+            {"cluster_group": [groups[0].pk, groups[1].name]},
         ]
         for params in filter_params:
             self.assertQuerysetEqualAndNotEmpty(
@@ -491,7 +514,7 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
         types = list(ClusterType.objects.all()[:2])
         filter_params = [
             {"cluster_type_id": [types[0].pk, types[1].pk]},
-            {"cluster_type": [types[0].pk, types[1].slug]},
+            {"cluster_type": [types[0].pk, types[1].name]},
         ]
         for params in filter_params:
             self.assertQuerysetEqualAndNotEmpty(
@@ -520,7 +543,7 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
 
     def test_role(self):
         roles = self.roles[:2]
-        params = {"role": [roles[0].pk, roles[1].slug]}
+        params = {"role": [roles[0].pk, roles[1].name]}
         self.assertQuerysetEqualAndNotEmpty(
             self.filterset(params, self.queryset).qs, self.queryset.filter(role__in=[roles[0], roles[1]])
         )
@@ -529,11 +552,12 @@ class VirtualMachineTestCase(FilterTestCases.FilterTestCase, FilterTestCases.Ten
         platforms = self.platforms[:2]
         filter_params = [
             {"platform_id": [platforms[0].pk, platforms[1].pk]},
-            {"platform": [platforms[0].pk, platforms[1].slug]},
+            {"platform": [platforms[0].pk, platforms[1].name]},
         ]
         for params in filter_params:
             self.assertQuerysetEqualAndNotEmpty(
-                self.filterset(params, self.queryset).qs, self.queryset.filter(platform__in=platforms).distinct()
+                self.filterset(params, self.queryset).qs,
+                self.queryset.filter(platform__in=[platforms[0], platforms[1]]).distinct(),
             )
 
     def test_mac_address(self):
@@ -585,7 +609,6 @@ class VMInterfaceTestCase(FilterTestCases.FilterTestCase):
         ["mtu"],
         ["name"],
         ["parent_interface"],
-        ["status", "status__slug"],
         ["tagged_vlans", "tagged_vlans__pk"],
         ["tagged_vlans", "tagged_vlans__vid"],
         ["untagged_vlan", "untagged_vlan__pk"],
@@ -596,11 +619,10 @@ class VMInterfaceTestCase(FilterTestCases.FilterTestCase):
 
     @classmethod
     def setUpTestData(cls):
-
         cluster_types = (
-            ClusterType.objects.create(name="Cluster Type 1", slug="cluster-type-1"),
-            ClusterType.objects.create(name="Cluster Type 2", slug="cluster-type-2"),
-            ClusterType.objects.create(name="Cluster Type 3", slug="cluster-type-3"),
+            ClusterType.objects.create(name="Cluster Type 1", description="A"),
+            ClusterType.objects.create(name="Cluster Type 2", description="B"),
+            ClusterType.objects.create(name="Cluster Type 3", description="C"),
         )
 
         clusters = (
@@ -609,10 +631,11 @@ class VMInterfaceTestCase(FilterTestCases.FilterTestCase):
             Cluster.objects.create(name="Cluster 3", cluster_type=cluster_types[2]),
         )
 
+        vm_status = Status.objects.get_for_model(VirtualMachine).first()
         vms = (
-            VirtualMachine.objects.create(name="Virtual Machine 1", cluster=clusters[0]),
-            VirtualMachine.objects.create(name="Virtual Machine 2", cluster=clusters[1]),
-            VirtualMachine.objects.create(name="Virtual Machine 3", cluster=clusters[2]),
+            VirtualMachine.objects.create(name="Virtual Machine 1", cluster=clusters[0], status=vm_status),
+            VirtualMachine.objects.create(name="Virtual Machine 2", cluster=clusters[1], status=vm_status),
+            VirtualMachine.objects.create(name="Virtual Machine 3", cluster=clusters[2], status=vm_status),
         )
 
         statuses = Status.objects.get_for_model(VMInterface)
@@ -628,7 +651,7 @@ class VMInterfaceTestCase(FilterTestCases.FilterTestCase):
                 enabled=True,
                 mtu=100,
                 mac_address="00-00-00-00-00-01",
-                status=statuses.get(slug="active"),
+                status=statuses[0],
                 description="This is a description of Interface1",
                 mode=InterfaceModeChoices.MODE_ACCESS,
                 untagged_vlan=cls.vlan1,
@@ -639,7 +662,7 @@ class VMInterfaceTestCase(FilterTestCases.FilterTestCase):
                 enabled=True,
                 mtu=200,
                 mac_address="00-00-00-00-00-02",
-                status=statuses.get(slug="active"),
+                status=statuses[0],
                 description="This is a description of Interface2",
                 mode=InterfaceModeChoices.MODE_ACCESS,
                 untagged_vlan=cls.vlan2,
@@ -650,7 +673,7 @@ class VMInterfaceTestCase(FilterTestCases.FilterTestCase):
                 enabled=False,
                 mtu=300,
                 mac_address="00-00-00-00-00-03",
-                status=statuses.get(slug="planned"),
+                status=statuses[1],
                 description="This is a description of Interface3",
                 mode=InterfaceModeChoices.MODE_TAGGED,
             ),
@@ -664,7 +687,7 @@ class VMInterfaceTestCase(FilterTestCases.FilterTestCase):
                 enabled=False,
                 mtu=300,
                 mac_address="00-00-00-00-00-04",
-                status=statuses.get(slug="planned"),
+                status=statuses[1],
                 description="This is a description of Interface4",
                 mode=InterfaceModeChoices.MODE_TAGGED,
             ),
@@ -676,7 +699,7 @@ class VMInterfaceTestCase(FilterTestCases.FilterTestCase):
                 enabled=False,
                 mtu=300,
                 mac_address="00-00-00-00-00-05",
-                status=statuses.get(slug="maintenance"),
+                status=statuses[2],
                 description="This is a description of Interface5",
                 mode=InterfaceModeChoices.MODE_TAGGED_ALL,
             ),
@@ -686,10 +709,10 @@ class VMInterfaceTestCase(FilterTestCases.FilterTestCase):
         vminterfaces[2].tagged_vlans.add(cls.vlan2)
 
         # Assign primary IPs for filtering
-        ip_address4 = IPAddress.objects.ip_family(4).first()
+        ip_address4 = IPAddress.objects.filter(ip_version=4).first()
         vminterfaces[0].add_ip_addresses(ip_address4)
         ip_address4.validated_save()
-        ip_address6 = IPAddress.objects.ip_family(6).first()
+        ip_address6 = IPAddress.objects.filter(ip_version=6).first()
         vminterfaces[1].add_ip_addresses(ip_address6)
         ip_address6.validated_save()
 

nautobot/virtualization/tests/test_models.py

@@ -11,9 +11,10 @@ from nautobot.virtualization.models import VirtualMachine, ClusterType, Cluster,
 
 class ClusterTestCase(TestCase):  # TODO: change to BaseModelTestCase
     def test_cluster_validation(self):
-        cluster_type = ClusterType.objects.create(name="Test Cluster Type 1")
+        cluster_type = ClusterType.objects.create(name="Cluster Type 1")
         location_type = LocationType.objects.create(name="Location Type 1")
-        location = Location.objects.create(name="Location 1", location_type=location_type)
+        location_status = Status.objects.get_for_model(Location).first()
+        location = Location.objects.create(name="Location 1", location_type=location_type, status=location_status)
         cluster = Cluster(name="Test Cluster 1", cluster_type=cluster_type, location=location)
         with self.assertRaises(ValidationError) as cm:
             cluster.validated_save()
@@ -26,12 +27,11 @@ class VirtualMachineTestCase(TestCase):  # TODO: change to BaseModelTestCase
     def setUp(self):
         statuses = Status.objects.get_for_model(VirtualMachine)
 
-        cluster_type = ClusterType.objects.create(name="Test Cluster Type 1", slug="test-cluster-type-1")
+        cluster_type = ClusterType.objects.create(name="Cluster Type 1")
         self.cluster = Cluster.objects.create(name="Test Cluster 1", cluster_type=cluster_type)
-        self.status = statuses.get(slug="active")
+        self.status = statuses[0]
 
     def test_vm_duplicate_name_per_cluster(self):
-
         vm1 = VirtualMachine(
             cluster=self.cluster,
             name="Test VM 1",
@@ -49,7 +49,7 @@ class VirtualMachineTestCase(TestCase):  # TODO: change to BaseModelTestCase
         with self.assertRaises(ValidationError):
             vm2.full_clean()
 
-        tenant = Tenant.objects.create(name="Test Tenant 1", slug="test-tenant-1")
+        tenant = Tenant.objects.first()
         vm1.tenant = tenant
         vm1.save()
         vm2.tenant = tenant
@@ -66,15 +66,21 @@ class VirtualMachineTestCase(TestCase):  # TODO: change to BaseModelTestCase
 
 
 class VMInterfaceTestCase(TestCase):  # TODO: change to BaseModelTestCase
-    def setUp(self):
+    @classmethod
+    def setUpTestData(cls):
         location = Location.objects.filter(location_type=LocationType.objects.get(name="Campus")).first()
-        self.vlan = VLAN.objects.create(name="VLAN 1", vid=100, location=location)
-        clustertype = ClusterType.objects.create(name="Test Cluster Type 1", slug="test-cluster-type-1")
+        vlan_status = Status.objects.get_for_model(VLAN).first()
+        cls.vlan = VLAN.objects.create(name="VLAN 1", vid=100, location=location, status=vlan_status)
+        clustertype = ClusterType.objects.create(name="Cluster Type 1")
         cluster = Cluster.objects.create(name="Test Cluster 1", cluster_type=clustertype)
-        self.virtualmachine = VirtualMachine.objects.create(cluster=cluster, name="Test VM 1")
+        vm_status = Status.objects.get_for_model(VirtualMachine).first()
+        cls.virtualmachine = VirtualMachine.objects.create(cluster=cluster, name="Test VM 1", status=vm_status)
+        cls.int_status = Status.objects.get_for_model(VMInterface).first()
 
     def test_tagged_vlan_raise_error_if_mode_not_set_to_tagged(self):
-        interface = VMInterface.objects.create(virtual_machine=self.virtualmachine, name="Interface 1")
+        interface = VMInterface.objects.create(
+            virtual_machine=self.virtualmachine, name="Interface 1", status=self.int_status
+        )
         with self.assertRaises(ValidationError) as err:
             interface.tagged_vlans.add(self.vlan)
         self.assertEqual(
@@ -83,7 +89,9 @@ class VMInterfaceTestCase(TestCase):  # TODO: change to BaseModelTestCase
 
     def test_add_ip_addresses(self):
         """Test the `add_ip_addresses` helper method on `VMInterface`"""
-        vm_interface = VMInterface.objects.create(name="Int1", virtual_machine=self.virtualmachine)
+        vm_interface = VMInterface.objects.create(
+            name="Int1", virtual_machine=self.virtualmachine, status=self.int_status
+        )
         ips = list(IPAddress.objects.all()[:10])
 
         # baseline (no vm_interface to ip address relationships exists)
@@ -103,7 +111,9 @@ class VMInterfaceTestCase(TestCase):  # TODO: change to BaseModelTestCase
 
     def test_remove_ip_addresses(self):
         """Test the `remove_ip_addresses` helper method on `VMInterface`"""
-        vm_interface = VMInterface.objects.create(name="Int1", virtual_machine=self.virtualmachine)
+        vm_interface = VMInterface.objects.create(
+            name="Int1", virtual_machine=self.virtualmachine, status=self.int_status
+        )
         ips = list(IPAddress.objects.all()[:10])
 
         # baseline (no vm_interface to ip address relationships exists)
@@ -125,3 +135,16 @@ class VMInterfaceTestCase(TestCase):  # TODO: change to BaseModelTestCase
         count = vm_interface.remove_ip_addresses(ips)
         self.assertEqual(count, 4)
         self.assertEqual(IPAddressToInterface.objects.filter(vm_interface=vm_interface).count(), 0)
+
+        # Test the pre_delete signal for IPAddressToInterface instances
+        vm_interface.add_ip_addresses(ips)
+        self.virtualmachine.primary_ip4 = vm_interface.ip_addresses.all().filter(host__family=4).first()
+        self.virtualmachine.primary_ip6 = vm_interface.ip_addresses.all().filter(host__family=6).first()
+        self.virtualmachine.save()
+
+        vm_interface.remove_ip_addresses(self.virtualmachine.primary_ip4)
+        self.virtualmachine.refresh_from_db()
+        self.assertEqual(self.virtualmachine.primary_ip4, None)
+        vm_interface.remove_ip_addresses(self.virtualmachine.primary_ip6)
+        self.virtualmachine.refresh_from_db()
+        self.assertEqual(self.virtualmachine.primary_ip6, None)