Python Mock - 模拟几个打开的

Question

读完这篇文章后：如何模拟 with 语句中使用的 open （使用 Python 中的 Mock 框架）？

我'我能够使用以下命令模拟 python 中的 open 函数：

with patch(open_name, create=True) as mock_open:
    mock_open.return_value = MagicMock(spec=file)
    m_file = mock_open.return_value.__enter__.return_value
    m_file.read.return_value = 'text1'

    diffman = Diffman()
    diffman.diff(path1, path2)

当我的测试方法使用一个 open 语句时，它效果很好。这是我测试的方法：

def diff(self, a, b):
    with open(a, 'r') as old:
        with open(b, 'r') as new:
            oldtext = old.read()
            newtext = new.read()

oldtext 和 newtext 的值相同（此处为“text1”）。

我希望旧文本为“text1”，新文本为“text2”。

我该怎么做？

Answer 1

这是获得您想要的东西的快速方法。它有点作弊，因为被测试方法中的两个文件对象是同一个对象，我们只是在每次读取后更改读取调用的返回值。如果您希望文件对象不同，您可以在多个层中使用相同的技术，但这会非常混乱，并且可能不必要地掩盖测试的意图。

将此行： 替换

    m_file.read.return_value = 'text1'

为：

    reads = ['text1', 'text2']
    m_file.read.side_effect = lambda: reads.pop(0)

Answer 2

也许一个好的解决方案就是以一种更适合轻松测试的方式编写代码。在“diff”的情况下，让 diff 将已打开的文件对象作为参数似乎很容易（无可否认，没有太多其他上下文）。这可能是在代码中进行的一个相当小的更改，并且使测试变得非常容易，因为当您测试它时，您可以轻松地向 diff() 提供模拟文件对象，而不是尝试跳过模拟同一文件的两个实例内置函数作为上下文管理器在...本身...或其他东西内调用；-)

import StringIO

diff(a, b):
  oldtext = a.read()
  newtext = b.read()

def test_diff():
  a = StringIO.StringIO('text1')
  b = StringIO.StringIO('text2')

  res = diff(a, b)
  <some assertion here>

这适合您的情况吗？

Answer 3

刚刚找到另一个解决方案：

open_mock1 = mock.mock_open(read_data="file1 data")
open_mock2 = mock.mock_open(read_data="file2 data")

def open_mock_side(file_name, *args, **kwargs):
    # can be refactored to use a dict
    if file_name.endswith("folder/file1.txt"):
        return open_mock1()
    if file_name.endswith("folder/file2.txt"):
        return open_mock2()

open_mock = mock.mock_open()
open_mock.side_effect = open_mock_side
open_mock.return_value = None

with mock.patch('package.of.function_to_test.open', open_mock):
     function_to_test(...)