Django測試可以避免permission_required裝飾器嗎？

Question

我正在測試一系列廣泛使用'permission_required'裝飾器的Django應用程序。這在我擁有的大多數視圖中以302 HTTP響應派生。

我的問題是：有什麼辦法可以避免或停用測試中的'permission_required'，所以當我打電話給我的意見時，我可以得到200響應，而不是302？

謝謝！

Answer 1

具有超級用戶只需登錄你的測試案例設置方法

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

class TestThatNeedsLogin(TestCase): 
    def setUp(self): 
     User.objects.create_superuser(
      'user1', 
      'user1@example.com', 
      'pswd', 
     ) 
     self.client.login(username="user1", password="pswd") 

    def tearDown(self): 
     self.client.logout() 

    def test_something(self): 
     response = self.client.get("/") 
     self.assertEqual(200, response.status_code) 

Answer 2

你可以猴子打補丁：

import django.contrib.auth.decorators 

real_permission_required = decorators.permission_required 
# return a function that returns the exact function that was decorated, ignoring arguments 
decorators.permission_required = lambda *args, **kwargs: lambda func: func 

你需要它的使用之前，也就是在定義時它的裝飾物的，以確保這種情況發生。 （例如，當包含該模塊時）。

它也必須在它被反彈到另一個範圍之前發生。在import django.contrib.auth.decorators之後沒問題，但在from django.contrib.auth.decorators import permission_required之前。

Answer 3

嘛。我找到的解決方案是在TestCase類的setUp方法中創建一個超級用戶。我做了那樣：

def setUp(self): 
    self.client = Client() 
    self.user = User.objects.create_superuser(
     'testuser', 
     'test@example.com', 
     'easy_password', 
    ) 

然後，當我想測試一個URL，我這樣做：

def test_search_customers(self): 
    url = reverse('customer_search') 
    response = self.client.get(url) 
    # Not logged user. Must return a 302 HTTP code. 
    self.assertEquals(response.status_code, 302) 
    self.assertEquals(response['Location'], 'http://testserver/unauthorized/?next=/search/customers/') 
    # HERE I LOG IN MY SUPERUSER 
    self.client.login(username='testuser', password='easy_password') 
    response = self.client.get(url, follow=True) 
    # Same URL requested with a logged user with permissions. Must return 200 HTTP code. 
    self.assertEquals(response.status_code, 200) 

這是它爲我工作:)

謝謝所有。乾杯，

何塞