nautobot 2.4.17__py3-none-any.whl → 2.4.19__py3-none-any.whl

nautobot/ipam/templates/ipam/prefix_create.html

@@ -0,0 +1,34 @@
+{% extends 'generic/object_create.html' %}
+{% load form_helpers %}
+
+{% block form %}
+    <div class="panel panel-default">
+        <div class="panel-heading"><strong>Prefix</strong></div>
+        <div class="panel-body">
+            {% render_field form.prefix %}
+            {% render_field form.namespace %}
+            {% render_field form.type %}
+            {% render_field form.status %}
+            {% render_field form.role %}
+            {% render_field form.rir %}
+            {% render_field form.date_allocated %}
+            {% render_field form.description %}
+        </div>
+    </div>
+    <div class="panel panel-default">
+        <div class="panel-heading"><strong>VRF Assignment</strong></div>
+        <div class="panel-body">
+            {% render_field form.vrfs %}
+        </div>
+    </div>
+    <div class="panel panel-default">
+        <div class="panel-heading"><strong>Location/VLAN Assignment</strong></div>
+        <div class="panel-body">
+            {% render_field form.locations %}
+            {% render_field form.vlan_group %}
+            {% render_field form.vlan %}
+        </div>
+    </div>
+    {% include 'inc/tenancy_form_panel.html' %}
+    {% include 'inc/extras_features_edit_form_fields.html' %}
+{% endblock %}

nautobot/ipam/templates/ipam/prefix_edit.html

@@ -1,34 +1 @@
-{% extends 'generic/object_create.html' %}
-{% load form_helpers %}
-
-{% block form %}
-    <div class="panel panel-default">
-        <div class="panel-heading"><strong>Prefix</strong></div>
-        <div class="panel-body">
-            {% render_field form.prefix %}
-            {% render_field form.namespace %}
-            {% render_field form.type %}
-            {% render_field form.status %}
-            {% render_field form.role %}
-            {% render_field form.rir %}
-            {% render_field form.date_allocated %}
-            {% render_field form.description %}
-        </div>
-    </div>
-    <div class="panel panel-default">
-        <div class="panel-heading"><strong>VRF Assignment</strong></div>
-        <div class="panel-body">
-            {% render_field form.vrfs %}
-        </div>
-    </div>
-    <div class="panel panel-default">
-        <div class="panel-heading"><strong>Location/VLAN Assignment</strong></div>
-        <div class="panel-body">
-            {% render_field form.locations %}
-            {% render_field form.vlan_group %}
-            {% render_field form.vlan %}
-        </div>
-    </div>
-    {% include 'inc/tenancy_form_panel.html' %}
-    {% include 'inc/extras_features_edit_form_fields.html' %}
-{% endblock %}
+{% extends 'ipam/prefix_create.html' %}

nautobot/ipam/templates/ipam/prefix_retrieve.html

@@ -0,0 +1,3 @@
+
+{% extends 'generic/object_retrieve.html' %}
+{% load helpers %}

nautobot/ipam/templates/ipam/service_retrieve.html

@@ -1,7 +1,2 @@
 {% extends 'generic/object_retrieve.html' %}
-{% load helpers %}
-
-{% block extra_breadcrumbs %}
-            <li>{{ object.parent|hyperlinked_object }}</li>
-{% endblock extra_breadcrumbs %}
-
+{% comment %}3.0 TODO: remove this template, which only exists for backward compatibility with 2.4 and earlier{% endcomment %}

nautobot/ipam/templates/ipam/vlan_retrieve.html

@@ -1,8 +1,2 @@
 {% extends 'generic/object_retrieve.html' %}
-{% load helpers %}
-
-{% block extra_breadcrumbs %}
-    {% if object.vlan_group %}
-        <li><a href="{% url 'ipam:vlan_list' %}?vlan_group={{ object.vlan_group.name }}">{{ object.vlan_group }}</a></li>
-    {% endif %}
-{% endblock extra_breadcrumbs %}
+{% comment %}3.0 TODO: remove this template, which only exists for backward compatibility with 2.4 and earlier{% endcomment %}

