Merge branch 'master' into aureplop/fewer_requests

bda/models.py

@@ -18,7 +18,10 @@ class Tirage(models.Model):
     fermeture = models.DateTimeField("Date et heure de fermerture du tirage")
     tokens = models.TextField("Graine(s) du tirage", blank=True)
     active = models.BooleanField("Tirage actif", default=False)
-    appear_catalogue = models.BooleanField("Tirage à afficher dans le catalogue", default=False)
+    appear_catalogue = models.BooleanField(
+        "Tirage à afficher dans le catalogue",
+        default=False
+    )
     enable_do_tirage = models.BooleanField("Le tirage peut être lancé",
                                            default=False)
 

bda/tests.py

@@ -1,22 +1,79 @@
-# -*- coding: utf-8 -*-
+import json
-"""
-This file demonstrates writing tests using the unittest module. These will pass
-when you run "manage.py test".
 
-Replace this with more appropriate tests for your application.
+from django.test import TestCase, Client
-"""
+from django.utils import timezone
 
-from __future__ import division
+from .models import Tirage, Spectacle, Salle, CategorieSpectacle
-from __future__ import print_function
-from __future__ import unicode_literals
 
 
-from django.test import TestCase
+class TestBdAViews(TestCase):
+    def setUp(self):
+        self.tirage = Tirage.objects.create(
+            title="Test tirage",
+            appear_catalogue=True,
+            ouverture=timezone.now(),
+            fermeture=timezone.now(),
+        )
+        self.category = CategorieSpectacle.objects.create(name="Category")
+        self.location = Salle.objects.create(name="here")
+        Spectacle.objects.bulk_create([
+            Spectacle(
+                title="foo", date=timezone.now(), location=self.location,
+                price=0, slots=42, tirage=self.tirage, listing=False,
+                category=self.category
+            ),
+            Spectacle(
+                title="bar", date=timezone.now(), location=self.location,
+                price=1, slots=142, tirage=self.tirage, listing=False,
+                category=self.category
+            ),
+            Spectacle(
+                title="baz", date=timezone.now(), location=self.location,
+                price=2, slots=242, tirage=self.tirage, listing=False,
+                category=self.category
+            ),
+        ])
 
+    def test_catalogue(self):
+        """Test the catalogue JSON API"""
+        client = Client()
 
-class SimpleTest(TestCase):
+        # The `list` hooh
-    def test_basic_addition(self):
+        resp = client.get("/bda/catalogue/list")
-        """
+        self.assertJSONEqual(
-        Tests that 1 + 1 always equals 2.
+            resp.content.decode("utf-8"),
-        """
+            [{"id": self.tirage.id, "title": self.tirage.title}]
-        self.assertEqual(1 + 1, 2)
+        )
+
+        # The `details` hook
+        resp = client.get(
+            "/bda/catalogue/details?id={}".format(self.tirage.id)
+        )
+        self.assertJSONEqual(
+            resp.content.decode("utf-8"),
+            {
+                "categories": [{
+                    "id": self.category.id,
+                    "name": self.category.name
+                }],
+                "locations": [{
+                    "id": self.location.id,
+                    "name": self.location.name
+                }],
+            }
+        )
+
+        # The `descriptions` hook
+        resp = client.get(
+            "/bda/catalogue/descriptions?id={}".format(self.tirage.id)
+        )
+        raw = resp.content.decode("utf-8")
+        try:
+            results = json.loads(raw)
+        except ValueError:
+            self.fail("Not valid JSON: {}".format(raw))
+        self.assertEqual(len(results), 3)
+        self.assertEqual(
+            {(s["title"], s["price"], s["slots"]) for s in results},
+            {("foo", 0, 42), ("bar", 1, 142), ("baz", 2, 242)}
+        )

bda/views.py

@@ -22,7 +22,6 @@ from django.core.urlresolvers import reverse
 from django.conf import settings
 from django.utils import timezone, formats
 from django.views.generic.list import ListView
-from django.core.exceptions import ObjectDoesNotExist
 from gestioncof.decorators import cof_required, buro_required
 from bda.models import (
     Spectacle, Participant, ChoixSpectacle, Attribution, Tirage,
@@ -657,29 +656,35 @@ def catalogue(request, request_type):
     if request_type == "list":
         # Dans ce cas on retourne la liste des tirages et de leur id en JSON
         data_return = list(
-            Tirage.objects.filter(appear_catalogue=True).values('id', 'title'))
+            Tirage.objects.filter(appear_catalogue=True).values('id', 'title')
+        )
         return JsonResponse(data_return, safe=False)
     if request_type == "details":
         # Dans ce cas on retourne une liste des catégories et des salles
