python - 测试 Django View 时模拟模块

Question

我有一个自定义函数来验证请求。我试图在测试期间模拟这个模块，但到目前为止运气不好

这是我的观点

from auth_utils import authenticate_request, UnauthenticatedRequest

def my_view(request):
    try:
        authenticate_request(request)
    except UnauthenticatedRequest:
        return Http404()
    return render(request, 'ok.html', {'status': 'ok'})

在测试中我试图模拟 authenticate_request 所以它不会引发错误

class TestMyView(MyAPITestCase, TestCase):

    @mock.patch('auth_utils.authenticate_request', side_effect=None)
    def setUp(self, mock_auth):
        self.response = self.client.get(reverse('my-view'))

    def test_should_return_ok(self):
        self.assertEqual(self.response.context.get('status'), 'ok')

无法让它工作。有什么建议么 ？

谢谢，Python 2.7，Django 1.8。

Answer 1

您需要在导入的地方模拟函数 authenticate_request，而不是定义的地方。

因此，例如，如果 my_view 定义在 myapp/views.py 中，则 authenticate_request 被导入到 myapp 中。意见。所以你想调用如下内容:

@mock.patch('myapp.views.authenticate_request', side_effect=None)