diff --git a/cookbook/tests/views/test_views_general.py b/cookbook/tests/views/test_views_general.py index efdc33372..79e3cdc8c 100644 --- a/cookbook/tests/views/test_views_general.py +++ b/cookbook/tests/views/test_views_general.py @@ -7,10 +7,10 @@ class TestViewsGeneral(TestViews): def test_index(self): r = self.client.get(reverse('index')) - self.assertEqual(r.status_code, 200) + self.assertEqual(r.status_code, 302) r = self.anonymous_client.get(reverse('index')) - self.assertEqual(r.status_code, 200) + self.assertEqual(r.status_code, 302) def test_books(self): url = reverse('view_books') diff --git a/cookbook/views/views.py b/cookbook/views/views.py index dc7583649..c79b54ebe 100644 --- a/cookbook/views/views.py +++ b/cookbook/views/views.py @@ -20,14 +20,16 @@ from cookbook.tables import RecipeTable def index(request): if not request.user.is_authenticated: return HttpResponseRedirect(reverse_lazy('view_search')) + try: + page_map = { + UserPreference.SEARCH: reverse_lazy('view_search'), + UserPreference.PLAN: reverse_lazy('view_plan'), + UserPreference.BOOKS: reverse_lazy('view_books'), + } - page_map = { - UserPreference.SEARCH: reverse_lazy('view_search'), - UserPreference.PLAN: reverse_lazy('view_plan'), - UserPreference.BOOKS: reverse_lazy('view_books'), - } - - return HttpResponseRedirect(page_map.get(request.user.userpreference.default_page)) + return HttpResponseRedirect(page_map.get(request.user.userpreference.default_page)) + except UserPreference.DoesNotExist: + return HttpResponseRedirect(reverse_lazy('view_search')) def search(request):