用补丁模拟两个功能进行单元测试

我有一个要单元测试的功能，其中包含调用其他两个功能的功能。我不确定如何使用补丁同时模拟两个函数。我在下面提供了我的意思的示例。当我进行鼻子测试时，测试通过了，但是我觉得必须有一种更清洁的方法来进行测试，而我实际上并不了解有关f.close（）的内容。

目录结构如下所示：

program/
  program/
    data.py
  tests/
    data_test.py

data.py：

import cPickle

def write_out(file_path, data):
    f = open(file_path, 'wb')
    cPickle.dump(data, f)
    f.close()

data_test.py：

from mock import MagicMock, patch

def test_write_out():
    path = '~/collection'
    mock_open = MagicMock()
    mock_pickle = MagicMock()
    f_mock = MagicMock()
    with patch('__builtin__.open', mock_open):
        f = mock_open.return_value
        f.method.return_value = path
        with patch('cPickle.dump', mock_pickle):
            write_out(path, 'data')
            mock_open.assert_called_once_with('~/collection', 'wb')
            f.close.assert_any_call()
            mock_pickle.assert_called_once_with('data', f)

结果：

$ nosetests
.
----------------------------------------------------------------------
Ran 1 test in 0.008s
OK

您可以通过使用补丁装饰器并像这样嵌套它们来简化测试（MagicMock默认情况下它们是对象）：

@patch('cPickle.dump')
@patch('__builtin__.open')
def test_write_out(mock_open, mock_pickle):
    path = '~/collection'
    f = mock_open.return_value
    f.method.return_value = path

    write_out(path, 'data')

    mock_open.assert_called_once_with('~/collection', 'wb')
    mock_pickle.assert_called_once_with('data', f)
    f.close.assert_any_call()

调用MagicMock实例将返回一个新MagicMock实例，因此您可以像检查其他任何模拟对象一样检查返回的值是否已被调用。在这种情况下f，MagicMock命名为'open()'（尝试打印f）。

除了响应@Matti John之外，您还可以使用patchinside函数test_write_out：

from mock import MagicMock, patch

def test_write_out():
    path = '~/collection'
    with patch('__builtin__.open') as mock_open, \
            patch('cPickle.dump') as mock_pickle:

        f = mock_open.return_value
        ...

这里有一个简单的例子，如何提高测试ConflictError中create_collection使用模拟功能：

import os
from unittest import TestCase
from mock import patch
from ..program.data import ConflictError, create_collection


class TestCreateCollection(TestCase):
    def test_path_exists(self):
        with patch.object(os.path, 'exists') as mock_method:
            mock_method.return_value = True

            self.assertRaises(ConflictError, create_collection, 'test')

请也请参阅模拟文档和Michael Foord对模拟的精彩介绍。