使用Mock Python返回Iterable

Question

我正在尝试使用Mock来模拟python中的一个函数.这是我的代码:

    resp, content = request(...)

request()函数需要返回两个值.这是我试过的:

    with patch("syncdatetime.py") as sync_mock:
        sync_mock.request.return_value = [obj, '']

但是当我运行测试时,我得到错误"模拟对象不可迭代".请求函数返回Mock类型的对象而不是列表.如何修改请求函数以便返回列表？

Answer 1

我怀疑你的问题是你没有使用你认为自己的模拟实例.默认情况下,Mock实例在调用时返回Mock.

>>> m = mock.Mock()
>>> type(m())
<class 'mock.mock.Mock'>    

看起来你的调用request是返回一个Mock,因为return_value尚未初始化,这意味着resp, content = request()正在尝试解压缩一个Mock对象.

>>> m = mock.Mock()
>>> (a, b) = m()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: 'Mock' object is not iterable

您不应该使用side_effect返回列表.分配return_value应该有效.

>>> m = mock.Mock()
>>> m.return_value = ['a', 'b']
>>> (a, b) = m()
>>> a
'a'
>>> b
'b'

Answer 2

请注意披露,我是新来的模拟,所以我不是专家,但是,我刚遇到同样的问题,发现将side_effect属性设置为返回数组的函数修复了一些事情

从您的示例代码中,更改:

with patch("syncdatetime.py") as sync_mock:
    sync_mock.request.return_value = [obj, '']

至

with patch("syncdatetime.py") as sync_mock:
    sync_mock.request.side_effect = function_returning_list

并定义

def function_returning_list(arg_list_of_choice):
    #anything you want to do goes here
    #then
    return your_list