diff --git a/bda/tests/utils.py b/bda/tests/utils.py index 5b486dd0..68f51fb6 100644 --- a/bda/tests/utils.py +++ b/bda/tests/utils.py @@ -1,4 +1,9 @@ +from datetime import timedelta + from django.contrib.auth.models import User +from django.utils import timezone + +from ..models import CategorieSpectacle, Salle, Spectacle, Tirage def create_user(username, is_cof=False, is_buro=False): @@ -15,3 +20,17 @@ def user_is_cof(user): def user_is_staff(user): return (user is not None) and user.profile.is_buro + + +def create_spectacle(**kwargs): + defaults = { + "title": "Title", + "category": CategorieSpectacle.objects.first(), + "date": (timezone.now() + timedelta(days=7)).date(), + "location": Salle.objects.first(), + "price": 10.0, + "slots": 20, + "tirage": Tirage.objects.first(), + "listing": False, + } + return Spectacle.objects.create(**dict(defaults, **kwargs))