From 1664554083773472858906fed609e64dee552314 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aur=C3=A9lien=20Delobelle?= <aurelien.delobelle@gmail.com>
Date: Sun, 28 Oct 2018 21:13:04 +0100
Subject: [PATCH] bda.tests -- Add Spectacle factories

---
 bda/tests/utils.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

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))