mirror of
https://github.com/TandoorRecipes/recipes.git
synced 2026-01-05 22:28:31 -05:00
imrpoved index redirect + fixed tests
This commit is contained in:
@@ -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')
|
||||||
|
|||||||
@@ -20,14 +20,16 @@ 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 = {
|
||||||
|
UserPreference.SEARCH: reverse_lazy('view_search'),
|
||||||
|
UserPreference.PLAN: reverse_lazy('view_plan'),
|
||||||
|
UserPreference.BOOKS: reverse_lazy('view_books'),
|
||||||
|
}
|
||||||
|
|
||||||
page_map = {
|
return HttpResponseRedirect(page_map.get(request.user.userpreference.default_page))
|
||||||
UserPreference.SEARCH: reverse_lazy('view_search'),
|
except UserPreference.DoesNotExist:
|
||||||
UserPreference.PLAN: reverse_lazy('view_plan'),
|
return HttpResponseRedirect(reverse_lazy('view_search'))
|
||||||
UserPreference.BOOKS: reverse_lazy('view_books'),
|
|
||||||
}
|
|
||||||
|
|
||||||
return HttpResponseRedirect(page_map.get(request.user.userpreference.default_page))
|
|
||||||
|
|
||||||
|
|
||||||
def search(request):
|
def search(request):
|
||||||
|
|||||||
Reference in New Issue
Block a user