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

nautobot/extras/tests/test_plugins.py

@@ -10,6 +10,7 @@ from django.urls import NoReverseMatch, reverse
 import netaddr
 
 from nautobot.circuits.models import Circuit, CircuitType, Provider
+from nautobot.core.celery import app
 from nautobot.core.testing import APIViewTestCases, TestCase, ViewTestCases, extract_page_body
 from nautobot.dcim.models import Device, DeviceType, Manufacturer, Location, LocationType
 from nautobot.dcim.tests.test_views import create_test_device
@@ -17,7 +18,7 @@ from nautobot.tenancy.models import Tenant, TenantGroup
 from nautobot.tenancy.filters import TenantFilterSet
 from nautobot.tenancy.forms import TenantFilterForm
 from nautobot.extras.choices import CustomFieldTypeChoices, RelationshipTypeChoices
-from nautobot.extras.jobs import get_job, get_job_classpaths, get_jobs
+from nautobot.extras.jobs import get_job
 from nautobot.extras.models import CustomField, Relationship, RelationshipAssociation, Role, Secret, Status
 from nautobot.extras.plugins.exceptions import PluginImproperlyConfigured
 from nautobot.extras.plugins.utils import load_plugin
@@ -119,31 +120,8 @@ class PluginTest(TestCase):
         from example_plugin.jobs import ExampleJob
 
         self.assertIn(ExampleJob, registry.get("plugin_jobs", []))
-
-        self.assertEqual(
-            ExampleJob,
-            get_job("plugins/example_plugin.jobs/ExampleJob"),
-        )
-        self.assertIn(
-            "plugins/example_plugin.jobs/ExampleJob",
-            get_job_classpaths(),
-        )
-        jobs_dict = get_jobs()
-        self.assertIn("plugins", jobs_dict)
-        self.assertIn("example_plugin.jobs", jobs_dict["plugins"])
-        self.assertEqual(
-            "ExamplePlugin jobs",
-            jobs_dict["plugins"]["example_plugin.jobs"].get("name"),
-        )
-        self.assertIn("jobs", jobs_dict["plugins"]["example_plugin.jobs"])
-        self.assertIn(
-            "ExampleJob",
-            jobs_dict["plugins"]["example_plugin.jobs"]["jobs"],
-        )
-        self.assertEqual(
-            ExampleJob,
-            jobs_dict["plugins"]["example_plugin.jobs"]["jobs"]["ExampleJob"],
-        )
+        self.assertEqual(ExampleJob, get_job("example_plugin.jobs.ExampleJob"))
+        self.assertIn("example_plugin.jobs.ExampleJob", app.tasks)
 
     def test_git_datasource_contents_registration(self):
         """
@@ -186,20 +164,6 @@ class PluginTest(TestCase):
         with self.assertRaises(PluginImproperlyConfigured):
             ExampleConfigWithMiddleware.validate({}, settings.VERSION)
 
-    def test_caching_config(self):
-        """
-        Check that plugin caching configuration is registered and valid.
-        """
-        self.assertIn("example_plugin.*", settings.CACHEOPS)
-
-        # Establish example config to have invalid cache_config (list)
-        class ExampleConfigWithBadCacheConfig(example_config):
-            caching_config = []
-
-        # Validation should fail when a caching_config is not a dict
-        with self.assertRaises(PluginImproperlyConfigured):
-            ExampleConfigWithBadCacheConfig.validate({}, settings.VERSION)
-
     def test_min_version(self):
         """
         Check enforcement of minimum Nautobot version.
@@ -286,14 +250,9 @@ class PluginTest(TestCase):
         """
         Validate that example plugin is adding new items to `registry["nav_menu"]`.
         """
-        self.assertTrue(registry["nav_menu"]["tabs"].get("Example Menu"))
-        self.assertTrue(registry["nav_menu"]["tabs"]["Example Menu"]["groups"].get("Example Group 1"))
+        self.assertIn("Example App", registry["nav_menu"]["Inventory"]["groups"])
         # Modified this statement since we are passing the url into registry directly instead of the reverse url string
