Question

让我们说我想测试这个非常复杂的函数：

def func(hostname, username, password):
    ftp = FTP(hostname, username, password)
    ftp.retrbinary('RETR README', open('README', 'wb').write)

其中一项测试将是：

@patch('FTP')
def test_func_happy_path():
    mock_ftp = Mock()
    mock_ftp.retrbinary = Mock()
    MockFTP.return_value = mock_ftp()
    func('localhost', 'fred', 's3Kr3t')
    assert mock_ftp.retrbinary.called

然而，这将创建一个名为README的本地文件，我显然不想要它。

有没有办法模拟/修补open以便不创建任何文件？

显然，作为一种解决方法，我可以确保将文件写入temporary directory，我可以将其作为参数传递给func或在func内创建并返回。

请注意，使用装饰器@patch('__builtin__.open')会引发以下期望：

self = <Mock name=u'open()' spec='FTP' id='51439824'>, name = 'write'
    def __getattr__(self, name):
        if name in ('_mock_methods', '_mock_unsafe'):
            raise AttributeError(name)
        elif self._mock_methods is not None:
            if name not in self._mock_methods or name in _all_magics:
>               raise AttributeError("Mock object has no attribute %r" % name)
AttributeError: Mock object has no attribute 'write'

我正在将回调传递给ftp.retrbinary，而不是函数调用。

Answer 1

所以，考虑到你不关心你的开放会发生什么，你可以直接嘲笑它以便停止写作。为此，您可以按照与模拟FTP类似的方法进行操作。因此，考虑到这一点，您可以设置如下测试代码：

import unittest
from mock import patch, Mock
from my_code import func

class SirTestsAlot(unittest.TestCase):

    @patch('my_code.open')
    @patch('my_code.FTP')
    def test_func_happy_path(self, MockFTP, m_open):
        MockFTP.return_value = Mock()
        mock_ftp_obj = MockFTP()

        m_open.return_value = Mock()

        func('localhost', 'fred', 's3Kr3t')

        assert mock_ftp_obj.retrbinary.called
        assert m_open.called
        # To leverage off of the other solution proposed, you can also
        # check assert called with here too
        m_open.assert_called_once_with('README', 'wb')


if __name__ == '__main__':
    unittest.main()

正如您所看到的，我们在这里所做的是，我们正在嘲笑我们正在测试的地方。因此，考虑到这一点，我们正在针对open嘲笑FTP和my_code。

现在在my_code内，没有任何改变：

from ftplib import FTP


def func(hostname, username, password):
    ftp = FTP(hostname, username, password)

    ftp.retrbinary('RETR README', open('README', 'wb').write)

运行此测试套件成功返回。

Answer 2

另一种方法涉及使用mock_open：

sudo service mongodb start

在单元测试中模拟FTP

2 个答案: