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

nautobot/core/tests/test_filters.py

@@ -4,8 +4,9 @@ from django.contrib.contenttypes.models import ContentType
 from django.db import models as django_models
 from django.shortcuts import reverse
 from django.test import TestCase
+from django.test.utils import isolate_apps
+
 import django_filters
-from taggit.managers import TaggableManager
 from tree_queries.models import TreeNodeForeignKey
 
 from nautobot.core import filters, testing
@@ -33,50 +34,67 @@ class TreeNodeMultipleChoiceFilterTest(TestCase):
         super().setUp()
 
         self.location_type = dcim_models.LocationType.objects.get(name="Campus")
+        status = extras_models.Status.objects.get_for_model(dcim_models.Location).first()
         self.parent_location_1 = dcim_models.Location.objects.create(
-            name="Test Parent Location 1", slug="test-parent-location-1", location_type=self.location_type
+            name="Test Parent Location 1",
+            slug="test-parent-location-1",
+            location_type=self.location_type,
+            status=status,
         )
         self.parent_location_2 = dcim_models.Location.objects.create(
-            name="Test Parent Location 2", slug="test-parent-location-2", location_type=self.location_type
+            name="Test Parent Location 2",
+            slug="test-parent-location-2",
+            location_type=self.location_type,
+            status=status,
         )
         self.parent_location_2a = dcim_models.Location.objects.create(
             name="Test Parent Location 2A",
             slug="test-parent-location-2a",
             parent=self.parent_location_2,
             location_type=self.location_type,
+            status=status,
         )
         self.parent_location_2ab = dcim_models.Location.objects.create(
             name="Test Parent Location 2A-B",
             slug="test-parent-location-2a-b",
             parent=self.parent_location_2a,
             location_type=self.location_type,
+            status=status,
         )
         self.child_location_1 = dcim_models.Location.objects.create(
             parent=self.parent_location_1,
             name="Test Child Location 1",
             slug="test-child-location-1",
             location_type=self.location_type,
+            status=status,
         )
         self.child_location_2 = dcim_models.Location.objects.create(
             parent=self.parent_location_2,
             name="Test Child Location 2",
             slug="test-child-location-2",
             location_type=self.location_type,
+            status=status,
         )
         self.child_location_2a = dcim_models.Location.objects.create(
             parent=self.parent_location_2a,
             name="Test Child Location 2a",
             slug="test-child-location-2a",
             location_type=self.location_type,
+            status=status,
         )
         self.child_location_2ab = dcim_models.Location.objects.create(
             parent=self.parent_location_2ab,
             name="Test Child Location 2a-b",
             slug="test-child-location-2a-b",
             location_type=self.location_type,
+            status=status,
         )
         self.child_location_0 = dcim_models.Location.objects.create(
-            parent=None, name="Test Child Location 0", slug="test-child-location0", location_type=self.location_type
+            parent=None,
+            name="Test Child Location 0",
+            slug="test-child-location0",
+            location_type=self.location_type,
+            status=status,
         )
         self.queryset = dcim_models.Location.objects.filter(name__icontains="Test Child Location")
 
@@ -336,35 +354,37 @@ class NaturalKeyOrPKMultipleChoiceFilterTest(TestCase, testing.NautobotTestCaseM
         )
 
 
-class TestModel(django_models.Model):
-    """
-    Test model used by BaseFilterSetTest for filter validation. Should never appear in a schema migration.
-    """
-
-    charfield = django_models.CharField(max_length=10)
-    choicefield = django_models.IntegerField(choices=(("A", 1), ("B", 2), ("C", 3)))
-    charchoicefield = django_models.CharField(
-        choices=(("1", "Option 1"), ("2", "Option 2"), ("3", "Option 3")), max_length=10
-    )
-    datefield = django_models.DateField()
-    datetimefield = django_models.DateTimeField()
-    decimalfield = django_models.DecimalField(max_digits=9, decimal_places=6)
-    floatfield = django_models.FloatField()
-    integerfield = django_models.IntegerField()
-    macaddressfield = core_fields.MACAddressCharField()
-    textfield = django_models.TextField()
-    timefield = django_models.TimeField()
-    treeforeignkeyfield = TreeNodeForeignKey(to="self", on_delete=django_models.CASCADE)
-
-    tags = TaggableManager(through=extras_models.TaggedItem)
-
-
+@isolate_apps("nautobot.core.tests")
 class BaseFilterSetTest(TestCase):
     """
     Ensure that a BaseFilterSet automatically creates the expected set of filters for each filter type.
     """
 
     class TestFilterSet(filters.BaseFilterSet):
