python - 使unittest.mock.Mock()返回字典列表

Question

我正在尝试模拟方法 extra_get() 的调用，该方法通常返回一个字典列表。据我从模拟docs了解，如果我想返回iterable，我应该设置side_effect参数。

client.extra_get = mock.Mock(
            **{'side_effect': [{'foo': 'bar'}]})

但随后以下代码调用该模拟方法:

extra = client.extra_get(request, type_id)
result = {x.key: x.value for x in extra}
return result

字典理解失败，因为extra没有列表，而是字典{'foo': 'bar'}。我做错了什么？如何使 Mock 方法返回字典列表？

Answer 1

with mock.patch.object(client, 'extra_get', return_value=[{...}, {...}]) as mock_get:
  # fill in the rest