在python中使用不同的值模拟两次方法

Question

我必须模拟下面显示的方法：

实际python方法

import json
def some_method(self):
    output_one = json.loads(varone)
    output_two = json.loads(vartwo)

测试方法

import json
self.stubs = stubout.StubOutForTesting()
self.stubs.Set(json, "loads", lambda *a: output_one)
self.stubs.Set(json, "loads", lambda *a: output_two)

结果是只有 output_two ，因为 output_one 被覆盖。 我应该如何模拟一个方法两次并且每次都期望不同的输出。

Answer 1

使用side_effect。

docs中的示例：

>>> mock = Mock()
>>> mock.side_effect = [3, 2, 1]
>>> mock(), mock(), mock()
(3, 2, 1)

Answer 2

您需要将代码移动到两个测试函数中。

def test_output_one():
    self.stubs = stubout.StubOutForTesting()
    self.stubs.Set(json, "loads", lambda *a: output_one)
    ... code for the first test here ...

def test_output_two():
    self.stubs = stubout.StubOutForTesting()
    self.stubs.Set(json, "loads", lambda *a: output_two)
    ... code for the second test here ...