在单元测试中模拟FTP

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以便不创建任何文件？

显然，作为一种变通方法，我可以确保将文件写入一个临时目录 ，该目录可以作为参数传递给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()

如您所见，我们在这里所做的是关于测试位置的嘲笑。 因此，考虑到这一点，我们正在针对my_code出open和FTP 。

现在在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 ：

from unittest.mock import patch, mock_open
import ftplib


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


@patch('ftplib.FTP')
def test_func_happy_path(MockFTP):
    mock_ftp = MockFTP.return_value  # returns another `MagicMock`
    with patch('__main__.open', mock_open(), create=True) as m:
        func('localhost', 'fred', 's3Kr3t')

    assert mock_ftp.retrbinary.called
    m.assert_called_once_with('README', 'wb')


test_func_happy_path()