基于输入参数模拟python函数

我们在python中使用Mock已经有一段时间了。

现在，我们有这样一种情况，我们想模拟一个函数

def foo(self, my_param):
    #do something here, assign something to my_result
    return my_result

通常，模拟的方法是(假设foo是对象的一部分)

self.foo = MagicMock(return_value="mocked!")

甚至，如果我调用foo()几次，我可以使用

self.foo = MagicMock(side_effect=["mocked once", "mocked twice!"])

现在，我面临这样一种情况:当输入参数具有特定值时，我想返回一个固定值。如果my_param等于something那么我要返回my_cool_mock

这似乎在python的mockito上可用

when(dummy).foo("something").thenReturn("my_cool_mock")

我一直在寻找如何实现同样的Mock没有成功?

什么好主意吗?

当前回答

Side effect接受一个函数(也可以是lambda函数)，所以对于简单的情况，你可以使用:

m = MagicMock(side_effect=(lambda x: x+1))

2016-08-30 05:19:38

其他回答

Side effect接受一个函数(也可以是lambda函数)，所以对于简单的情况，你可以使用:

m = MagicMock(side_effect=(lambda x: x+1))

2016-08-30 05:19:38

为了展示另一种方法:

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

2016-12-08 21:08:13

如果你想使用一个带参数的函数，而你要模拟的函数不带参数，你也可以使用来自functools的partial。例如:

def mock_year(year):
    return datetime.datetime(year, 11, 28, tzinfo=timezone.utc)

@patch('django.utils.timezone.now', side_effect=partial(mock_year, year=2020))

这将返回一个不接受参数的可调用对象(如Django的timezone.now())，但我的mock_year函数可以。

2019-09-02 07:38:25

如在多次调用方法的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

这就成功了

2013-04-23 06:19:43

你也可以使用@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`

2019-06-13 19:40:50

基于输入参数模拟python函数

推荐文章

最新文章

标签