如何用python unittest模拟mock对象里面的列表？

Question

我试图从Mock对象返回一个列表，但无法让它工作。

那是代码：

def execute_call(model, payload):
    result = model.execute(payload)
    code = result[0]

这是我试过的测试：

def test_get_code(self, session):
    execute_call = Mock()
    execute_call.return_value = ['123', '123', '123']
    session.return_value = Mock(execute=execute_call)

结果：

code = result[0]
TypeError: 'Mock' object does not support indexing

我无法弄清楚如何使结果[0]起作用。 'execute_call'应该是mock，但return_value需要是一个列表。

任何想法都表示赞赏。

Answer 1

def execute_call(model, payload):
    result = model.execute(payload)
    code = result[0]

def test_execute_call(self):
    mock_model = mock.MagicMock()
    mock_model.execute = mock.MagicMock(return_value=(a,b,c)
    self.execute_call(mock_model, payload)
    assert whatever

您需要将模拟对象传递给函数，以便可以使用它。