如何模拟 Python 中的列表？

Question

例如，我有一些尝试访问列表的代码：

def some_code():
    script_dir = os.path.dirname(sys.argv[0])

我需要模拟 sys.argv[0]。所以我将 @patch.object 添加到我的测试中：

import os
import sys

THIS_DIR = os.path.dirname(os.path.abspath(__file__))
@mock.patch.object(sys, 'argv', mock.Mock(return_value=[THIS_DIR]))
def test_mytest():
    some_code()

但这不起作用。 Pytest 引发错误：

>     script_dir = os.path.dirname(sys.argv[0])
E     TypeError: 'Mock' object does not support indexing

我做错了什么？

Answer 1

不要使用 Mock 对象来替换列表，只需将 sys.argv 替换为不同的列表即可。您仍然可以使用 mock.patch 来管理替换和撤消操作：

@mock.patch.object(sys, 'argv', [THIS_DIR])

请注意，您可能希望传递文件名，而不是目录（除非您的测试目录嵌套在您要使用的实际目录下）。

您的模拟配置将 sys.argv 替换为可调用对象，因此只有 sys.argv() 会生成您的 [THIS_DIR] 列表。

演示：

>>> import sys
>>> from unittest import mock
>>> sys.argv
['']
>>> with mock.patch.object(sys, 'argv', ['foo', 'bar', 'baz']):
...     sys.argv
...
['foo', 'bar', 'baz']
>>> sys.argv
['']

如何模拟 Python 中的列表？

How to mock a list in Python?

python

unit-testing

mocking

pytest

python-mock