-        self.assertTrue(
-            registry["nav_menu"]["tabs"]["Example Menu"]["groups"]["Example Group 1"]["items"].get(
-                "/plugins/example-plugin/models/"
-            )
-        )
+        self.assertIn("Example Model", registry["nav_menu"]["Inventory"]["groups"]["Example App"]["items"])
 
     def test_nautobot_database_ready_signal(self):
         """
@@ -404,7 +363,6 @@ class PluginDetailViewTest(TestCase):
 )
 class PluginAPITest(APIViewTestCases.APIViewTestCase):
     model = ExampleModel
-    brief_fields = ["display", "id", "name", "url"]
     bulk_update_data = {
         "number": 2600,
     }
@@ -443,7 +401,7 @@ class PluginAPITest(APIViewTestCases.APIViewTestCase):
         self.assertEqual(detail_url, self._get_detail_url(instance))
 
     @override_settings(EXEMPT_VIEW_PERMISSIONS=[])
-    @mock.patch("nautobot.extras.api.serializers.reverse")
+    @mock.patch("nautobot.core.api.serializers.reverse")
     def test_get_notes_url_on_object_raise_no_reverse_match(self, mock_reverse):
         # This test is to check the error handling when the user implemented a model without NotesMixin
         # But include the NotesSerializerMixin in the Model APIViewSet with NautobotModelViewSet
@@ -462,7 +420,7 @@ class PluginAPITest(APIViewTestCases.APIViewTestCase):
         url = self._get_detail_url(instance)
 
         # Check if log messages and API data is properly rendered
-        with self.assertLogs(logger="nautobot.extras.api", level="WARNING") as cm:
+        with self.assertLogs(logger="nautobot.core.api", level="WARNING") as cm:
             response = self.client.get(url, **self.header)
             self.assertHttpStatus(response, 200)
             self.assertIn("notes_url", response.data)
@@ -473,12 +431,21 @@ class PluginAPITest(APIViewTestCases.APIViewTestCase):
                     f"Notes feature is not available for model {model_name}. "
                     "Please make sure to: "
                     f"1. Include NotesMixin from nautobot.extras.model.mixins in the {model_name} class definition "
-                    f"2. Include NotesViewSetMixin from nautobot.extras.api.mixins in the {model_name}ViewSet "
+                    f"2. Include NotesViewSetMixin from nautobot.extras.api.views in the {model_name}ViewSet "
                     "before including NotesSerializerMixin in the model serializer"
                 ),
                 cm.output[0],
             )
 
+    # TODO: Unskip after resolving #2908, #2909
+    @skip("DRF's built-in OrderingFilter triggering natural key attribute error in our base")
+    def test_list_objects_ascending_ordered(self):
+        pass
+
+    @skip("DRF's built-in OrderingFilter triggering natural key attribute error in our base")
+    def test_list_objects_descending_ordered(self):
+        pass
+
 
 @skipIf(
     "example_plugin" not in settings.PLUGINS,
@@ -499,15 +466,18 @@ class PluginCustomValidationTest(TestCase):
         with self.assertRaises(ValidationError):
             location.clean()
 
-    @skip
     def test_relationship_association_validator_raises_exception(self):
-        status = Status.objects.get_for_model(IPAddress).first()
         namespace = Namespace.objects.first()
-        prefix = Prefix.objects.create(prefix=netaddr.IPNetwork("192.168.22.0/24"), namespace=namespace)
+        prefix_status = Status.objects.get_for_model(Prefix).first()
+        prefix = Prefix.objects.create(
+            prefix=netaddr.IPNetwork("192.168.10.0/24"), status=prefix_status, namespace=namespace
+        )
+        Prefix.objects.create(prefix=netaddr.IPNetwork("192.168.22.0/24"), status=prefix_status, namespace=namespace)
+        status = Status.objects.get_for_model(IPAddress).first()
         ipaddress = IPAddress.objects.create(address="192.168.22.1/24", status=status, namespace=namespace)
         relationship = Relationship.objects.create(
-            name="Test Relationship",
-            slug="test-relationship",
+            label="Test Relationship",
+            key="test_relationship",
             source_type=ContentType.objects.get_for_model(Prefix),
             destination_type=ContentType.objects.get_for_model(IPAddress),
             type=RelationshipTypeChoices.TYPE_ONE_TO_MANY,
@@ -534,31 +504,35 @@ class FilterExtensionTest(TestCase):
             Tenant.objects.create(name="Tenant 3", tenant_group=tenant_groups[2], description="tenant-3.nautobot.com"),
         )
         location_type = LocationType.objects.get(name="Campus")
+        location_status = Status.objects.get_for_model(Location).first()
         Location.objects.create(
             name="Location 1",
             slug="location-1",
             tenant=tenants[0],
             location_type=location_type,
+            status=location_status,
         )
         Location.objects.create(
             name="Location 2",
             slug="location-2",
             tenant=tenants[1],
             location_type=location_type,
+            status=location_status,
         )
         Location.objects.create(
             name="Location 3",
             slug="location-3",
             tenant=tenants[2],
             location_type=location_type,
+            status=location_status,
         )
 
-        manufactures = Manufacturer.objects.all()[:3]
+        manufacturers = Manufacturer.objects.all()[:3]
 
         roles = Role.objects.get_for_model(Device)
 
         DeviceType.objects.create(
-            manufacturer=manufactures[0],
+            manufacturer=manufacturers[0],
             model="Model 1",
             slug="model-1",
             part_number="Part Number 1",
@@ -566,7 +540,7 @@ class FilterExtensionTest(TestCase):
             is_full_depth=True,
         )
         DeviceType.objects.create(
-            manufacturer=manufactures[1],
+            manufacturer=manufacturers[1],
             model="Model 2",
             slug="model-2",
             part_number="Part Number 2",
@@ -574,7 +548,7 @@ class FilterExtensionTest(TestCase):
             is_full_depth=True,
         )
         DeviceType.objects.create(
-            manufacturer=manufactures[2],
+            manufacturer=manufacturers[2],
             model="Model 3",
             slug="model-3",
             part_number="Part Number 3",
@@ -582,12 +556,14 @@ class FilterExtensionTest(TestCase):
             is_full_depth=False,
         )
 
+        device_status = Status.objects.get_for_model(Device).first()
         Device.objects.create(
             name="Device 1",
             device_type=DeviceType.objects.get(slug="model-1"),
             role=roles[0],
             tenant=tenants[0],
             location=Location.objects.get(slug="location-1"),
+            status=device_status,
         )
         Device.objects.create(
             name="Device 2",
@@ -595,6 +571,7 @@ class FilterExtensionTest(TestCase):
             role=roles[1],
             tenant=tenants[1],
             location=Location.objects.get(slug="location-2"),
+            status=device_status,
         )
         Device.objects.create(
             name="Device 3",
@@ -602,6 +579,7 @@ class FilterExtensionTest(TestCase):
             role=roles[3],
             tenant=tenants[2],
             location=Location.objects.get(slug="location-3"),
+            status=device_status,
         )
 
     def test_basic_custom_filter(self):