+        """Filterset for testing various fields."""
+
+        class TestModel(django_models.Model):
+            """
+            Test model used by BaseFilterSetTest for filter validation. Should never appear in a schema migration.
+            """
+
+            charfield = django_models.CharField(max_length=10)
+            choicefield = django_models.IntegerField(choices=(("A", 1), ("B", 2), ("C", 3)))
+            charchoicefield = django_models.CharField(
+                choices=(("1", "Option 1"), ("2", "Option 2"), ("3", "Option 3")), max_length=10
+            )
+            datefield = django_models.DateField()
+            datetimefield = django_models.DateTimeField()
+            decimalfield = django_models.DecimalField(max_digits=9, decimal_places=6)
+            floatfield = django_models.FloatField()
+            integerfield = django_models.IntegerField()
+            macaddressfield = core_fields.MACAddressCharField()
+            textfield = django_models.TextField()
+            timefield = django_models.TimeField()
+            treeforeignkeyfield = TreeNodeForeignKey(to="self", on_delete=django_models.CASCADE)
+
+            tags = core_fields.TagsField()
+
         charfield = django_filters.CharFilter()
         macaddressfield = filters.MACAddressFilter()
         modelchoicefield = django_filters.ModelChoiceFilter(
@@ -379,26 +399,32 @@ class BaseFilterSetTest(TestCase):
         multivaluecharfield = filters.MultiValueCharFilter(field_name="charfield")
         treeforeignkeyfield = filters.TreeNodeMultipleChoiceFilter(queryset=TestModel.objects.all())
 
-        class Meta:
-            model = TestModel
-            fields = (
-                "charfield",
-                "choicefield",
-                "charchoicefield",
-                "datefield",
-                "datetimefield",
-                "decimalfield",
-                "floatfield",
-                "integerfield",
-                "macaddressfield",
-                "modelchoicefield",
-                "modelmultiplechoicefield",
-                "multiplechoicefield",
-                "tags",
-                "textfield",
-                "timefield",
-                "treeforeignkeyfield",
-            )
+        # declared this way because `class Meta: model = TestModel` gives a NameError otherwise.
+        Meta = type(
+            "Meta",
+            (),
+            {
+                "model": TestModel,
+                "fields": (
+                    "charfield",
+                    "choicefield",
+                    "charchoicefield",
+                    "datefield",
+                    "datetimefield",
+                    "decimalfield",
+                    "floatfield",
+                    "integerfield",
+                    "macaddressfield",
+                    "modelchoicefield",
+                    "modelmultiplechoicefield",
+                    "multiplechoicefield",
+                    "tags",
+                    "textfield",
+                    "timefield",
+                    "treeforeignkeyfield",
+                ),
+            },
+        )
 
     @classmethod
     def setUpTestData(cls):
@@ -791,10 +817,11 @@ class DynamicFilterLookupExpressionTest(TestCase):
         cls.locations[1].asn = 65101
         cls.locations[2].asn = 65201
 
+        rack_status = extras_models.Status.objects.get_for_model(dcim_models.Rack).first()
         racks = (
-            dcim_models.Rack(name="Rack 1", location=cls.locations[0]),
-            dcim_models.Rack(name="Rack 2", location=cls.locations[1]),
-            dcim_models.Rack(name="Rack 3", location=cls.locations[2]),
+            dcim_models.Rack(name="Rack 1", location=cls.locations[0], status=rack_status),
+            dcim_models.Rack(name="Rack 2", location=cls.locations[1], status=rack_status),
+            dcim_models.Rack(name="Rack 3", location=cls.locations[2], status=rack_status),
         )
         dcim_models.Rack.objects.bulk_create(racks)
 
@@ -845,13 +872,26 @@ class DynamicFilterLookupExpressionTest(TestCase):
         )
         dcim_models.Device.objects.bulk_create(devices)
 
+        intf_status = extras_models.Status.objects.get_for_model(dcim_models.Interface).first()
         interfaces = (
-            dcim_models.Interface(device=devices[0], name="Interface 1", mac_address="00-00-00-00-00-01"),
-            dcim_models.Interface(device=devices[0], name="Interface 2", mac_address="aa-00-00-00-00-01"),
-            dcim_models.Interface(device=devices[1], name="Interface 3", mac_address="00-00-00-00-00-02"),
-            dcim_models.Interface(device=devices[1], name="Interface 4", mac_address="bb-00-00-00-00-02"),
-            dcim_models.Interface(device=devices[2], name="Interface 5", mac_address="00-00-00-00-00-03"),
-            dcim_models.Interface(device=devices[2], name="Interface 6", mac_address="cc-00-00-00-00-03"),
+            dcim_models.Interface(
+                device=devices[0], name="Interface 1", mac_address="00-00-00-00-00-01", status=intf_status
+            ),
+            dcim_models.Interface(
+                device=devices[0], name="Interface 2", mac_address="aa-00-00-00-00-01", status=intf_status
+            ),
+            dcim_models.Interface(
+                device=devices[1], name="Interface 3", mac_address="00-00-00-00-00-02", status=intf_status
+            ),
+            dcim_models.Interface(
+                device=devices[1], name="Interface 4", mac_address="bb-00-00-00-00-02", status=intf_status
+            ),
+            dcim_models.Interface(
+                device=devices[2], name="Interface 5", mac_address="00-00-00-00-00-03", status=intf_status
+            ),
+            dcim_models.Interface(
+                device=devices[2], name="Interface 6", mac_address="cc-00-00-00-00-03", status=intf_status
+            ),
         )
         dcim_models.Interface.objects.bulk_create(interfaces)
 
@@ -1168,7 +1208,10 @@ class GetFiltersetTestValuesTest(testing.FilterTestCases.BaseFilterTestCase):
     def test_insufficient_unique_values(self):
         location_type = dcim_models.LocationType.objects.get(name="Campus")
         dcim_models.Location.objects.create(
-            name="getfiltersettestUniqueLocation", description="UniqueLocation description", location_type=location_type
+            name="getfiltersettestUniqueLocation",
+            description="UniqueLocation description",
+            location_type=location_type,
+            status=self.status,
         )
         with self.assertRaisesMessage(ValueError, self.exception_message):
             self.get_filterset_test_values("description")
@@ -1201,11 +1244,12 @@ class SearchFilterTest(TestCase, testing.NautobotTestCaseMixin):
     def setUp(self):
         super().setUp()
         self.lt = dcim_models.LocationType.objects.get(name="Campus")
+        status = extras_models.Status.objects.get_for_model(dcim_models.Location).first()
         self.parent_location_1 = dcim_models.Location.objects.create(
-            name="Test Parent Location 1", slug="test-parent-location-1", location_type=self.lt
+            name="Test Parent Location 1", slug="test-parent-location-1", location_type=self.lt, status=status
         )
         self.parent_location_2 = dcim_models.Location.objects.create(
-            name="Test Parent Location 2", slug="test-parent-location-2", location_type=self.lt
+            name="Test Parent Location 2", slug="test-parent-location-2", location_type=self.lt, status=status
         )
         self.child_location_1 = dcim_models.Location.objects.create(
             parent=self.parent_location_1,
@@ -1213,6 +1257,7 @@ class SearchFilterTest(TestCase, testing.NautobotTestCaseMixin):
             slug="test-child-location1",
             location_type=self.lt,
             asn=1234,
+            status=status,
         )
         self.child_location_2 = dcim_models.Location.objects.create(
             parent=self.parent_location_2,
@@ -1220,18 +1265,21 @@ class SearchFilterTest(TestCase, testing.NautobotTestCaseMixin):
             slug="test-child-location2",
             location_type=self.lt,
             asn=12345,
+            status=status,
         )
         self.child_location_3 = dcim_models.Location.objects.create(
             parent=None,
             name="Test Child Location 3",
             slug="test-child-location3",
             location_type=self.lt,
+            status=status,
         )
         self.child_location_4 = dcim_models.Location.objects.create(
             parent=None,
             name="Test Child Location4",
             slug="test-child-location4",
             location_type=self.lt,
+            status=status,
         )
         self.queryset = dcim_models.Location.objects.all()
 