nautobot/ipam/templates/ipam/vrf_edit.html

@@ -7,7 +7,7 @@
         <div class="panel-body">
             {% render_field form.name %}
             {% render_field form.namespace %}
-	        {% render_field form.rd %}
+            {% render_field form.rd %}
             {% render_field form.status %}
             {% render_field form.description %}
         </div>

nautobot/ipam/tests/test_api.py

@@ -413,6 +413,11 @@ class PrefixTest(APIViewTestCases.APIViewTestCase):
         }
         cls.choices_fields = ["type"]
 
+        # Generic `test_update_object()` will grab and update the first Prefix
+        first_pfx = Prefix.objects.first()
+        cls.update_data = cls.create_data[0].copy()
+        cls.update_data["namespace"] = first_pfx.namespace.pk  # can't change network and namespace in the same update
+
     def test_legacy_api_behavior(self):
         """
         Tests for the 2.0/2.1 REST API of Prefixes.

nautobot/ipam/tests/test_models.py

@@ -284,6 +284,393 @@ class IPAddressToInterfaceTest(TestCase):
         self.assertEqual(remaining_assignments.count(), 1)
         self.assertIn(assignment_nested_module, remaining_assignments)
 
+    def test_ip_address_to_interface_delete_signal_no_save_when_device_has_no_primary_ips(self):
+        """
+        Test that Device is not saved when removing an IP from an interface and the device
+        has no primary IPs assigned.
+        """
+        # Create an IP address
+        ip_addr = IPAddress.objects.create(address="192.0.2.1/24", status=self.status, namespace=self.namespace)
+
+        # Assign IP to interface
+        assignment_device_int1 = IPAddressToInterface.objects.create(interface=self.test_int1, ip_address=ip_addr)
+
+        # Mock the device save method to verify it's not called
+        with patch.object(self.test_device, "save") as mock_save:
+            # Remove the IP assignment from interface - this should NOT trigger a device save
+            assignment_device_int1.delete()
+
+            # Assert save was not called since no primary IPs were affected
+            mock_save.assert_not_called()
+
+    def test_ip_address_to_interface_delete_signal_no_save_when_removing_non_primary_ip_from_device(self):
+        """
+        Test that Device is not saved when removing an IP from an interface and the IP
+        is not the device's primary IP.
+        """
+        # Test removing non-primary IPv4 assignment
+        with self.subTest("IPv4 non-primary IP removal"):
+            # Create IPv4 addresses
+            primary_ipv4_addr = IPAddress.objects.create(
+                address="192.0.2.1/24", status=self.status, namespace=self.namespace
+            )
+            other_ipv4_addr = IPAddress.objects.create(
+                address="192.0.2.2/24", status=self.status, namespace=self.namespace
+            )
+
+            # Assign primary IP to interface first (required before setting as primary)
+            IPAddressToInterface.objects.create(interface=self.test_int2, ip_address=primary_ipv4_addr)
+
+            # Set it as the device's primary IP
+            self.test_device.primary_ip4 = primary_ipv4_addr
+            self.test_device.save()
+
+            # Assign the other IP to a different interface
+            assignment_other_ipv4 = IPAddressToInterface.objects.create(
+                interface=self.test_int1, ip_address=other_ipv4_addr
+            )
+
+            with patch.object(self.test_device, "save") as mock_save:
+                assignment_other_ipv4.delete()
+                mock_save.assert_not_called()
+
+        # Test removing non-primary IPv6 assignment
+        with self.subTest("IPv6 non-primary IP removal"):
+            # Create IPv6 prefix first
+            Prefix.objects.create(prefix="2001:db8::/64", status=self.status, namespace=self.namespace)
+
+            # Create IPv6 addresses
+            primary_ipv6_addr = IPAddress.objects.create(
+                address="2001:db8::1/64", status=self.status, namespace=self.namespace
+            )
+            other_ipv6_addr = IPAddress.objects.create(
+                address="2001:db8::2/64", status=self.status, namespace=self.namespace
+            )
+
+            # Assign primary IP to interface first (required before setting as primary)
+            IPAddressToInterface.objects.create(interface=self.test_int2, ip_address=primary_ipv6_addr)
+
+            # Set it as the device's primary IP
+            self.test_device.primary_ip6 = primary_ipv6_addr
+            self.test_device.save()
+
+            # Assign the other IP to a different interface
+            assignment_other_ipv6 = IPAddressToInterface.objects.create(
+                interface=self.test_int1, ip_address=other_ipv6_addr
+            )
+
+            with patch.object(self.test_device, "save") as mock_save:
+                assignment_other_ipv6.delete()
+                mock_save.assert_not_called()
+
+    def test_ip_address_to_interface_delete_signal_save_when_removing_primary_ip_from_device(self):
+        """
+        Test that Device is saved when removing an IP from an interface and the IP is the device's
+        primary IP and no other assignments exist.
+        """
+        # Test removing primary IPv4 assignment
+        with self.subTest("IPv4 primary IP removal"):
+            # Create primary IPv4 address
+            primary_ipv4_addr = IPAddress.objects.create(
+                address="192.0.2.1/24", status=self.status, namespace=self.namespace
+            )
+
+            # Assign primary IPv4 to an interface first
+            assignment_primary_ipv4 = IPAddressToInterface.objects.create(
+                interface=self.test_int1, ip_address=primary_ipv4_addr
+            )
+
+            # Set it as the device's primary IP
+            self.test_device.primary_ip4 = primary_ipv4_addr
+            self.test_device.save()
+
+            # Mock the device save method to verify it's called
+            with patch.object(self.test_device, "save") as mock_save:
+                # Remove the primary IP assignment - this SHOULD trigger a device save
+                assignment_primary_ipv4.delete()
+
+                # Assert save was called to nullify primary_ip4
+                mock_save.assert_called_once()
+
+        # Test removing primary IPv6 assignment
+        with self.subTest("IPv6 primary IP removal"):
+            # Create IPv6 prefix first
+            Prefix.objects.create(prefix="2001:db8::/64", status=self.status, namespace=self.namespace)
+
+            # Create primary IPv6 address
+            primary_ipv6_addr = IPAddress.objects.create(
+                address="2001:db8::1/64", status=self.status, namespace=self.namespace
+            )
+
+            # Assign primary IPv6 to an interface first
+            assignment_primary_ipv6 = IPAddressToInterface.objects.create(
+                interface=self.test_int1, ip_address=primary_ipv6_addr
+            )
+
+            # Set it as the device's primary IP
+            self.test_device.primary_ip6 = primary_ipv6_addr
+            self.test_device.save()
+
+            # Mock the device save method to verify it's called
+            with patch.object(self.test_device, "save") as mock_save:
+                # Remove the primary IP assignment - this SHOULD trigger a device save
+                assignment_primary_ipv6.delete()
+
+                # Assert save was called to nullify primary_ip6
+                mock_save.assert_called_once()
+
+    def test_ip_address_to_interface_delete_signal_no_save_when_device_primary_ip_has_multiple_assignments(self):
+        """
+        Test that Device is not saved when removing an IP from an interface and the IP is the device's
+        primary IP but is assigned to other interfaces on that device.
+        """
+        # Test IPv4 primary IP with multiple assignments
+        with self.subTest("IPv4 primary IP with multiple assignments"):
+            # Create primary IPv4 address
+            primary_ipv4_addr = IPAddress.objects.create(
+                address="192.0.2.1/24", status=self.status, namespace=self.namespace
+            )
+            # Assign primary IPv4 to multiple interfaces
+            assignment_ipv4_int1 = IPAddressToInterface.objects.create(
+                interface=self.test_int1, ip_address=primary_ipv4_addr
+            )
+            IPAddressToInterface.objects.create(interface=self.test_int2, ip_address=primary_ipv4_addr)
+
+            # Set it as the device's primary IP
+            self.test_device.primary_ip4 = primary_ipv4_addr
+            self.test_device.save()
+
+            # Mock the device save method to verify it's not called
+            with patch.object(self.test_device, "save") as mock_save:
+                # Remove primary IP from one interface - should NOT trigger save since other assignment exists
+                assignment_ipv4_int1.delete()
+
+                # Assert save was not called since IP is still assigned to test_int2
+                mock_save.assert_not_called()
+
+        # Test IPv6 primary IP with multiple assignments
+        with self.subTest("IPv6 primary IP with multiple assignments"):
+            # Create IPv6 prefix first
+            Prefix.objects.create(prefix="2001:db8::/64", status=self.status, namespace=self.namespace)
+
+            # Create primary IPv6 address
+            primary_ipv6_addr = IPAddress.objects.create(
+                address="2001:db8::1/64", status=self.status, namespace=self.namespace
+            )
+
+            # Assign primary IPv6 to multiple interfaces
+            assignment_ipv6_int1 = IPAddressToInterface.objects.create(
+                interface=self.test_int1, ip_address=primary_ipv6_addr
+            )
+            IPAddressToInterface.objects.create(interface=self.test_int2, ip_address=primary_ipv6_addr)
+
+            # Set it as the device's primary IP
+            self.test_device.primary_ip6 = primary_ipv6_addr
+            self.test_device.save()
+
+            # Mock the device save method to verify it's not called
+            with patch.object(self.test_device, "save") as mock_save:
+                # Remove primary IP from one interface - should NOT trigger save since other assignment exists
+                assignment_ipv6_int1.delete()
+
+                # Assert save was not called since IP is still assigned to test_int2
+                mock_save.assert_not_called()
+
+    def test_ip_address_to_interface_delete_signal_no_save_when_vm_has_no_primary_ips(self):
+        """
+        Test that VM is not saved when removing an IP from a VM interface and the VM
+        has no primary IPs assigned.
+        """
+        # Create an IP address
+        ip_addr = IPAddress.objects.create(address="192.0.2.1/24", status=self.status, namespace=self.namespace)
+
+        # VM has no primary IPs by default (both are None)
+
+        # Assign IP to VM interface
+        assignment_vm_int1 = IPAddressToInterface.objects.create(vm_interface=self.test_vmint1, ip_address=ip_addr)
+
+        # Mock the VM save method to verify it's not called
+        with patch.object(self.test_vm, "save") as mock_save:
+            # Remove the IP assignment from VM interface - this should NOT trigger a VM save
+            assignment_vm_int1.delete()
+
+            # Assert save was not called since no primary IPs were affected
+            mock_save.assert_not_called()
+
+    def test_ip_address_to_interface_delete_signal_no_save_when_removing_non_primary_ip_from_vm(self):
+        """
+        Test that VM is not saved when removing an IP from a VM interface and the IP
+        is not the VM's primary IP.
+        """
+        # Test removing non-primary IPv4 assignment
+        with self.subTest("IPv4 non-primary IP removal"):
+            # Create IPv4 addresses
+            primary_ipv4_addr = IPAddress.objects.create(
+                address="192.0.2.1/24", status=self.status, namespace=self.namespace
+            )
+            other_ipv4_addr = IPAddress.objects.create(
+                address="192.0.2.2/24", status=self.status, namespace=self.namespace
+            )
+
+            # Assign primary IP to VM interface first (required before setting as primary)
+            IPAddressToInterface.objects.create(vm_interface=self.test_vmint2, ip_address=primary_ipv4_addr)
+
+            # Set it as the VM's primary IP
+            self.test_vm.primary_ip4 = primary_ipv4_addr
+            self.test_vm.save()
+
+            # Assign the other IP to a different VM interface
+            assignment_vm_int1 = IPAddressToInterface.objects.create(
+                vm_interface=self.test_vmint1, ip_address=other_ipv4_addr
+            )
+
+            with patch.object(self.test_vm, "save") as mock_save:
+                assignment_vm_int1.delete()
+                mock_save.assert_not_called()
+
+        # Test removing non-primary IPv6 assignment
+        with self.subTest("IPv6 non-primary IP removal"):
+            # Create IPv6 prefix first
+            Prefix.objects.create(prefix="2001:db8::/64", status=self.status, namespace=self.namespace)
+
+            # Create IPv6 addresses
+            primary_ipv6_addr = IPAddress.objects.create(
+                address="2001:db8::1/64", status=self.status, namespace=self.namespace
+            )
+            other_ipv6_addr = IPAddress.objects.create(
+                address="2001:db8::2/64", status=self.status, namespace=self.namespace
+            )
+
+            # Assign primary IP to VM interface first (required before setting as primary)
+            IPAddressToInterface.objects.create(vm_interface=self.test_vmint2, ip_address=primary_ipv6_addr)
+
+            # Set it as the VM's primary IP
+            self.test_vm.primary_ip6 = primary_ipv6_addr
+            self.test_vm.save()
+
+            # Assign the other IP to a different VM interface
+            assignment_vm_int1 = IPAddressToInterface.objects.create(
+                vm_interface=self.test_vmint1, ip_address=other_ipv6_addr
+            )
+
+            with patch.object(self.test_vm, "save") as mock_save:
+                assignment_vm_int1.delete()
+                mock_save.assert_not_called()
+
+    def test_ip_address_to_interface_delete_signal_save_when_removing_primary_ip_from_vm(self):
+        """
+        Test that VM is saved when removing an IP from a VM interface and the IP is the VM's
+        primary IP and no other assignments exist.
+        """
+        # Test removing primary IPv4 assignment
+        with self.subTest("IPv4 primary IP removal"):
+            # Create primary IPv4 address
+            primary_ipv4_addr = IPAddress.objects.create(
+                address="192.0.2.1/24", status=self.status, namespace=self.namespace
+            )
+
+            # Assign primary IPv4 to VM interface first
+            assignment_primary_ipv4 = IPAddressToInterface.objects.create(
+                vm_interface=self.test_vmint1, ip_address=primary_ipv4_addr
+            )
+
+            # Set it as the VM's primary IP
+            self.test_vm.primary_ip4 = primary_ipv4_addr
+            self.test_vm.save()
+
+            # Mock the VM save method to verify it's called
+            with patch.object(self.test_vm, "save") as mock_save:
+                # Remove the primary IP assignment - this SHOULD trigger a VM save
+                assignment_primary_ipv4.delete()
+
+                # Assert save was called to nullify primary_ip4
+                mock_save.assert_called_once()
+
+        # Test removing primary IPv6 assignment
+        with self.subTest("IPv6 primary IP removal"):
+            # Create IPv6 prefix first
+            Prefix.objects.create(prefix="2001:db8::/64", status=self.status, namespace=self.namespace)
+
+            # Create primary IPv6 address
+            primary_ipv6_addr = IPAddress.objects.create(
+                address="2001:db8::1/64", status=self.status, namespace=self.namespace
+            )
+
+            # Assign primary IPv6 to VM interface first
+            assignment_primary_ipv6 = IPAddressToInterface.objects.create(
+                vm_interface=self.test_vmint1, ip_address=primary_ipv6_addr
+            )
+
+            # Set it as the VM's primary IP
+            self.test_vm.primary_ip6 = primary_ipv6_addr
+            self.test_vm.save()
+
+            # Mock the VM save method to verify it's called
+            with patch.object(self.test_vm, "save") as mock_save:
+                # Remove the primary IP assignment - this SHOULD trigger a VM save
+                assignment_primary_ipv6.delete()
+
+                # Assert save was called to nullify primary_ip6
+                mock_save.assert_called_once()
+
+    def test_ip_address_to_interface_delete_signal_no_save_when_vm_primary_ip_has_multiple_assignments(self):
+        """
+        Test that VM is not saved when removing an IP from a VM interface and the IP is the VM's
+        primary IP but is assigned to other VM interfaces on that VM.
+        """
+        # Test IPv4 primary IP with multiple assignments
+        with self.subTest("IPv4 primary IP with multiple assignments"):
+            # Create primary IPv4 address
+            primary_ipv4_addr = IPAddress.objects.create(
+                address="192.0.2.1/24", status=self.status, namespace=self.namespace
+            )
+
+            # Assign primary IPv4 to multiple VM interfaces
+            assignment_ipv4_vmint1 = IPAddressToInterface.objects.create(
+                vm_interface=self.test_vmint1, ip_address=primary_ipv4_addr
+            )
+            IPAddressToInterface.objects.create(vm_interface=self.test_vmint2, ip_address=primary_ipv4_addr)
+
+            # Set it as the VM's primary IP
+            self.test_vm.primary_ip4 = primary_ipv4_addr
+            self.test_vm.save()
+
+            # Mock the VM save method to verify it's not called
+            with patch.object(self.test_vm, "save") as mock_save:
+                # Remove primary IP from one VM interface - should NOT trigger save since other assignment exists
+                assignment_ipv4_vmint1.delete()
+
+                # Assert save was not called since IP is still assigned to test_vmint2
+                mock_save.assert_not_called()
+
+        # Test IPv6 primary IP with multiple assignments
+        with self.subTest("IPv6 primary IP with multiple assignments"):
+            # Create IPv6 prefix first
+            Prefix.objects.create(prefix="2001:db8::/64", status=self.status, namespace=self.namespace)
+
+            # Create primary IPv6 address
+            primary_ipv6_addr = IPAddress.objects.create(
+                address="2001:db8::1/64", status=self.status, namespace=self.namespace
+            )
+
+            # Assign primary IPv6 to multiple VM interfaces
+            assignment_ipv6_vmint1 = IPAddressToInterface.objects.create(
+                vm_interface=self.test_vmint1, ip_address=primary_ipv6_addr
+            )
+            IPAddressToInterface.objects.create(vm_interface=self.test_vmint2, ip_address=primary_ipv6_addr)
+
+            # Set it as the VM's primary IP
+            self.test_vm.primary_ip6 = primary_ipv6_addr
+            self.test_vm.save()
+
+            # Mock the VM save method to verify it's not called
+            with patch.object(self.test_vm, "save") as mock_save:
+                # Remove primary IP from one VM interface - should NOT trigger save since other assignment exists
+                assignment_ipv6_vmint1.delete()
+
+                # Assert save was not called since IP is still assigned to test_vmint2
+                mock_save.assert_not_called()
+
 
 class TestVarbinaryIPField(TestCase):
     """Tests for `nautobot.ipam.fields.VarbinaryIPField`."""

nautobot/ipam/tests/test_querysets.py

@@ -802,6 +802,23 @@ class PrefixQuerysetTestCase(TestCase):
                 status=self.status,
             )
 
+        # same but for IPv6
+        container_v6 = netaddr.IPNetwork("2001:db8::/120")
+        Prefix.objects.create(
+            prefix=container_v6,
+            type=choices.PrefixTypeChoices.TYPE_CONTAINER,
+            namespace=namespace,
+            status=self.status,
+        )
+        for prefix_length in range(121, 129):
+            network = list(container_v6.subnet(prefix_length))[1]
+            Prefix.objects.create(
+                prefix=network,
+                type=choices.PrefixTypeChoices.TYPE_NETWORK,
+                namespace=namespace,
+                status=self.status,
+            )
+
         for last_octet in range(1, 255):
             ip = netaddr.IPAddress(f"10.0.0.{last_octet}")
             expected_prefix_length = 33 - len(bin(last_octet)[2:])  # [1] = 32, [2,3] = 31, [4,5,6,7] = 30, etc.
@@ -819,3 +836,32 @@ class PrefixQuerysetTestCase(TestCase):
                     .order_by("-prefix_length")
                     .first(),
                 )
+
+            ip = netaddr.IPAddress(f"2001:db8::{last_octet:x}")
+            expected_prefix_length = 129 - len(bin(last_octet)[2:])  # [1] = 128, [2,3] = 127, [4,5,6,7] = 126, etc.
+            with self.subTest(ip=ip, expected_prefix_length=expected_prefix_length):
+                closest_parent = Prefix.objects.filter(namespace=namespace).get_closest_parent(ip, include_self=True)
+                expected_parent = list(container_v6.subnet(expected_prefix_length))[1]
+                self.assertEqual(closest_parent.prefix, expected_parent)
+                self.assertEqual(
+                    closest_parent,
+                    Prefix.objects.filter(
+                        network__lte=ip.value,
+                        broadcast__gte=ip.value,
+                        namespace=namespace,
+                    )
+                    .order_by("-prefix_length")
+                    .first(),
+                )
+
+        slash32 = Prefix.objects.create(prefix="10.1.1.154/32", namespace=namespace, status=self.status)
+        slash31 = Prefix.objects.create(prefix="10.1.1.154/31", namespace=namespace, status=self.status)
+        slash30 = Prefix.objects.create(prefix="10.1.1.154/30", namespace=namespace, status=self.status)
+        self.assertEqual(slash31, Prefix.objects.get_closest_parent(slash32.prefix))
+        self.assertEqual(slash30, Prefix.objects.get_closest_parent(slash31.prefix))
+
+        slash126 = Prefix.objects.create(prefix="::1/126", namespace=namespace, status=self.status)
+        slash127 = Prefix.objects.create(prefix="::1/127", namespace=namespace, status=self.status)
+        slash128 = Prefix.objects.create(prefix="::1/128", namespace=namespace, status=self.status)
+        self.assertEqual(slash126, Prefix.objects.get_closest_parent(slash127.prefix))
+        self.assertEqual(slash127, Prefix.objects.get_closest_parent(slash128.prefix))

nautobot/ipam/tests/test_views.py

@@ -303,6 +303,8 @@ class PrefixTestCase(ViewTestCases.PrimaryObjectViewTestCase, ViewTestCases.List
             status=ip_status,
             namespace=self.namespace,
         )
+        # add this permission to can test button Add an IP Address
+        self.add_permissions("ipam.add_ipaddress")
         url = reverse("ipam:prefix_ipaddresses", args=(instance.pk,))
         response = self.client.get(url)
         self.assertHttpStatus(response, 200)
@@ -314,6 +316,35 @@ class PrefixTestCase(ViewTestCases.PrimaryObjectViewTestCase, ViewTestCases.List
             f'<li role="presentation" class="active"><a href="{url}">IP Addresses <span class="badge">2</span></a></li>'
         )
         self.assertInHTML(ip_address_tab, content)
+        # Checks if the button is in the content.
+        self.assertInHTML("""<span class="mdi mdi-plus-thick" aria-hidden="true"></span>Add an IP Address""", content)
+
+    @override_settings(EXEMPT_VIEW_PERMISSIONS=["*"])
+    def test_prefix_child_prefixes_table_list(self):
+        instance = Prefix.objects.create(
+            prefix="5.5.10.0/23",
+            namespace=self.namespace,
+            type=PrefixTypeChoices.TYPE_NETWORK,
+            status=self.statuses[1],
+        )
+        Prefix.objects.create(
+            prefix="5.5.10.0/30",
+            namespace=self.namespace,
+            type=PrefixTypeChoices.TYPE_POOL,
+            status=self.statuses[1],
+        )
+        # add this permission to can test button Add Child Prefix
+        self.add_permissions("ipam.add_prefix")
+        url = reverse("ipam:prefix_prefixes", args=(instance.pk,))
+        response = self.client.get(url)
+        self.assertHttpStatus(response, 200)
+        content = extract_page_body(response.content.decode(response.charset))
+        # This validates that both parent prefix and child prefix IPAddresses are present in parent prefix IPAddresses list
+        self.assertIn("5.5.10.0/30", strip_tags(content))
+        ip_address_tab = f'<li role="presentation" class="active"><a href="{url}">Child Prefixes <span class="badge">1</span></a></li>'
+        self.assertInHTML(ip_address_tab, content)
+        # Checks if the button is in the content.
+        self.assertInHTML("""<span class="mdi mdi-plus-thick" aria-hidden="true"></span>Add Child Prefix""", content)
 
 
 class IPAddressTestCase(ViewTestCases.PrimaryObjectViewTestCase):
@@ -1036,6 +1067,9 @@ class VLANTestCase(ViewTestCases.PrimaryObjectViewTestCase):
 
 class ServiceTestCase(ViewTestCases.PrimaryObjectViewTestCase):
     model = Service
+    allowed_number_of_tree_queries_per_view_type = {
+        "retrieve": 1,
+    }
 
     @classmethod
     def setUpTestData(cls):