Question

我有一个创建临时目录的功能，切换到该临时目录，执行一些工作，然后切换回原始目录。我正在尝试编写一个测试它的单元测试。我没有问题验证当前目录是否已更改为临时目录并再次更改，但我在验证重要内容发生在这些调用之间时遇到了问题。

我最初的想法是将函数抽象为三个子函数，以便我可以测试调用顺序。我可以用模拟替换三个子函数中的每一个来验证它们是否被调用 - 但是，我仍然会遇到验证顺序的问题。在一个模拟器上，我可以使用assert_has_calls，但是我在什么对象上调用该函数？

以下是我试图测试的课程：

import shutil
import os
import subprocess
import tempfile
import pkg_resources


class Converter:
    def __init__(self, encoded_file_path):
        self.encoded_file_path = encoded_file_path
        self.unencoded_file_path = None
        self.original_path = None
        self.temp_dir = None

    def change_to_temp_dir(self):
        self.original_path = os.getcwd()
        self.temp_dir = tempfile.mkdtemp()
        os.chdir(self.temp_dir)

    def change_to_original_dir(self):
        os.chdir(self.original_path)
        shutil.rmtree(self.temp_dir)

    def do_stuff(self):
        pass

    def run(self):
        self.change_to_temp_dir()
        self.do_stuff()
        self.change_to_original_dir()

就我编写测试用例而言：

def test_converter(self, pkg_resources, tempfile, subprocess, os, shutil):
    encoded_file_path = Mock()

    converter = Converter(encoded_file_path)
    converter.change_to_temp_dir = Mock()
    converter.do_stuff= Mock()
    converter.change_to_original_dir = Mock()

    assert converter.encoded_file_path == encoded_file_path
    assert converter.unencoded_file_path is None

    converter.run()

现在我已经模拟了每个函数，我可以验证它们是被调用的，但不是在ORDER中。我该怎么做呢？

Answer 1

一种解决方法是创建一个单独的模拟对象，为其附加方法并使用assert_has_calls()检查调用顺序：

converter = Converter(encoded_file_path)
converter.change_to_temp_dir = Mock()
converter.do_stuff = Mock()
converter.change_to_original_dir = Mock()

m = Mock()
m.configure_mock(first=converter.change_to_temp_dir,
                 second=converter.do_stuff,
                 third=converter.change_to_original_dir)

converter.run()
m.assert_has_calls([call.first(), call.second(), call.third()])

在python unittest中断言执行顺序

1 个答案: