NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-24 17:38:37 -06:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Fix ID list creation in API tests

Browse Source
This commit is contained in:
jeremystretch 2021-11-04 11:31:39 -04:00
parent 7b1335825b
commit 8f1acb700d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
netbox/utilities/testing/api.py
View File

@ -345,7 +345,7 @@ class APIViewTestCases:
obj_perm.users.add(self.user) obj_perm.users.add(self.user)
obj_perm.object_types.add(ContentType.objects.get_for_model(self.model)) obj_perm.object_types.add(ContentType.objects.get_for_model(self.model))
id_list = self._get_queryset().values_list('id', flat=True)[:3] id_list = list(self._get_queryset().values_list('id', flat=True)[:3])
self.assertEqual(len(id_list), 3, "Insufficient number of objects to test bulk update") self.assertEqual(len(id_list), 3, "Insufficient number of objects to test bulk update")
data = [ data = [
{'id': id, **self.bulk_update_data} for id in id_list {'id': id, **self.bulk_update_data} for id in id_list
@ -416,7 +416,7 @@ class APIViewTestCases:
# Target the three most recently created objects to avoid triggering recursive deletions # Target the three most recently created objects to avoid triggering recursive deletions
# (e.g. with MPTT objects) # (e.g. with MPTT objects)
id_list = self._get_queryset().order_by('-id').values_list('id', flat=True)[:3] id_list = list(self._get_queryset().order_by('-id').values_list('id', flat=True)[:3])
self.assertEqual(len(id_list), 3, "Insufficient number of objects to test bulk deletion") self.assertEqual(len(id_list), 3, "Insufficient number of objects to test bulk deletion")
data = [{"id": id} for id in id_list] data = [{"id": id} for id in id_list]