nautobot/core/tests/test_forms.py

@@ -376,87 +376,6 @@ class AddFieldToFormClassTest(TestCase):
             )
 
 
-class CSVDataFieldTest(TestCase):
-    def setUp(self):
-        self.field = forms.CSVDataField(from_form=ipam_forms.IPAddressCSVForm)
-
-    def test_clean(self):
-        input_ = """
-        address,status,vrf
-        192.0.2.1/32,Active,Test VRF
-        """
-        output = (
-            {"address": None, "status": None, "vrf": None},
-            [{"address": "192.0.2.1/32", "status": "Active", "vrf": "Test VRF"}],
-        )
-        self.assertEqual(self.field.clean(input_), output)
-
-    def test_clean_invalid_header(self):
-        input_ = """
-        address,status,vrf,xxx
-        192.0.2.1/32,Active,Test VRF,123
-        """
-        with self.assertRaises(django_forms.ValidationError):
-            self.field.clean(input_)
-
-    def test_clean_missing_required_header(self):
-        input_ = """
-        status,vrf
-        Active,Test VRF
-        """
-        with self.assertRaises(django_forms.ValidationError):
-            self.field.clean(input_)
-
-    def test_clean_default_to_field(self):
-        input_ = """
-        address,status,vrf.name
-        192.0.2.1/32,Active,Test VRF
-        """
-        output = (
-            {"address": None, "status": None, "vrf": "name"},
-            [{"address": "192.0.2.1/32", "status": "Active", "vrf": "Test VRF"}],
-        )
-        self.assertEqual(self.field.clean(input_), output)
-
-    def test_clean_pk_to_field(self):
-        input_ = """
-        address,status,vrf.pk
-        192.0.2.1/32,Active,123
-        """
-        output = (
-            {"address": None, "status": None, "vrf": "pk"},
-            [{"address": "192.0.2.1/32", "status": "Active", "vrf": "123"}],
-        )
-        self.assertEqual(self.field.clean(input_), output)
-
-    def test_clean_custom_to_field(self):
-        input_ = """
-        address,status,vrf.rd
-        192.0.2.1/32,Active,123:456
-        """
-        output = (
-            {"address": None, "status": None, "vrf": "rd"},
-            [{"address": "192.0.2.1/32", "status": "Active", "vrf": "123:456"}],
-        )
-        self.assertEqual(self.field.clean(input_), output)
-
-    def test_clean_invalid_to_field(self):
-        input_ = """
-        address,status,vrf.xxx
-        192.0.2.1/32,Active,123:456
-        """
-        with self.assertRaises(django_forms.ValidationError):
-            self.field.clean(input_)
-
-    def test_clean_to_field_on_non_object(self):
-        input_ = """
-        address,status.foo,vrf
-        192.0.2.1/32,Bar,Test VRF
-        """
-        with self.assertRaises(django_forms.ValidationError):
-            self.field.clean(input_)
-
-
 class DynamicModelChoiceFieldTest(TestCase):
     """Tests for DynamicModelChoiceField."""
 
@@ -474,17 +393,21 @@ class DynamicModelChoiceFieldTest(TestCase):
 
     def test_prepare_value_valid_uuid(self):
         """A UUID PK referring to an actual object should be handled correctly."""
+        ipaddr_status = extras_models.Status.objects.get_for_model(ipam_models.IPAddress).first()
+        prefix_status = extras_models.Status.objects.get_for_model(ipam_models.Prefix).first()
         namespace = ipam_models.Namespace.objects.first()
-        ipam_models.Prefix.objects.create(prefix="10.1.1.0/24", namespace=namespace)
-        address = ipam_models.IPAddress.objects.create(address="10.1.1.1/24", namespace=namespace)
+        ipam_models.Prefix.objects.create(prefix="10.1.1.0/24", namespace=namespace, status=prefix_status)
+        address = ipam_models.IPAddress.objects.create(address="10.1.1.1/24", namespace=namespace, status=ipaddr_status)
         self.assertEqual(self.field.prepare_value(address.pk), address.pk)
         self.assertEqual(self.field_with_to_field_name.prepare_value(address.pk), address.address)
 
     def test_prepare_value_valid_object(self):
         """An object reference should be handled correctly."""
