mirror of
https://github.com/ae-utbm/sith.git
synced 2026-03-31 15:59:42 +00:00
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
import pytest
|
|
from model_bakery import baker, seq
|
|
from model_bakery.recipe import Recipe
|
|
|
|
from club.models import Club, ClubRole
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_order_auto():
|
|
"""Test that newly created roles are put in the right place."""
|
|
club = baker.make(Club)
|
|
recipe = Recipe(ClubRole, club=club, name=seq("role "))
|
|
# bulk create initial roles
|
|
roles = recipe.make(
|
|
is_board=iter([True, True, False]),
|
|
is_presidency=iter([True, False, False]),
|
|
order=iter([1, 2, 3]),
|
|
_quantity=3,
|
|
_bulk_create=True,
|
|
)
|
|
# then create the remaining roles one by one (like they will be in prod)
|
|
# each new role should be placed at the end of its category
|
|
role_a = recipe.make(is_board=True, is_presidency=True, order=None)
|
|
role_b = recipe.make(is_board=True, is_presidency=False, order=None)
|
|
role_c = recipe.make(is_board=False, is_presidency=False, order=None)
|
|
assert list(club.roles.order_by("order")) == [
|
|
roles[0],
|
|
role_a,
|
|
roles[1],
|
|
role_b,
|
|
roles[2],
|
|
role_c,
|
|
]
|