模拟连续呼叫

问题描述

我有一些Python代码，其中我连续调用with open("f1") as f1: d1 = json.load(f1) # some_more_logic_here with open("f2") as f2: d2 = json.load(f2) 上下文管理器。

mock_open(read_data=json.dumps(some_dictionary)

我不确定如何模拟这些背靠背的电话。我可以用com.xxx.yyy.someClass$$Lambda$12345/7654321@qwert来模拟第一次调用，但是不能使用与第一个不同的对象来模拟第二个上下文管理器调用。任何帮助将不胜感激。

解决方法

我无法获得mock_open的控球权。但是使用低级模拟很容易：

from unittest.mock import patch,MagicMock

from yourmodule import yourfunc


def test_yourfunc():
    first = MagicMock()
    second = MagicMock()
    first.__enter__.return_value.read.return_value = '{"k1": "v1"}'
    second.__enter__.return_value.read.return_value = '{"k2": "v2"}'
    with patch("yourmodule.open",side_effect=[first,second]):
        d1,d2 = yourfunc()
    assert d1 == {"k1": "v1"}
    assert d2 == {"k2": "v2"}