From f2a509e0844df2cce99281fb196509a2a86f8da2 Mon Sep 17 00:00:00 2001 From: Brian Tiemann Date: Tue, 19 Nov 2024 14:54:54 -0500 Subject: [PATCH 1/2] Add config tests --- netbox_branching/tests/test_config.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 netbox_branching/tests/test_config.py diff --git a/netbox_branching/tests/test_config.py b/netbox_branching/tests/test_config.py new file mode 100644 index 0000000..37b9775 --- /dev/null +++ b/netbox_branching/tests/test_config.py @@ -0,0 +1,27 @@ +from django.test import TransactionTestCase, override_settings + +from netbox.registry import registry +from netbox_branching.models import Branch +from netbox_branching.utilities import register_models + + +class ConfigTestCase(TransactionTestCase): + serialized_rollback = True + + @override_settings(PLUGINS_CONFIG={ + 'netbox_branching': { + 'exempt_models': ['ipam.prefix'], + } + }) + def test_exempt_models(self): + register_models() + self.assertNotIn('prefix', registry['model_features']['branching']['ipam']) + + @override_settings(PLUGINS_CONFIG={ + 'netbox_branching': { + 'schema_prefix': 'dummy_', + } + }) + def test_schema_prefix(self): + branch = Branch(name='Branch 5') + self.assertEqual(branch.schema_name, f'dummy_{branch.schema_id}') \ No newline at end of file From 319fec8c28735fd12bf2a240c5e26043353da96f Mon Sep 17 00:00:00 2001 From: Brian Tiemann Date: Tue, 19 Nov 2024 15:03:41 -0500 Subject: [PATCH 2/2] Add newline --- netbox_branching/tests/test_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/netbox_branching/tests/test_config.py b/netbox_branching/tests/test_config.py index 37b9775..b78f152 100644 --- a/netbox_branching/tests/test_config.py +++ b/netbox_branching/tests/test_config.py @@ -24,4 +24,4 @@ def test_exempt_models(self): }) def test_schema_prefix(self): branch = Branch(name='Branch 5') - self.assertEqual(branch.schema_name, f'dummy_{branch.schema_id}') \ No newline at end of file + self.assertEqual(branch.schema_name, f'dummy_{branch.schema_id}')