71

我有一个正在测试的类，它具有另一个类的依赖项（其实例被传递给 CUT 的 init 方法）。我想使用 Python Mock 库来模拟这个类。

我所拥有的是这样的：

mockobj = Mock(spec=MyDependencyClass)
mockobj.methodfromdepclass.return_value = "the value I want the mock to return"
assertTrue(mockobj.methodfromdepclass(42), "the value I want the mock to return")

cutobj = ClassUnderTest(mockobj)

这很好，但是“methodfromdepclass”是一个参数化方法，因此我想创建一个模拟对象，根据传递给 methodfromdepclass 的参数，它返回不同的值。

我想要这种参数化行为的原因是我想创建包含不同值的 ClassUnderTest 的多个实例（其值由从 mockobj 返回的内容产生）。

有点像我在想的（这当然行不通）：

mockobj = Mock(spec=MyDependencyClass)
mockobj.methodfromdepclass.ifcalledwith(42).return_value = "you called me with arg 42"
mockobj.methodfromdepclass.ifcalledwith(99).return_value = "you called me with arg 99"

assertTrue(mockobj.methodfromdepclass(42), "you called me with arg 42")
assertTrue(mockobj.methodfromdepclass(99), "you called me with arg 99")

cutinst1 = ClassUnderTest(mockobj, 42)
cutinst2 = ClassUnderTest(mockobj, 99)

# now cutinst1 & cutinst2 contain different values

我如何实现这种“ifcallwith”的语义？

4

4 回答 4

97

尝试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
于 2011-10-05T18:15:04.587 回答
60

甜一点：

mockobj.method.side_effect = lambda x: {123: 100, 234: 10000}[x]

或多个参数：

mockobj.method.side_effect = lambda *x: {(123, 234): 100, (234, 345): 10000}[x]

或使用默认值：

mockobj.method.side_effect = lambda x: {123: 100, 234: 10000}.get(x, 20000)

或两者的组合：

mockobj.method.side_effect = lambda *x: {(123, 234): 100, (234, 345): 10000}.get(x, 20000)

我们高高兴兴地去。

于 2012-11-29T23:04:56.597 回答
11

我在做自己的测试时遇到了这个问题。如果您不关心捕获对 methodfromdepclass() 的调用，而只需要它返回某些内容，那么以下内容可能就足够了：

def makeFakeMethod(mapping={}):
    def fakeMethod(inputParam):
        return mapping[inputParam] if inputParam in mapping else MagicMock()
    return fakeMethod

mapping = {42:"Called with 42", 59:"Called with 59"}
mockobj.methodfromdepclass = makeFakeMethod(mapping)

这是一个参数化版本：

def makeFakeMethod():
    def fakeMethod(param):
        return "Called with " + str(param)
    return fakeMethod
于 2012-06-03T07:58:02.727 回答
2

和这里一样，除了side_effect在unittest.mock.Mock中使用之外，您还可以使用@mock.patch.objectwith new_callable，它允许您使用模拟对象修补对象的属性。

假设一个模块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-13T22:54:06.770 回答