Adjust num queries

allegro · May 16, 2024 · d5d2ea0 · d5d2ea0
1 parent 1112b75
commit d5d2ea0
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/src/ralph/data_center/tests/test_view.py b/src/ralph/data_center/tests/test_view.py
@@ -39,7 +39,7 @@ class DataCenterAssetViewTest(ClientMixin, TestCase):
     def test_changelist_view(self):
         self.login_as_user()
         DataCenterAssetFullFactory.create_batch(10)
-        with self.assertNumQueries(20):
+        with self.assertNumQueries(21):
             self.client.get(
                 reverse('admin:data_center_datacenterasset_changelist'),
             )
@@ -54,7 +54,7 @@ def test_changelist_view(self):
         VirtualServerFullFactory.create_batch(5)
         CloudHostFullFactory.create_batch(4)
         ClusterFactory.create_batch(4)
-        with self.assertNumQueries(21):
+        with self.assertNumQueries(22):
             result = self.client.get(
                 reverse('admin:data_center_dchost_changelist'),
             )

diff --git a/src/ralph/data_importer/tests/test_fields.py b/src/ralph/data_importer/tests/test_fields.py
@@ -69,7 +69,7 @@ def test_users_through_field(self):
         self.assertEqual(self.licence2.users.all().count(), 3)
 
         # Add and remove
-        with self.assertNumQueries(4):
+        with self.assertNumQueries(5):
             field.save(
                 self.licence,
                 {'users': ','.join([i.username for i in self.users])}
@@ -88,7 +88,7 @@ def test_users_through_field(self):
         self.assertEqual(self.licence.users.all().count(), 5)
 
         # Remove
-        with self.assertNumQueries(3):
+        with self.assertNumQueries(4):
             field.save(
                 self.licence,
                 {'users': ','.join([i.username for i in users[:4]])}
@@ -143,7 +143,7 @@ def test_through_field_only_remove(self):
         field = self._get_base_objects_through_field()
         self.assertEqual(self.licence.base_objects.all().count(), 2)
         ids = [self.back_office_assets[0].pk]
-        with self.assertNumQueries(3):
+        with self.assertNumQueries(4):
             field.save(
                 self.licence,
                 {'base_objects': ','.join(map(str, ids))}
@@ -165,7 +165,7 @@ def test_through_field_add_and_remove(self):
             self.back_office_assets[3].pk,
 
         ]
-        with self.assertNumQueries(4):
+        with self.assertNumQueries(5):
             field.save(
                 self.licence,
                 {'base_objects': ','.join(map(str, ids))}

diff --git a/src/ralph/lib/mixins/models.py b/src/ralph/lib/mixins/models.py
@@ -4,7 +4,8 @@
 from django.utils import timezone
 from django.utils.translation import ugettext_lazy as _
 from djmoney.models.fields import MoneyField
-from taggit.managers import TaggableManager as TaggableManagerOriginal, _TaggableManager
+from taggit.managers import TaggableManager as TaggableManagerOriginal
+from taggit.managers import _TaggableManager  # noqa
 
 from ralph.lib.mixins.fields import TaggitTagField
 from ralph.settings import DEFAULT_CURRENCY_CODE