測試用戶是否已成功登錄

Question

如何在提交註冊表單後測試用戶是否已登錄？

我嘗試了以下方法，但即使在將登錄邏輯添加到我的註冊視圖之前，它也會返回True。

def test_that_user_gets_logged_in(self): 
    response = self.client.post(reverse('auth-registration'), 
           { 'username':'foo', 
            'password1':'bar', 
            'password2':'bar' }) 

    user = User.objects.get(username='foo') 
    assert user.is_authenticated() 

多數民衆贊成被測試代碼：

class RegistrationView(CreateView): 
    template_name = 'auth/registration.html' 
    form_class = UserCreationForm 
    success_url = '/' 

    def auth_login(self, request, username, password): 
     ''' 
     Authenticate always needs to be called before login because it 
     adds which backend did the authentication which is required by login. 
     ''' 

     user = authenticate(username=username, password=password) 
     login(request, user) 

    def form_valid(self, form): 
     ''' 
     Overwrite form_valid to login. 
     ''' 

     #save the user 
     response = super(RegistrationView, self).form_valid(form) 

     #Get the user creditials 
     username = form.cleaned_data['username'] 
     password = form.cleaned_data['password1'] 

     #authenticate and login 
     self.auth_login(self.request, username, password) 

     return response 

Answer 1

這不是最好的答案。見https://stackoverflow.com/a/35871564/307511

Chronial給 如何使下面這種說法一個很好的例子。他的回答 比我現在的代碼更好。

---

最簡單的方法來測試用戶是否登錄是通過測試客戶端對象：

：

self.assertIn('_auth_user_id', self.client.session) 

你也可以檢查是否有特定的用戶登錄

self.assertEqual(int(self.client.session['_auth_user_id']), user.pk) 

作爲附加信息，response.request對象不是HttpRequest對象;相反，它是一個普通的字典，其中包含有關實際請求的一些信息，因此無論如何它都不具有user屬性。

此外，測試response.context對象並不安全，因爲您沒有上下文。

Answer 2

在User模型總是返回True方法is_authenticated()。在request.user是AnonymousUser的實例的情況下，False返回request.user.is_authenticated()，is_authenticated()方法總是返回False。 雖然測試，你可以看看response.context['request'].user.is_authenticated()。

您也可以嘗試在測試這需要登錄才能訪問另一個頁面，看看是否response.status回報200或302（從login_required重定向）。

Answer 3

你在哪裏初始化你的self.client？你的setUp方法還有什麼？我有一個類似的測試，你的代碼應該可以正常工作。下面是我如何做到這一點：

from django.contrib.auth.models import User 
from django.test import TestCase 
from django.test.client import Client 


class UserTestCase(TestCase): 

    def setUp(self): 
     self.client = Client() 

    def testLogin(self): 
     print User.objects.all() # returns [] 
     response = self.client.post(reverse('auth-registration'), 
          { 'username':'foo', 
           'password1':'bar', 
           'password2':'bar' }) 
     print User.objects.all() # returns one user 
     print User.objects.all()[0].is_authenticated() # returns True 

編輯

如果我註釋掉我的登錄邏輯，我不self.client.post(後得到任何用戶。如果您確實想檢查用戶是否已通過身份驗證，請使用self.client訪問需要用戶身份驗證的另一個網址。從上面的，獲得持續的另一頁：

response = self.client.get(reverse('another-page-which-requires-authentication')) 
print response.status_code 

上面應該返回200，以確認用戶身份驗證。除此之外，它將使用302代碼重定向到登錄頁面。

Answer 4

Django的TestClient的有login method如果用戶成功登錄返回true。

Answer 5

可以使用auth模塊的get_user方法。它表示它想要一個請求作爲參數，但它只使用請求的session屬性。而且恰巧我們的Client有這個屬性。

from django.contrib import auth 
user = auth.get_user(self.client) 
assert user.is_authenticated() 

Answer 6

還有另一種簡潔的方式，使用在wsgi_requestresponse：

response = self.client.post('/signup', data) 
assert response.wsgi_request.user.is_authenticated() 

和@Chronial的方式也可提供wsgi_request：

from django.contrib import auth 
user = auth.get_user(response.wsgi_request) 
assert user.is_authenticated() 

因爲response.wsgi_request對象具有session屬性。我覺得使用response.wsgi_request.user更簡單。