mock.patch修饰符：缺少1个必需的位置参数：“self”

Question

mock.patch修饰符：缺少1个必需的位置参数：“self”

4

我正在尝试在测试方法运行期间，修补settings模块中的一个变量:

from unittest import mock

class Test(...):

    @mock.patch('settings.TARGET_SCORES_PER_SECTION', True)
    def test_register_user(self):

我遇到了这个错误：

ERROR: tests.test_user.transplant_class.<locals>.C (test_register_user)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/usr/local/Cellar/python/3.6.5_1/Frameworks/Python.framework/Versions/3.6/lib/python3.6/unittest/mock.py", line 1179, in patched
    return func(*args, **keywargs)
TypeError: test_register_user() missing 1 required positional argument: 'self'

我尝试了不同的方法，但没有找到解决方案。

我做错了什么？

这个版本运行得很好：

    def test_register_user(self):
        with mock.patch('settings.TARGET_SCORES_PER_SECTION', True):
            self._test_register_user()

    def _test_register_user(self):

我希望使用修饰器的方式能够达到同样的效果。

- warvariuc

不要忘记导入 settings 并将修补后的模拟对象添加到测试函数参数中，就像这样 def test_register_user(self, mock_target_scores):。 - progmatico

根据我的测试，你们两个的陈述都是错误的。 - warvariuc

错误有点简略。你测试了哪些代码？先检查我的答案，希望能帮到你。 - progmatico

事实上，这是错误的，因为您需要修补整个设置，而不是属性。如果您不想模拟整个设置模块，而只想模拟属性（使用PropertyMock），请参见此处获取替代方案。 - progmatico

虽然我的答案在这两种形式中都可以工作，但是我无法通过错误地更改补丁和 def 参数来重现您的确切错误消息。您有可能只是忘记了将 self 放入类方法中吗？ - progmatico

显示剩余2条评论

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- progmatico · Answer 1

如果我理解你想做什么，这将为你提供如何实现它的思路。

我已经让settings.py只保存对TARGET_SCORES_PER_SECTION赋值为False。在补丁中需要提到 __main__ 只是一个小麻烦，它不能接受非限定名称。

import unittest
from unittest import TestCase
from unittest.mock import patch
import settings


def register_user():
    return settings.TARGET_SCORES_PER_SECTION


class Test(TestCase):

    @patch('__main__.settings')
    def test_register_user(self, mock_settings):
        # setup your mock
        mock_settings.TARGET_SCORES_PER_SECTION = True
        # test your function
        self.assertTrue(register_user())

if __name__ == '__main__':
    unittest.main()

编辑：

我认为为了更好地理解OP的问题，如果您愿意，上面的代码也可以更改为以下内容：

import unittest
from unittest import TestCase
from unittest.mock import patch
import settings


def register_user():
    return settings.TARGET_SCORES_PER_SECTION


class Test(TestCase):

    @patch('settings.TARGET_SCORES_PER_SECTION', True)
    def test_register_user(self):
        self.assertTrue(register_user())

    if __name__ == '__main__':
        unittest.main()

True 是 new 参数。该对象用于修补您在修补字符串中提到的位置。这可能更加优雅，但需要您手头有一个配置好的模拟对象。