From d5d2ea0e1847c2f5bc71b57707f31892ee96fb40 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20Szulc?= <pawel.szulc@allegro.com>
Date: Thu, 16 May 2024 12:26:39 +0200
Subject: [PATCH] Adjust num queries

---
 src/ralph/data_center/tests/test_view.py     | 4 ++--
 src/ralph/data_importer/tests/test_fields.py | 8 ++++----
 src/ralph/lib/mixins/models.py               | 3 ++-
 3 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/src/ralph/data_center/tests/test_view.py b/src/ralph/data_center/tests/test_view.py
index d4df78e0f2..0e320a997d 100644
--- 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
index f5474d2818..938d2268ae 100644
--- 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
index 75b85bd27c..8253e503db 100644
--- 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