Fix data_importer tests

allegro · May 6, 2024 · 63f897b · 63f897b
1 parent dcfcb53
commit 63f897b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/ralph/data_importer/tests/test_fields.py b/src/ralph/data_importer/tests/test_fields.py
@@ -134,7 +134,7 @@ def test_through_field_only_add(self):
 
         self.assertEqual(self.licence.base_objects.all().count(), 4)
         self.assertCountEqual(
-            self.licence.base_objects.values_list('pk', flat=True), ids
+            [bo.pk for bo in self.licence.base_objects.all()], ids
         )
         # Make sure it doesn't touch other licences
         self.assertEqual(self.licence2.base_objects.all().count(), 2)
@@ -151,7 +151,7 @@ def test_through_field_only_remove(self):
 
         self.assertEqual(self.licence.base_objects.all().count(), 1)
         self.assertCountEqual(
-            self.licence.base_objects.values_list('pk', flat=True), ids
+            [bo.pk for bo in self.licence.base_objects.all()], ids
         )
         # Make sure it doesn't touch other licences
         self.assertEqual(self.licence2.base_objects.all().count(), 2)
@@ -172,7 +172,7 @@ def test_through_field_add_and_remove(self):
             )
         self.assertEqual(self.licence.base_objects.all().count(), 3)
         self.assertCountEqual(
-            self.licence.base_objects.values_list('pk', flat=True), ids
+            [bo.pk for bo in self.licence.base_objects.all()], ids
         )
         # Make sure it doesn't touch other licences
         self.assertEqual(self.licence2.base_objects.all().count(), 2)