Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from django.contrib.auth.decorators import login_required
- from django.shortcuts import render
- @login_required(login_url='/users/login')
- def secure(request):
- user = request.user
- return render(request, 'secure.html', {'email': user.email})
- #tests.py
- class SimpleTest(TestCase):
- def setUp(self):
- user = User.objects.create_user('temporary', 'temporary@gmail.com', 'temporary')
- def test_secure_page(self):
- self.client.login(username='temporary', password='temporary')
- response = self.client.get('/manufacturers/', follow=True)
- user = User.objects.get(username='temporary')
- self.assertEqual(response.context['user'].email, 'temporary@gmail.com')
Add Comment
Please, Sign In to add comment