ipfabric_netbox 4.3.0b5__py3-none-any.whl → 4.3.0b7__py3-none-any.whl

ipfabric_netbox/tests/test_views.py

@@ -1,4 +1,5 @@
 import random
+from datetime import timedelta
 from unittest.mock import patch
 from uuid import uuid4
 
@@ -63,8 +64,8 @@ class IPFabricSourceTestCase(
     ViewTestCases.DeleteObjectViewTestCase,
     ViewTestCases.ListObjectsViewTestCase,
     ViewTestCases.BulkDeleteObjectsViewTestCase,
-    # ViewTestCases.BulkRenameObjectsViewTestCase,
-    # ViewTestCases.BulkEditObjectsViewTestCase,
+    ViewTestCases.BulkRenameObjectsViewTestCase,
+    ViewTestCases.BulkEditObjectsViewTestCase,
     # ViewTestCases.BulkImportObjectsViewTestCase,
 ):
     model = IPFabricSource
@@ -84,7 +85,7 @@ class IPFabricSourceTestCase(
             ),
             IPFabricSource(
                 name="IP Fabric Source 2",
-                type=IPFabricSourceTypeChoices.REMOTE,
+                type=IPFabricSourceTypeChoices.LOCAL,
                 url="https://ipfabric2.example.com",
                 status=DataSourceStatusChoices.COMPLETED,
                 parameters={"auth": "token2", "verify": False, "timeout": 60},
@@ -488,7 +489,6 @@ class IPFabricSnapshotTestCase(
         self.assertTrue(response_snapshot.ipf_data.exists())
 
 
-# TODO: Get back to these tests
 class IPFabricDataTestCase(
     PluginPathMixin,
     # ViewTestCases.GetObjectViewTestCase,
@@ -607,12 +607,12 @@ class IPFabricSyncTestCase(
     ViewTestCases.DeleteObjectViewTestCase,
     ViewTestCases.ListObjectsViewTestCase,
     ViewTestCases.BulkDeleteObjectsViewTestCase,
-    # ViewTestCases.BulkRenameObjectsViewTestCase,
-    # ViewTestCases.BulkEditObjectsViewTestCase,
+    ViewTestCases.BulkRenameObjectsViewTestCase,
+    ViewTestCases.BulkEditObjectsViewTestCase,
     # ViewTestCases.BulkImportObjectsViewTestCase,
 ):
     model = IPFabricSync
-    user_permissions = ("ipfabric_netbox.start_ipfabricsync",)
+    user_permissions = ("ipfabric_netbox.sync_ipfabricsync",)
 
     @classmethod
     def setUpTestData(cls):
@@ -657,6 +657,8 @@ class IPFabricSyncTestCase(
                 status=DataSourceStatusChoices.NEW,
                 parameters=get_parameters(),
                 last_synced=timezone.now(),
+                scheduled=timezone.now() + timedelta(hours=6),
+                interval=123456,
             ),
             IPFabricSync(
                 name="Sync Job 2",
@@ -839,9 +841,9 @@ class IPFabricTransformMapGroupTestCase(
     ViewTestCases.DeleteObjectViewTestCase,
     ViewTestCases.ListObjectsViewTestCase,
     ViewTestCases.BulkDeleteObjectsViewTestCase,
-    # ViewTestCases.BulkRenameObjectsViewTestCase,
-    # ViewTestCases.BulkEditObjectsViewTestCase,
-    # ViewTestCases.BulkImportObjectsViewTestCase,
+    ViewTestCases.BulkRenameObjectsViewTestCase,
+    ViewTestCases.BulkEditObjectsViewTestCase,
+    ViewTestCases.BulkImportObjectsViewTestCase,
 ):
     model = IPFabricTransformMapGroup
 
@@ -870,6 +872,21 @@ class IPFabricTransformMapGroupTestCase(
             "description": "Test group description",
         }
 
+        cls.bulk_edit_data = {
+            "description": "Bulk updated description",
+        }
+
+        cls.csv_data = (
+            "name,description",
+            "First imported group,import test 1",
+            "Second imported group,import test 2",
+        )
+        cls.csv_update_data = (
+            "id,name,description",
+            f"{groups[0].pk},First renamed group,changed import test 1",  # noqa: E231
+            f"{groups[1].pk},Second renamed group,changed import test 2",  # noqa: E231
+        )
+
 
 class IPFabricTransformMapTestCase(
     PluginPathMixin,
@@ -880,9 +897,9 @@ class IPFabricTransformMapTestCase(
     ViewTestCases.DeleteObjectViewTestCase,
     ViewTestCases.ListObjectsViewTestCase,
     ViewTestCases.BulkDeleteObjectsViewTestCase,
-    # ViewTestCases.BulkRenameObjectsViewTestCase,
-    # ViewTestCases.BulkEditObjectsViewTestCase,
-    # ViewTestCases.BulkImportObjectsViewTestCase,
+    ViewTestCases.BulkRenameObjectsViewTestCase,
+    ViewTestCases.BulkEditObjectsViewTestCase,
+    ViewTestCases.BulkImportObjectsViewTestCase,
 ):
     model = IPFabricTransformMap
     user_permissions = (
@@ -893,7 +910,11 @@ class IPFabricTransformMapTestCase(
     @classmethod
     def setUpTestData(cls):
         # Number of IPFabricTransformMaps created during migration
-        cls.default_maps_count = 14
+        # Hardcoded since we need to make sure we have the correct count
+        cls.default_maps = IPFabricTransformMap.objects.filter(group__isnull=True)
+        assert cls.default_maps.count() == 14
+        # Remove all transform maps created in migrations to not interfere with tests
+        IPFabricTransformMap.objects.filter(group__isnull=True).delete()
 
         # Create required dependencies
         group = IPFabricTransformMapGroup.objects.create(
@@ -904,14 +925,16 @@ class IPFabricTransformMapTestCase(
             name="Test Cloning Group",
             description="Test group for cloning transform maps",
         )
+        bulk_edit_group = IPFabricTransformMapGroup.objects.create(
+            name="Test Bulk Edit Group",
+            description="Test group for bulk editing transform maps",
+        )
 
-        # Create IPFabricTransformMap instances with groups for testing
         maps = (
             IPFabricTransformMap(
                 name="Device Transform Map",
                 source_model="device",
                 target_model=ContentType.objects.get(app_label="dcim", model="device"),
-                group=group,
             ),
             IPFabricTransformMap(
                 name="Site Transform Map",
@@ -919,10 +942,30 @@ class IPFabricTransformMapTestCase(
                 target_model=ContentType.objects.get(app_label="dcim", model="site"),
                 group=group,
             ),
+            IPFabricTransformMap(
+                name="VLAN Transform Map",
+                source_model="vlan",
+                target_model=ContentType.objects.get(app_label="ipam", model="vlan"),
+                group=group,
+            ),
         )
         for map_obj in maps:
             map_obj.save()
 
+        IPFabricTransformField.objects.create(
+            transform_map=maps[0],
+            source_field="hostname",
+            target_field="name",
+            template="{{ object.hostname }}",
+        )
+        IPFabricRelationshipField.objects.create(
+            transform_map=maps[0],
+            source_model=ContentType.objects.get(app_label="dcim", model="site"),
+            target_field="site",
+            template="{{ object.siteName }}",
+            coalesce=True,
+        )
+
         cls.form_data = {
             "name": "Test Transform Map X",
             "source_model": "device",
@@ -932,6 +975,22 @@ class IPFabricTransformMapTestCase(
             "group": group.pk,
         }
 
+        cls.bulk_edit_data = {
+            "group": bulk_edit_group.pk,
+        }
+
+        cls.csv_data = (
+            "name,source_model,target_model,group",
+            "Manufacturer Transform Map,device,dcim.manufacturer,Test Group",
+            "IPAddress Transform Map,ipaddress,ipam.ipaddress,Test Group",
+            "Platform Transform Map,device,dcim.platform,",
+        )
+        cls.csv_update_data = (
+            "id,name,source_model,target_model,group",
+            f"{maps[0].pk},Prefix Transform Map,prefix,ipam.prefix,Test Group",  # noqa: E231
+            f"{maps[1].pk},Manufacturer Transform Map,device,dcim.manufacturer,",  # noqa: E231
+        )
+
     @override_settings(EXEMPT_VIEW_PERMISSIONS=["*"])
     def test_get_relationships(self):
         transform_map = self._get_queryset().first()
@@ -1166,7 +1225,8 @@ class IPFabricTransformMapTestCase(
         # Should show form errors
         self.assertHttpStatus(response, 200)
         self.assertContains(
-            response, "A transform map with this group and target model already exists."
+            response,
+            "A transform map with group 'Test Group' and target model 'DCIM | site' already exists.",
         )
         self.assertIn("X-Debug-HTMX-Partial", response)
 
@@ -1237,10 +1297,7 @@ class IPFabricTransformMapTestCase(
     @override_settings(EXEMPT_VIEW_PERMISSIONS=["*"])
     def test_restore_view_post_success(self):
         """Test POST request to restore view successfully deletes ungrouped maps and rebuilds them."""
-        default_maps = list(IPFabricTransformMap.objects.filter(group__isnull=True))
-        self.assertEqual(len(default_maps), self.default_maps_count)
-
-        # Remove all transform maps created in migrations
+        # Remove all existing transform maps
         IPFabricTransformMap.objects.filter(group__isnull=True).delete()
         self.assertEqual(
             IPFabricTransformMap.objects.filter(group__isnull=True).count(), 0
@@ -1258,9 +1315,9 @@ class IPFabricTransformMapTestCase(
         # Verify ungrouped transform maps were restored
         self.assertEqual(
             IPFabricTransformMap.objects.filter(group__isnull=True).count(),
-            self.default_maps_count,
+            self.default_maps.count(),
         )
-        for map in default_maps:
+        for map in self.default_maps:
             self.assertTrue(
                 IPFabricTransformMap.objects.filter(
                     name=map.name,
@@ -1288,140 +1345,139 @@ class IPFabricTransformMapTestCase(
         self.assertIn(response.status_code, [302, 403])
 
 
-# TODO: These have no detail view, so it's failing of finding reverse URL
-# class IPFabricTransformFieldTestCase(
-#     PluginPathMixin,
-#     # ViewTestCases.GetObjectViewTestCase,
-#     # ViewTestCases.GetObjectChangelogViewTestCase,
-#     ViewTestCases.CreateObjectViewTestCase,
-#     ViewTestCases.EditObjectViewTestCase,
-#     ViewTestCases.DeleteObjectViewTestCase,
-#     ViewTestCases.ListObjectsViewTestCase,
-#     ViewTestCases.BulkDeleteObjectsViewTestCase,
-#     # ViewTestCases.BulkRenameObjectsViewTestCase,
-#     # ViewTestCases.BulkEditObjectsViewTestCase,
-#     # ViewTestCases.BulkImportObjectsViewTestCase,
-# ):
-#     model = IPFabricTransformField
-#
-#     @classmethod
-#     def setUpTestData(cls):
-#         # Create required dependencies
-#         group = IPFabricTransformMapGroup.objects.create(
-#             name="Test Group",
-#             description="Test group for transform fields",
-#         )
-#
-#         transform_map = IPFabricTransformMap.objects.create(
-#             name="Test Transform Map",
-#             source_model="devices",
-#             target_model=ContentType.objects.get(app_label="dcim", model="device"),
-#             group=group,
-#         )
-#
-#         # Create three IPFabricTransformField instances for testing
-#         fields = (
-#             IPFabricTransformField(
-#                 transform_map=transform_map,
-#                 source_field="hostname",
-#                 target_field="name",
-#                 coalesce=False,
-#                 template="{{ object.hostname }}",
-#             ),
-#             IPFabricTransformField(
-#                 transform_map=transform_map,
-#                 source_field="vendor",
-#                 target_field="manufacturer",
-#                 coalesce=True,
-#                 template="{{ object.vendor }}",
-#             ),
-#             IPFabricTransformField(
-#                 transform_map=transform_map,
-#                 source_field="model",
-#                 target_field="device_type",
-#                 coalesce=False,
-#                 template="{{ object.model }}",
-#             ),
-#         )
-#         for field in fields:
-#             field.save()
-#
-#         cls.form_data = {
-#             "transform_map": transform_map.pk,
-#             "source_field": "serial_number",
-#             "target_field": "serial",
-#             "coalesce": False,
-#             "template": "{{ object.serial_number }}",
-#         }
-#
-#
-# class IPFabricRelationshipFieldTestCase(
-#     PluginPathMixin,
-#     # ViewTestCases.GetObjectViewTestCase,
-#     # ViewTestCases.GetObjectChangelogViewTestCase,
-#     ViewTestCases.CreateObjectViewTestCase,
-#     ViewTestCases.EditObjectViewTestCase,
-#     ViewTestCases.DeleteObjectViewTestCase,
-#     ViewTestCases.ListObjectsViewTestCase,
-#     ViewTestCases.BulkDeleteObjectsViewTestCase,
-#     # ViewTestCases.BulkRenameObjectsViewTestCase,
-#     # ViewTestCases.BulkEditObjectsViewTestCase,
-#     # ViewTestCases.BulkImportObjectsViewTestCase,
-# ):
-#     model = IPFabricRelationshipField
-#
-#     @classmethod
-#     def setUpTestData(cls):
-#         # Create required dependencies
-#         group = IPFabricTransformMapGroup.objects.create(
-#             name="Test Group",
-#             description="Test group for relationship fields",
-#         )
-#
-#         device_ct = ContentType.objects.get(app_label="dcim", model="device")
-#         site_ct = ContentType.objects.get(app_label="dcim", model="site")
-#
-#         transform_map = IPFabricTransformMap.objects.create(
-#             name="Test Transform Map",
-#             source_model="devices",
-#             target_model=device_ct,
-#             group=group,
-#         )
-#
-#         # Create three IPFabricRelationshipField instances for testing
-#         fields = (
-#             IPFabricRelationshipField(
-#                 transform_map=transform_map,
-#                 source_model=site_ct,
-#                 target_field="site",
-#                 coalesce=False,
-#                 template="{{ object.site_id }}",
-#             ),
-#             IPFabricRelationshipField(
-#                 transform_map=transform_map,
-#                 source_model=device_ct,
-#                 target_field="parent_device",
-#                 coalesce=True,
-#                 template="{{ object.parent_id }}",
-#             ),
-#             IPFabricRelationshipField(
-#                 transform_map=transform_map,
-#                 source_model=site_ct,
-#                 target_field="location",
-#                 coalesce=False,
-#                 template="{{ object.location_id }}",
-#             ),
-#         )
-#         for field in fields:
-#             field.save()
-#
-#         cls.form_data = {
-#             "transform_map": transform_map.pk,
-#             "source_model": site_ct.pk,
-#             "target_field": "site",
-#             "coalesce": False,
-#             "template": "{{ object.site_id }}",
-#         }
+class IPFabricTransformFieldTestCase(
+    PluginPathMixin,
+    # ViewTestCases.GetObjectViewTestCase,
+    # ViewTestCases.GetObjectChangelogViewTestCase,
+    ViewTestCases.CreateObjectViewTestCase,
+    ViewTestCases.EditObjectViewTestCase,
+    ViewTestCases.DeleteObjectViewTestCase,
+    # ViewTestCases.ListObjectsViewTestCase,
+    ViewTestCases.BulkDeleteObjectsViewTestCase,
+    # ViewTestCases.BulkRenameObjectsViewTestCase,
+    # ViewTestCases.BulkEditObjectsViewTestCase,
+    # ViewTestCases.BulkImportObjectsViewTestCase,
+):
+    model = IPFabricTransformField
+
+    @classmethod
+    def setUpTestData(cls):
+        # Create required dependencies
+        group = IPFabricTransformMapGroup.objects.create(
+            name="Test Group",
+            description="Test group for transform fields",
+        )
+
+        transform_map = IPFabricTransformMap.objects.create(
+            name="Test Transform Map",
+            source_model="devices",
+            target_model=ContentType.objects.get(app_label="dcim", model="device"),
+            group=group,
+        )
+
+        # Create three IPFabricTransformField instances for testing
+        fields = (
+            IPFabricTransformField(
+                transform_map=transform_map,
+                source_field="hostname",
+                target_field="name",
+                coalesce=False,
+                template="{{ object.hostname }}",
+            ),
+            IPFabricTransformField(
+                transform_map=transform_map,
+                source_field="vendor",
+                target_field="manufacturer",
+                coalesce=True,
+                template="{{ object.vendor }}",
+            ),
+            IPFabricTransformField(
+                transform_map=transform_map,
+                source_field="model",
+                target_field="device_type",
+                coalesce=False,
+                template="{{ object.model }}",
+            ),
+        )
+        for field in fields:
+            field.save()
+
+        cls.form_data = {
+            "transform_map": transform_map.pk,
+            "source_field": "serial_number",
+            "target_field": "serial",
+            "coalesce": False,
+            "template": "{{ object.serial_number }}",
+        }
+
+
+class IPFabricRelationshipFieldTestCase(
+    PluginPathMixin,
+    # ViewTestCases.GetObjectViewTestCase,
+    # ViewTestCases.GetObjectChangelogViewTestCase,
+    ViewTestCases.CreateObjectViewTestCase,
+    ViewTestCases.EditObjectViewTestCase,
+    ViewTestCases.DeleteObjectViewTestCase,
+    # ViewTestCases.ListObjectsViewTestCase,
+    ViewTestCases.BulkDeleteObjectsViewTestCase,
+    # ViewTestCases.BulkRenameObjectsViewTestCase,
+    # ViewTestCases.BulkEditObjectsViewTestCase,
+    # ViewTestCases.BulkImportObjectsViewTestCase,
+):
+    model = IPFabricRelationshipField
+
+    @classmethod
+    def setUpTestData(cls):
+        # Create required dependencies
+        group = IPFabricTransformMapGroup.objects.create(
+            name="Test Group",
+            description="Test group for relationship fields",
+        )
+
+        device_ct = ContentType.objects.get(app_label="dcim", model="device")
+        site_ct = ContentType.objects.get(app_label="dcim", model="site")
+
+        transform_map = IPFabricTransformMap.objects.create(
+            name="Test Transform Map",
+            source_model="devices",
+            target_model=device_ct,
+            group=group,
+        )
+
+        # Create three IPFabricRelationshipField instances for testing
+        fields = (
+            IPFabricRelationshipField(
+                transform_map=transform_map,
+                source_model=site_ct,
+                target_field="site",
+                coalesce=False,
+                template="{{ object.site_id }}",
+            ),
+            IPFabricRelationshipField(
+                transform_map=transform_map,
+                source_model=device_ct,
+                target_field="parent_device",
+                coalesce=True,
+                template="{{ object.parent_id }}",
+            ),
+            IPFabricRelationshipField(
+                transform_map=transform_map,
+                source_model=site_ct,
+                target_field="location",
+                coalesce=False,
+                template="{{ object.location_id }}",
+            ),
+        )
+        for field in fields:
+            field.save()
+
+        cls.form_data = {
+            "transform_map": transform_map.pk,
+            "source_model": site_ct.pk,
+            "target_field": "site",
+            "coalesce": False,
+            "template": "{{ object.site_id }}",
+        }
 
 
 class IPFabricIngestionTestCase(
@@ -1432,7 +1488,7 @@ class IPFabricIngestionTestCase(
     # ViewTestCases.EditObjectViewTestCase,
     # ViewTestCases.DeleteObjectViewTestCase,
     ViewTestCases.ListObjectsViewTestCase,
-    # ViewTestCases.BulkDeleteObjectsViewTestCase,
+    ViewTestCases.BulkDeleteObjectsViewTestCase,
     # ViewTestCases.BulkRenameObjectsViewTestCase,
     # ViewTestCases.BulkEditObjectsViewTestCase,
     # ViewTestCases.BulkImportObjectsViewTestCase,