在Python unittest中断言执行顺序

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()

现在我已经模拟了每个函数，我可以确认它们被调用了，但无法确认它们的调用顺序。我该怎么做呢？

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()])