如果使用不同的属性多次调用方法，如何模拟方法？

Question

我正在为基于 Python 和 Fabric 的代码编写单元测试。 我有以下方法，它依次使用不同的参数多次调用 Fabric API 的sudo方法。 我想知道如何在模拟sudo对象上调用断言。

**main_file.py**

from fabric.api import sudo

def do_something(path_to_file_to_be_copied, destination_path):
    # remove file if already exists
    sudo('rm ' + path_to_file_to_be_copied, warn_only=True)
    sudo('cp ' + path_to_file_to_be_copied + ' ' + destination_path)

我已经编写了如下测试文件：

**test_main_file.py**

import main_file

class MainFileTest(unittest.TestCase):

    @mock.patch('main_file.sudo')
    def test_do_something(self, mock_sudo):
        file_path = '/dummy/file/path.txt'
        dest_dir = '/path/to/dest/dir'
        main_file.do_something(file_path, dest_dir)
        mock_sudo.assert_called_with('rm ' + file_path)

上面的测试失败，因为模拟对象只记得最后一次调用。 也就是说，如果我写mock_sudo.assert_called_with(cp + file_path + ' ' + dest_dir)然后测试失败。

如何断言对sudo两个调用？

Answer 1

尝试assert_any_call断言是否有任何调用，而不仅仅是最近的调用。

或者，您可以使用call_args_list来获取已调用模拟的 args 列表。

Answer 2

assert_has_calls将为您完成这项工作。

assert_has_calls ：断言已经使用指定的调用调用了模拟。 检查 mock_calls 列表中的调用。
如果 any_order 为真，则调用可以按任何顺序排列，但它们必须全部出现在 mock_calls 中。

import main_file
class MainFileTest(unittest.TestCase):

    @mock.patch('main_file.sudo')
    def test_do_something(self, mock_sudo):
        file_path = '/dummy/file/path.txt'
        dest_dir = '/path/to/dest/dir'
        main_file.do_something(file_path, dest_dir)
        mock_sudo.assert_has_calls(['rm ' + file_path,
                                    'cp' + file_path + ' ' + dest_dir],
                                     any_order=True)

Answer 3

我写了一个帮助库来自动为我生成断言。

添加这些行以在main_file.do_something(file_path, dest_dir)之后为您的案例打印正确的断言：

import mock_autogen.generator
print(mock_autogen.generator.generate_asserts(mock_sudo))

你会得到以下输出：

from mock import call

assert 2 == mock_sudo.call_count
mock_sudo.assert_has_calls(
    calls=[call('rm /dummy/file/path.txt', warn_only=True),
           call('cp /dummy/file/path.txt /path/to/dest/dir'), ])

无需再猜测，使用模拟生成器:)