使unittest.mock.Mock（）返回字典列表

Question

I'm trying to mock a call of method extra_get() which usually returns a list of dicts. As far as I understand from the Mock docs , if I want to return iterable, I should set side_effect param.

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

But then the following code calls that mocked method:

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

And the dict comprehention fails because extra there is not a list, but dict {'foo': 'bar'} . What I'm doing wrong? How can I make Mock method return a list of dicts?

Answer 1

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