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) 

这是它为我工作:)

谢谢所有。干杯，

何塞