From 362be89ec5a7ed863a5aaa679712ca114b1ad5d0 Mon Sep 17 00:00:00 2001 From: Jason Novinger Date: Tue, 9 Sep 2025 09:45:16 -0500 Subject: [PATCH] Fixup ModuleTypeTestCase bulk import test to work with callback mechamisn --- netbox/dcim/tests/test_views.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/netbox/dcim/tests/test_views.py b/netbox/dcim/tests/test_views.py index b23f7e16d..8f33c3df1 100644 --- a/netbox/dcim/tests/test_views.py +++ b/netbox/dcim/tests/test_views.py @@ -1078,14 +1078,14 @@ class ModuleTypeTestCase(ViewTestCases.PrimaryObjectViewTestCase): 'dcim.add_modulebaytemplate', ) + def verify_module_type_profile(scenario_name): + # TODO: remove extra regression asserts once parent test supports testing all import fields + fan_module_type = ModuleType.objects.get(part_number='generic-fan') + fan_module_type_profile = ModuleTypeProfile.objects.get(name='Fan') + assert fan_module_type.profile == fan_module_type_profile + # run base test - super().test_bulk_import_objects_with_permission() - - # TODO: remove extra regression asserts once parent test supports testing all import fields - fan_module_type = ModuleType.objects.get(part_number='generic-fan') - fan_module_type_profile = ModuleTypeProfile.objects.get(name='Fan') - - assert fan_module_type.profile == fan_module_type_profile + super().test_bulk_import_objects_with_permission(post_import_callback=verify_module_type_profile) @override_settings(EXEMPT_VIEW_PERMISSIONS=['*'], EXEMPT_EXCLUDE_MODELS=[]) def test_bulk_import_objects_with_constrained_permission(self):