-        tirage_id = request.GET.get('id', '')
+        tirage_id = request.GET.get('id', None)
-        try:
+        if tirage_id is None:
-            tirage = Tirage.objects.get(id=tirage_id)
-        except ObjectDoesNotExist:        
             return HttpResponseBadRequest(
-                "Aucun tirage correspondant à l'id "
+                "Missing GET parameter: id <int>"
-                + tirage_id)
+            )
+        try:
+            tirage = get_object_or_404(Tirage, id=int(tirage_id))
         except ValueError:
             return HttpResponseBadRequest(
-                "Mauvais format d'identifiant : "
+                "Bad format: int expected for `id`"
-                + tirage_id)
+            )
+        shows = tirage.spectacle_set.values_list("id", flat=True)
         categories = list(
-            CategorieSpectacle.objects.filter(
+            CategorieSpectacle.objects
-                spectacle__in=tirage.spectacle_set.all())
+            .filter(spectacle__in=shows)
-            .distinct().values('id', 'name'))
+            .distinct()
+            .values('id', 'name')
+        )
         locations = list(
-            Salle.objects.filter(
+            Salle.objects
-                spectacle__in=tirage.spectacle_set.all())
+            .filter(spectacle__in=shows)
-            .distinct().values('id', 'name'))
+            .distinct()
+            .values('id', 'name')
+        )
         data_return = {'categories': categories, 'locations': locations}
         return JsonResponse(data_return, safe=False)
     if request_type == "descriptions":
@@ -687,33 +692,35 @@ def catalogue(request, request_type):
         # à la salle spécifiées
 
         tirage_id = request.GET.get('id', '')
-        categories = request.GET.get('category', '[0]')
+        categories = request.GET.get('category', '[]')
-        locations = request.GET.get('location', '[0]')
+        locations = request.GET.get('location', '[]')
         try:
-            category_id = json.loads(categories)
+            tirage_id = int(tirage_id)
-            location_id = json.loads(locations)
+            categories_id = json.loads(categories)
-            tirage = Tirage.objects.get(id=tirage_id)
+            locations_id = json.loads(locations)
-
+            # Integers expected
-            shows_qs = tirage.spectacle_set
+            if not all(isinstance(id, int) for id in categories_id):
-            if not(0 in category_id):
+                raise ValueError
-                shows_qs = shows_qs.filter(
+            if not all(isinstance(id, int) for id in locations_id):
-                    category__id__in=category_id)
+                raise ValueError
-            if not(0 in location_id):
-                shows_qs = shows_qs.filter(
-                    location__id__in=location_id)
-        except ObjectDoesNotExist:
-            return HttpResponseBadRequest(
-                "Impossible de trouver des résultats correspondant "
-                "à ces caractéristiques : "
-					 + "id = " + tirage_id 
-					 + ", catégories = " + categories
-					 + ", salles = " + locations)
         except ValueError:  # Contient JSONDecodeError
             return HttpResponseBadRequest(
-				    "Impossible de parser les paramètres donnés : "
+                "Parse error, please ensure the GET parameters have the "
-				    + "id = " + request.GET.get('id', '')
+                "following types:\n"
-				    + ", catégories = " + request.GET.get('category', '[0]')
+                "id: int, category: [int], location: [int]\n"
-				    + ", salles = " + request.GET.get('location', '[0]'))
+                "Data received:\n"
+                "id = {}, category = {}, locations = {}"
+                .format(request.GET.get('id', ''),
+                        request.GET.get('category', '[]'),
+                        request.GET.get('location', '[]'))
+            )
+        tirage = get_object_or_404(Tirage, id=tirage_id)
+
+        shows_qs = tirage.spectacle_set
+        if categories_id:
+            shows_qs = shows_qs.filter(category__id__in=categories_id)
+        if locations_id:
+            shows_qs = shows_qs.filter(location__id__in=locations_id)
 
         # On convertit les descriptions à envoyer en une liste facilement
         # JSONifiable (il devrait y avoir un moyen plus efficace en