nautobot 2.4.21__py3-none-any.whl → 3.0.0a3__py3-none-any.whl

nautobot/extras/tests/test_plugins.py

@@ -5,7 +5,7 @@ from django.conf import settings
 from django.contrib.contenttypes.models import ContentType
 from django.core.exceptions import ValidationError
 from django.template import engines
-from django.test import override_settings
+from django.test import override_settings, tag
 from django.urls import NoReverseMatch, reverse
 import django_tables2 as tables
 import netaddr
@@ -13,6 +13,7 @@ import netaddr
 from nautobot.circuits.models import Circuit, CircuitType, Provider
 from nautobot.core.testing import APIViewTestCases, disable_warnings, extract_page_body, TestCase, ViewTestCases
 from nautobot.core.utils.lookup import get_table_for_model
+from nautobot.data_validation.custom_validators import BaseValidator
 from nautobot.dcim.models import Device, DeviceType, Location, LocationType, Manufacturer
 from nautobot.dcim.tests.test_views import create_test_device
 from nautobot.extras import plugins
@@ -31,11 +32,8 @@ from nautobot.tenancy.forms import TenantFilterForm
 from nautobot.tenancy.models import Tenant, TenantGroup
 from nautobot.users.models import ObjectPermission
 
-from example_app import config as example_config
-from example_app.datasources import refresh_git_text_files
-from example_app.models import ExampleModel
-
 
+@tag("example_app")
 class AppTest(TestCase):
     def test_config(self):
         self.assertIn(
@@ -45,6 +43,8 @@ class AppTest(TestCase):
 
     def test_models(self):
         # Test saving an instance
+        from example_app.models import ExampleModel
+
         instance = ExampleModel(name="Instance 1", number=100)
         instance.save()
         self.assertIsNotNone(instance.pk)
@@ -86,6 +86,13 @@ class AppTest(TestCase):
             registry["plugin_custom_validators"]["extras.relationshipassociation"],
         )
 