+        ipaddr_status = extras_models.Status.objects.get_for_model(ipam_models.IPAddress).first()
+        prefix_status = extras_models.Status.objects.get_for_model(ipam_models.Prefix).first()
         namespace = ipam_models.Namespace.objects.first()
-        ipam_models.Prefix.objects.create(prefix="10.1.1.0/24", namespace=namespace)
-        address = ipam_models.IPAddress.objects.create(address="10.1.1.1/24", namespace=namespace)
+        ipam_models.Prefix.objects.create(prefix="10.1.1.0/24", namespace=namespace, status=prefix_status)
+        address = ipam_models.IPAddress.objects.create(address="10.1.1.1/24", namespace=namespace, status=ipaddr_status)
         self.assertEqual(self.field.prepare_value(address), address.pk)
         self.assertEqual(self.field_with_to_field_name.prepare_value(address), address.address)
 
@@ -506,10 +429,16 @@ class DynamicModelMultipleChoiceFieldTest(TestCase):
 
     def test_prepare_value_multiple_object(self):
         """A list of object values should be translated to a list of PKs."""
+        ipaddr_status = extras_models.Status.objects.get_for_model(ipam_models.IPAddress).first()
+        prefix_status = extras_models.Status.objects.get_for_model(ipam_models.Prefix).first()
         namespace = ipam_models.Namespace.objects.first()
-        ipam_models.Prefix.objects.create(prefix="10.1.1.0/24", namespace=namespace)
-        address_1 = ipam_models.IPAddress.objects.create(address="10.1.1.1/24", namespace=namespace)
-        address_2 = ipam_models.IPAddress.objects.create(address="10.1.1.2/24", namespace=namespace)
+        ipam_models.Prefix.objects.create(prefix="10.1.1.0/24", namespace=namespace, status=prefix_status)
+        address_1 = ipam_models.IPAddress.objects.create(
+            address="10.1.1.1/24", namespace=namespace, status=ipaddr_status
+        )
+        address_2 = ipam_models.IPAddress.objects.create(
+            address="10.1.1.2/24", namespace=namespace, status=ipaddr_status
+        )
         self.assertEqual(
             self.field.prepare_value([address_1, address_2]),
             [address_1.pk, address_2.pk],
@@ -582,13 +511,17 @@ class AddressFieldMixinTest(TestCase):
 
     def setUp(self):
         """Setting up shared variables for the AddressFieldMixin."""
+        ipaddr_status = extras_models.Status.objects.get_for_model(ipam_models.IPAddress).first()
+        prefix_status = extras_models.Status.objects.get_for_model(ipam_models.Prefix).first()
         self.namespace = ipam_models.Namespace.objects.first()
-        self.prefix = ipam_models.Prefix.objects.create(prefix="10.0.0.0/24", namespace=self.namespace)
-        self.ip = ipam_models.IPAddress.objects.create(address="10.0.0.1/32", namespace=self.namespace)
+        self.prefix = ipam_models.Prefix.objects.create(
+            prefix="10.0.0.0/24", namespace=self.namespace, status=prefix_status
+        )
+        self.ip = ipam_models.IPAddress.objects.create(
+            address="10.0.0.1/32", namespace=self.namespace, status=ipaddr_status
+        )
         self.initial = {"address": self.ip.address}
 
-    def test_address_initial(self):
-        """Ensure initial kwargs for address is passed in."""
         with mock.patch("nautobot.core.forms.forms.forms.ModelForm.__init__") as mock_init:
             ip_none = ipam_models.IPAddress()
             forms.AddressFieldMixin(initial=self.initial, instance=ip_none)
@@ -864,6 +797,7 @@ class DynamicFilterFormTest(TestCase):
                 form.fields["lookup_value"].widget.attrs,
                 {
                     "class": "form-control nautobot-select2-api lookup_value-input form-control",
+                    "data-depth": 0,
                     "data-multiple": 1,
                     "data-query-param-content_types": '["dcim.location"]',
                     "display-field": "display",