python - 如何在 Python 单元测试中模拟 __version__

Question

在我的 release_exporter/version.py 中，我有 __version__ = '1.0.6' 并且我想模拟它的 stdout。 check_version() 将最新版本打印到 stdout。

我通过做模拟stdout

@patch('sys.stdout', new_callable=io.StringIO)
def assert_stdout_1(self, n, expected_output, mock_stdout):
    request = check_version()
    self.assertIn(expected_output, mock_stdout.getvalue())

效果很好。

我试着做这样的事情来模拟版本号:

def test_check_version_pass(self):
    version.__version__ = '1'

    self.assert_stdout_1('', 'New version')

我知道这是错误的，因为我不是在 mock 它(尽管它工作正常)。然后我试着做

@patch.object(version.__version__, 'version.__version__', return_value='1')
def test_check_version_pass(self):

    self.assert_stdout_1('', 'New version')

但我得到一个错误，因为 AttributeError: 1.0.6 does not have the attribute 'version.__version__'。我也尝试使用 @patch('release_exporter.version.__version__', return_value='1') 但我得到一个错误，因为 TypeError: test_check_version_pass() takes 1 positional argument but 2 were given 。

我不知道我做错了什么。帮助!

Answer 1

好的。我知道了。使用 @patch.object() 装饰器时，请始终确保您正在使用它的方法有一个参数。例如在我的例子中(感谢 @Sraw ):

@patch.object(version, '__version__', return_value='1')
def test_check_version_pass(self, n):

    self.assert_stdout_1('', 'New version')

在这种情况下，我使用了任意参数 n。