在 django 测试中使用 mock 绕过装饰器

Question

我正在尝试编写一个简单的测试，但是我的视图用嵌套的 user_passes_test 语句装饰。他们检查诸如条带订阅和 is_authenticated 之类的内容。我发现了各种帖子，例如this，其中解决了如何绕过带有补丁的装饰器，但我不太清楚如何将所有东西集成在一起。

tests.py

@patch('dashboard.views.authorised_base_user_checks', lambda func: func)
def test_dashboard_root_exists(self):
    response = self.client.get('/dashboard/')
    self.assertEqual(200, response.status_code)

视图中的装饰器

def authorised_base_user_checks(view_func):
    decorated_view_func = login_required(user_active(subscriber_exists(subscriber_valid(view_func))))
    return decorated_view_func

views.py

@authorised_base_user_checks
def IndexView(request):
    ...

上面还是没有通过装饰器。

谢谢！

Answer 1

这种修补装饰器的方法很可能不起作用，因为views 模块的导入发生在修补之后。如果 view 已经被导入，那么装饰器已经被应用到 IndexView 并且修补装饰器功能将完全没有效果。

您可以重新加载视图模块来克服这个问题：

import imp
import dashboard.views

@patch('dashboard.views.authorised_base_user_checks', lambda func: func)
def test_dashboard_root_exists(self):
   # reload module to make sure view is decorated with patched decorator
   imp.reload(views)

   response = self.client.get('/dashboard/')
   self.assertEqual(200, response.status_code)

   # reload again
   patch.stopall()
   imp.reload(views)

免责声明：此代码仅演示该想法。您需要确保stopall 和最终重新加载总是发生，所以它们应该在finally 或tearDown 中。