如何让Python Mock先返回两次False，后续调用均返回True？

你遇到的StopIteration错误是因为：当给side_effect赋值为可迭代对象时，mock会依次返回该对象中的元素，一旦元素耗尽就会抛出迭代结束的异常；同时，只要设置了side_effect，return_value的配置会被忽略，所以后续调用找不到返回值就触发了报错。

下面提供两种可行的解决方案：

方法一：自定义side_effect函数（灵活控制返回逻辑） #

通过一个计数器记录方法的调用次数，根据次数返回对应的值：

def test_moc(self):
    from unittest import mock

    class A:
        def __init__(self):
            self.value = False
        def getValue(self):
            return self.value

    call_count = 0
    def get_value_side_effect():
        nonlocal call_count
        call_count += 1
        return False if call_count <= 2 else True

    # 使用spec参数确保mock符合A类的接口规范
    m = mock.Mock(spec=A)
    m.getValue.side_effect = get_value_side_effect

    self.assertFalse(m.getValue())
    self.assertFalse(m.getValue())
    self.assertTrue(m.getValue())
    self.assertTrue(m.getValue())
    self.assertTrue(m.getValue())

方法二：使用itertools拼接迭代器（简洁实现） #

利用itertools.chain和itertools.repeat生成一个永不耗尽的迭代器，先返回两次False，之后无限返回True：

def test_moc(self):
    from unittest import mock
    import itertools

    class A:
        def __init__(self):
            self.value = False
        def getValue(self):
            return self.value

    m = mock.Mock(spec=A)
    # 拼接两个迭代器：先取2个False，之后一直取True
    m.getValue.side_effect = itertools.chain([False, False], itertools.repeat(True))

    self.assertFalse(m.getValue())
    self.assertFalse(m.getValue())
    self.assertTrue(m.getValue())
    self.assertTrue(m.getValue())
    self.assertTrue(m.getValue())

内容的提问来源于stack exchange，提问作者CiaranWelsh