mirror of
https://github.com/ae-utbm/sith.git
synced 2025-04-16 11:00:21 +00:00
61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
from datetime import timedelta
|
|
|
|
from django.conf import settings
|
|
from django.core.cache import cache
|
|
from django.test import TestCase
|
|
from django.urls import reverse
|
|
from django.utils.timezone import now
|
|
from model_bakery import baker
|
|
from model_bakery.recipe import Recipe
|
|
|
|
from club.models import Club, Membership
|
|
from core.baker_recipes import old_subscriber_user, subscriber_user
|
|
from core.models import User
|
|
|
|
|
|
class TestClub(TestCase):
|
|
"""Set up data for test cases related to clubs and membership.
|
|
|
|
The generated dataset is the one created by the populate command,
|
|
plus the following modifications :
|
|
|
|
- `self.club` is a dummy club
|
|
- `self.club` has two board members :
|
|
simple_board_member (role 3) and president (role 10)
|
|
- `self.club` has one regular member : richard
|
|
- `self.club` has one former member : sli (who had role 2)
|
|
- None of the `self.club` members are in the AE club.
|
|
"""
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
# subscribed users - initial members
|
|
cls.president, cls.simple_board_member = subscriber_user.make(_quantity=2)
|
|
cls.richard = User.objects.get(username="rbatsbak")
|
|
cls.sli = User.objects.get(username="sli")
|
|
cls.root = baker.make(User, is_superuser=True)
|
|
cls.old_subscriber = old_subscriber_user.make()
|
|
cls.public = baker.make(User)
|
|
|
|
# subscribed users - not initial member
|
|
cls.krophil = User.objects.get(username="krophil")
|
|
cls.subscriber = subscriber_user.make()
|
|
|
|
cls.ae = Club.objects.get(pk=settings.SITH_MAIN_CLUB_ID)
|
|
cls.club = baker.make(Club)
|
|
cls.members_url = reverse("club:club_members", kwargs={"club_id": cls.club.id})
|
|
a_month_ago = now() - timedelta(days=30)
|
|
yesterday = now() - timedelta(days=1)
|
|
membership_recipe = Recipe(Membership, club=cls.club)
|
|
membership_recipe.make(
|
|
user=cls.simple_board_member, start_date=a_month_ago, role=3
|
|
)
|
|
membership_recipe.make(user=cls.richard, role=1)
|
|
membership_recipe.make(user=cls.president, start_date=a_month_ago, role=10)
|
|
membership_recipe.make( # sli was a member but isn't anymore
|
|
user=cls.sli, start_date=a_month_ago, end_date=yesterday, role=2
|
|
)
|
|
|
|
def setUp(self):
|
|
cache.clear()
|