如在多次调用方法的Python Mock对象中所述

解决办法是写我自己的side_effect

def my_side_effect(*args, **kwargs):
    if args[0] == 42:
        return "Called with 42"
    elif args[0] == 43:
        return "Called with 43"
    elif kwargs['foo'] == 7:
        return "Foo is seven"

mockobj.mockmethod.side_effect = my_side_effect

这就成功了

你也可以使用@mock.patch.object:

假设my_module.py模块使用pandas从数据库中读取数据，我们想通过模拟pd来测试这个模块。Read_sql_table方法(以table_name作为参数)。

你能做的是(在你的测试中)创建一个db_mock方法，根据提供的参数返回不同的对象:

def db_mock(**kwargs):
    if kwargs['table_name'] == 'table_1':
        # return some DataFrame
    elif kwargs['table_name'] == 'table_2':
        # return some other DataFrame

在你的测试函数中，你可以这样做:

import my_module as my_module_imported

@mock.patch.object(my_module_imported.pd, "read_sql_table", new_callable=lambda: db_mock)
def test_my_module(mock_read_sql_table):
    # You can now test any methods from `my_module`, e.g. `foo` and any call this 
    # method does to `read_sql_table` will be mocked by `db_mock`, e.g.
    ret = my_module_imported.foo(table_name='table_1')
    # `ret` is some DataFrame returned by `db_mock`

如在多次调用方法的Python Mock对象中所述

解决办法是写我自己的side_effect

def my_side_effect(*args, **kwargs):
    if args[0] == 42:
        return "Called with 42"
    elif args[0] == 43:
        return "Called with 43"
    elif kwargs['foo'] == 7:
        return "Foo is seven"

mockobj.mockmethod.side_effect = my_side_effect

这就成功了

如果side_effect_func是一个函数，则该函数返回的是 什么调用模拟返回。调用side_effect_func函数 和模拟的论点一样。这允许你改变回报 基于输入动态调用的值: >>>定义side_effect_func(value): …返回值+ 1 .．. >>> m = MagicMock(side_effect=side_effect_func) > > > m (1) 2 > > > m (2) 3. > > > m.mock_calls ((1)打电话,电话(2))

http://www.voidspace.org.uk/python/mock/mock.html#calling

虽然side_effect可以实现目标，但是为每个测试用例设置side_effect函数并不方便。

我写了一个轻量级的Mock(称为NextMock)来增强内置Mock来解决这个问题，这里是一个简单的例子:

from nextmock import Mock

m = Mock()

m.with_args(1, 2, 3).returns(123)

assert m(1, 2, 3) == 123
assert m(3, 2, 1) != 123

它还支持参数匹配器:

from nextmock import Arg, Mock

m = Mock()

m.with_args(1, 2, Arg.Any).returns(123)

assert m(1, 2, 1) == 123
assert m(1, 2, "123") == 123

希望这个包能让测试更愉快。请随时提供任何反馈。

为了展示另一种方法:

def mock_isdir(path):
    return path in ['/var/log', '/var/log/apache2', '/var/log/tomcat']

with mock.patch('os.path.isdir') as os_path_isdir:
    os_path_isdir.side_effect = mock_isdir