+        # Assert data-validation engine added one and only one validator for a given model.
+        # https://github.com/nautobot/nautobot/pull/7180
+        data_validation_validators = [
+            v for v in registry["plugin_custom_validators"]["dcim.location"] if issubclass(v, BaseValidator)
+        ]
+        self.assertEqual(1, len(data_validation_validators))
+
     def test_jinja_filter_registration(self):
         """
         Check that App custom Jinja filters are registered correctly.
@@ -126,6 +133,8 @@ class AppTest(TestCase):
         """
         Check that App DatasourceContents are registered.
         """
+        from example_app.datasources import refresh_git_text_files
+
         registered_datasources = registry.get("datasource_contents", {}).get("extras.gitrepository", [])
 
         app_datasource = DatasourceContent(
@@ -150,6 +159,8 @@ class AppTest(TestCase):
         """
         Check that App middleware is registered.
         """
+        from example_app import config as example_config
+
         self.assertIn(
             "example_app.middleware.ExampleMiddleware",
             settings.MIDDLEWARE,
@@ -167,6 +178,8 @@ class AppTest(TestCase):
         """
         Check enforcement of minimum Nautobot version.
         """
+        from example_app import config as example_config
+
         with self.assertRaises(PluginImproperlyConfigured):
             example_config.validate({}, "0.8")
 
@@ -174,6 +187,8 @@ class AppTest(TestCase):
         """
         Check enforcement of maximum Nautobot version.
         """
+        from example_app import config as example_config
+
         with self.assertRaises(PluginImproperlyConfigured):
             example_config.validate({}, "10.0")
 
@@ -181,6 +196,7 @@ class AppTest(TestCase):
         """
         Validate enforcement of required settings.
         """
+        from example_app import config as example_config
 
         class ExampleConfigWithRequiredSettings(example_config):
             required_settings = ["foo"]
@@ -202,6 +218,7 @@ class AppTest(TestCase):
         """
         Validate population of default config settings.
         """
+        from example_app import config as example_config
 
         class ExampleConfigWithDefaultSettings(example_config):
             default_settings = {
@@ -231,6 +248,8 @@ class AppTest(TestCase):
         """
         Validate that App installed Django apps and dependencies are are registered.
         """
+        from example_app import config as example_config
+
         self.assertIn(
             "example_app.ExampleAppConfig",
             settings.INSTALLED_APPS,
@@ -275,11 +294,13 @@ class AppTest(TestCase):
         self.assertEqual(secret.get_value(obj=secret), secret.parameters["constant"])
 
 
+@tag("example_app")
 class AppGenericViewTest(ViewTestCases.PrimaryObjectViewTestCase):
-    model = ExampleModel
-
     @classmethod
     def setUpTestData(cls):
+        from example_app.models import ExampleModel
+
+        cls.model = ExampleModel
         # Example objects to test.
         ExampleModel.objects.create(name="Example 1", number=1)
         ExampleModel.objects.create(name="Example 2", number=2)
@@ -303,6 +324,7 @@ class AppGenericViewTest(ViewTestCases.PrimaryObjectViewTestCase):
         }
 
 
+@tag("example_app")
 class AppListViewTest(TestCase):
     def test_list_plugins_anonymous(self):
         # Make the request as an unauthenticated user
@@ -374,6 +396,7 @@ class AppListViewTest(TestCase):
         )
 
 
+@tag("example_app")
 class PluginDetailViewTest(TestCase):
     def test_view_detail_anonymous(self):
         # Make the request as an unauthenticated user
@@ -405,8 +428,8 @@ class MarketplaceViewTest(TestCase):
         self.assertHttpStatus(response, 200)
 
 
+@tag("example_app")
 class AppAPITest(APIViewTestCases.APIViewTestCase):
-    model = ExampleModel
     bulk_update_data = {
         "number": 2600,
     }
@@ -428,6 +451,9 @@ class AppAPITest(APIViewTestCases.APIViewTestCase):
 
     @classmethod
     def setUpTestData(cls):
+        from example_app.models import ExampleModel
+
+        cls.model = ExampleModel
         # Example objects to test.
         ExampleModel.objects.create(name="Example 1", number=1)
         ExampleModel.objects.create(name="Example 2", number=2)
@@ -435,6 +461,8 @@ class AppAPITest(APIViewTestCases.APIViewTestCase):
 
     @override_settings(EXEMPT_VIEW_PERMISSIONS=["*"])
     def test_api_urls(self):
+        from example_app.models import ExampleModel
+
         # Test list URL resolution
         list_url = reverse("plugins-api:example_app-api:examplemodel-list")
         self.assertEqual(list_url, self._get_list_url())
@@ -501,6 +529,7 @@ class TestUserContextCustomValidator(CustomValidator):
         self.validation_error(f"TestUserContextCustomValidator: user is {self.context['user']}")
 
 
+@tag("example_app")
 class AppCustomValidationTest(TestCase):
     def setUp(self):
         # When creating a fresh test DB, wrapping model clean methods fails, which is normal.
@@ -563,13 +592,15 @@ class AppCustomValidationTest(TestCase):
             registry["plugin_custom_validators"]["dcim.locationtype"] = before
 
 
+@tag("example_app")
 class ExampleModelCustomActionViewTest(TestCase):
     """Test for custom action view `all_names` added to Example App"""
 
-    model = ExampleModel
-
     @classmethod
     def setUpTestData(cls):
+        from example_app.models import ExampleModel
+
+        cls.model = ExampleModel
         ExampleModel.objects.create(name="Example 1", number=100)
         ExampleModel.objects.create(name="Example 2", number=200)
         ExampleModel.objects.create(name="Example 3", number=300)
@@ -625,6 +656,7 @@ class ExampleModelCustomActionViewTest(TestCase):
             self.assertNotIn(example_model.name, response_body, msg=response_body)
 
 
+@tag("example_app")
 class FilterExtensionTest(TestCase):
     """
     Tests for adding filter extensions
