404 if calendar is not found

Author: fcurella

events/tests/test_views.py

@@ -41,6 +41,11 @@ def test_event_list(self):
         self.assertEqual(response.status_code, 200)
         self.assertEqual(len(response.context['object_list']), 1)
 
+        url = reverse('events:event_list_past', kwargs={"calendar_slug": 'unexisting'})
+        response = self.client.get(url)
+
+        self.assertEqual(response.status_code, 404)
+
     def test_event_list_past(self):
         url = reverse('events:event_list_past', kwargs={"calendar_slug": self.calendar.slug})
         response = self.client.get(url)

events/views.py

@@ -1,6 +1,7 @@
 # Create your views here.
 import datetime
 
+from django.shortcuts import get_object_or_404
 from django.utils import timezone
 from django.views.generic import DetailView, ListView
 
@@ -74,7 +75,7 @@ def get_queryset(self):
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
         context['events_today'] = Event.objects.until_datetime(timezone.now()).filter(calendar__slug=self.kwargs['calendar_slug'])[:2]
-        context['calendar'] = Calendar.objects.get(slug=self.kwargs['calendar_slug'])
+        context['calendar'] = get_object_or_404(Calendar, slug=self.kwargs['calendar_slug'])
         return context