python 如何在django上使用身份验证简单地测试视图。

Posted 2021-05-09

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', '[e-mail]', '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, '[e-mail]')