imrpoved index redirect + fixed tests

This commit is contained in:
vabene1111
2020-04-13 23:11:33 +02:00
parent 0b250c71aa
commit dab77e8e4f
2 changed files with 11 additions and 9 deletions

View File

@@ -7,10 +7,10 @@ class TestViewsGeneral(TestViews):
def test_index(self): def test_index(self):
r = self.client.get(reverse('index')) 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')) r = self.anonymous_client.get(reverse('index'))
self.assertEqual(r.status_code, 200) self.assertEqual(r.status_code, 302)
def test_books(self): def test_books(self):
url = reverse('view_books') url = reverse('view_books')

View File

@@ -20,7 +20,7 @@ from cookbook.tables import RecipeTable
def index(request): def index(request):
if not request.user.is_authenticated: if not request.user.is_authenticated:
return HttpResponseRedirect(reverse_lazy('view_search')) return HttpResponseRedirect(reverse_lazy('view_search'))
try:
page_map = { page_map = {
UserPreference.SEARCH: reverse_lazy('view_search'), UserPreference.SEARCH: reverse_lazy('view_search'),
UserPreference.PLAN: reverse_lazy('view_plan'), UserPreference.PLAN: reverse_lazy('view_plan'),
@@ -28,6 +28,8 @@ def index(request):
} }
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): def search(request):