post-rebase fix

ae-utbm · Dec 19, 2024 · a7b1406 · a7b1406
1 parent 871ef60
commit a7b1406
Show file tree

Hide file tree

Showing 3 changed files with 184 additions and 234 deletions.
diff --git a/counter/tests/test_product.py b/counter/tests/test_product.py
@@ -13,7 +13,7 @@
 from pytest_django.asserts import assertNumQueries
 
 from core.baker_recipes import board_user, subscriber_user
-from core.models import RealGroup, User
+from core.models import Group, User
 from counter.models import Product, ProductType
 
 
@@ -51,16 +51,14 @@ def test_resize_product_icon(model):
         (
             lambda: baker.make(
                 User,
-                groups=[RealGroup.objects.get(pk=settings.SITH_GROUP_COUNTER_ADMIN_ID)],
+                groups=[Group.objects.get(pk=settings.SITH_GROUP_COUNTER_ADMIN_ID)],
             ),
             200,
         ),
         (
             lambda: baker.make(
                 User,
-                groups=[
-                    RealGroup.objects.get(pk=settings.SITH_GROUP_ACCOUNTING_ADMIN_ID)
-                ],
+                groups=[Group.objects.get(pk=settings.SITH_GROUP_ACCOUNTING_ADMIN_ID)],
             ),
             200,
         ),

diff --git a/counter/tests/test_product_type.py b/counter/tests/test_product_type.py
@@ -6,7 +6,7 @@
 from ninja_extra.testing import TestClient
 
 from core.baker_recipes import board_user, subscriber_user
-from core.models import RealGroup, User
+from core.models import Group, User
 from counter.api import ProductTypeController
 from counter.models import ProductType
 
@@ -70,16 +70,14 @@ def test_move_above_product_type(product_types: list[ProductType]):
         (
             lambda: baker.make(
                 User,
-                groups=[RealGroup.objects.get(pk=settings.SITH_GROUP_COUNTER_ADMIN_ID)],
+                groups=[Group.objects.get(pk=settings.SITH_GROUP_COUNTER_ADMIN_ID)],
             ),
             200,
         ),
         (
             lambda: baker.make(
                 User,
-                groups=[
-                    RealGroup.objects.get(pk=settings.SITH_GROUP_ACCOUNTING_ADMIN_ID)
-                ],
+                groups=[Group.objects.get(pk=settings.SITH_GROUP_ACCOUNTING_ADMIN_ID)],
             ),
             200,
         ),