@@ -874,6 +906,7 @@ class LoadPluginTest(TestCase):
         load_plugin(app_name, settings)
 
 
+@tag("example_app", "example_app_with_view_override")
 class TestAppCoreViewOverrides(TestCase):
     """
     Validate that overridden core views work as expected.
@@ -908,6 +941,7 @@ class TestAppCoreViewOverrides(TestCase):
         )
 
 
+@tag("example_app")
 class PluginTemplateExtensionsTest(TestCase):
     """
     Test that registered TemplateExtensions inject content as expected
@@ -932,14 +966,14 @@ class PluginTemplateExtensionsTest(TestCase):
     def test_detail_view_left_page(self):
         response = self.client.get(reverse("dcim:location", kwargs={"pk": self.location.pk}))
         response_body = extract_page_body(response.content.decode(response.charset))
-        self.assertIn("App Injected Content - Left", response_body, msg=response_body)
+        self.assertIn("APP INJECTED CONTENT - LEFT", response_body, msg=response_body)
 
     def test_detail_view_right_page(self):
         response = self.client.get(reverse("dcim:location", kwargs={"pk": self.location.pk}))
         response_body = extract_page_body(response.content.decode(response.charset))
-        self.assertIn("App Injected Content - Right", response_body, msg=response_body)
+        self.assertIn("APP INJECTED CONTENT - RIGHT", response_body, msg=response_body)
 
     def test_detail_view_full_width_page(self):
         response = self.client.get(reverse("dcim:location", kwargs={"pk": self.location.pk}))
         response_body = extract_page_body(response.content.decode(response.charset))
-        self.assertIn("App Injected Content - Full Width", response_body, msg=response_body)
+        self.assertIn("APP INJECTED CONTENT - FULL WIDTH", response_body, msg=response_body)

nautobot/extras/tests/test_relationships.py

@@ -17,6 +17,7 @@ from nautobot.core.forms import (
 from nautobot.core.tables import RelationshipColumn
 from nautobot.core.testing import create_job_result_and_run_job, TestCase, TransactionTestCase
 from nautobot.core.testing.models import ModelTestCases
+from nautobot.core.utils.cache import construct_cache_key
 from nautobot.core.utils.lookup import get_route_for_model
 from nautobot.dcim.forms import DeviceForm
 from nautobot.dcim.models import (
@@ -185,8 +186,12 @@ class RelationshipBaseTest:
     def tearDown(self):
         """Ensure that relationship caches are cleared to avoid leakage into other tests."""
         with contextlib.suppress(redis.exceptions.ConnectionError):
-            cache.delete_pattern(f"{Relationship.objects.get_for_model_source.cache_key_prefix}.*")
-            cache.delete_pattern(f"{Relationship.objects.get_for_model_destination.cache_key_prefix}.*")
+            cache.delete_pattern(
+                f"{construct_cache_key(Relationship.objects, method_name='get_for_model_source', branch_aware=True)}(*)"
+            )
+            cache.delete_pattern(
+                f"{construct_cache_key(Relationship.objects, method_name='get_for_model_destination', branch_aware=True)}(*)"
+            )
 
 
 class RelationshipTest(RelationshipBaseTest, ModelTestCases.BaseModelTestCase):

nautobot/extras/tests/test_utils.py

@@ -35,10 +35,12 @@ class UtilsTestCase(TestCase):
     @mock.patch("celery.app.control.Inspect.active_queues")
     def test_get_celery_queues(self, mock_active_queues):
         with self.subTest("No queues"):
+            cache.clear()
             mock_active_queues.return_value = None
             self.assertDictEqual(get_celery_queues(), {})
 
         with self.subTest("1 worker 1 queue"):
+            cache.clear()
             mock_active_queues.return_value = {"celery@worker": [{"name": "queue1"}]}
             self.assertDictEqual(get_celery_queues(), {"queue1": 1})