如何对pywin32文件转换操作实现快速健壮的单元测试？

问题背景 #

我正在开发一个小型pywin32程序，可将文本文件转换为docx、pdf格式，当前功能运行稳定。但我尚未找到无需创建临时文件即可测试该转换器的单元测试方案，现有测试流程执行速度很慢。请问针对文件操作的单元测试，是否有行业公认的最佳实践？

原有代码 #

converter.py #

import win32com.client as win32


class Converter:
    def __init__(self):
        self.word = win32.Dispatch('Word.Application')
        self.word.Visible = False
        self._word_format = {
            'docx': 12,
            'pdf': 17
        }

    @property
    def word_format(self):
        return self._word_format

    def open_and_close(func):
        def wrapper(self, fpath):
            doc = self.word.Documents.Open(str(fpath))
            doc.Activate()
            func(self, fpath)
            doc.Close()
        return wrapper

    @open_and_close
    def to_docx(self, fpath):
        new_fpath = str(fpath).replace('.doc', '.docx')
        self.word.ActiveDocument.SaveAs(new_fpath, FileFormat=self.word_format['docx'])

    @open_and_close
    def to_pdf(self, fpath):
        new_fpath = str(fpath).replace('.doc', '.pdf')
        self.word.ActiveDocument.SaveAs(new_fpath, FileFormat=self.word_format['pdf'])

    def __del__(self):
        self.word.Quit()

test_converter.py（原有测试代码） #

import unittest
import tempfile
import os
from pathlib import Path

from converter import Converter


class TestConverterToDocx(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.converter = Converter()

    def setUp(self):
        self.tempdir = tempfile.TemporaryDirectory()
        for _ in range(5):
            tempfile.NamedTemporaryFile(dir=self.tempdir.name,
                                        suffix='.doc', delete=False)

    def tearDown(self):
        self.tempdir.cleanup()

    def test_to_docx(self):
        for root, dirs, files in os.walk(self.tempdir.name):
            for file in files:
                self.converter.to_docx(str(Path(root) / file))
        files = [file for file in os.listdir(self.tempdir.name)
                 if file.endswith('.docx')]
        self.assertEqual(len(files), 5)

    def test_to_pdf(self):
        for root, dirs, files in os.walk(self.tempdir.name):
            for file in files:
                self.converter.to_pdf(str(Path(root) / file))
        files = [file for file in os.listdir(self.tempdir.name)
                 if file.endswith('.pdf')]
        self.assertEqual(len(files), 5)

行业通用最佳实践 #

1. 明确区分单元测试与集成测试 #

你当前编写的属于集成测试，需要真实调用Word COM组件、读写磁盘文件，速度慢是这类测试的固有特性。单元测试的核心原则是隔离所有外部依赖，仅验证代码本身的逻辑正确性，不需要依赖真实的外部组件和文件系统。

2. 用Mock技术实现无依赖单元测试 #

使用Python标准库自带的unittest.mock对外部依赖打桩，不需要启动Word进程、也不需要创建真实文件，即可验证逻辑正确性：

• 打桩win32.Dispatch返回的Word对象，模拟所有COM调用行为
• 验证文件打开路径、保存路径、格式参数是否符合预期
• 完全脱离IO和外部组件，执行速度可提升数十倍

3. 精简集成测试用例 #

保留少量集成测试用例做全链路验证即可：

• 仅使用1-2个有效小体积doc文件作为测试样本，不需要生成5个空测试文件
• 临时目录可在测试类级别复用，不需要每个用例都创建销毁，减少IO开销

优化后的单元测试示例 #

import unittest
from unittest.mock import patch, MagicMock
from converter import Converter

class TestConverterUnit(unittest.TestCase):
    @patch('win32com.client.Dispatch')
    def setUp(self, mock_dispatch):
        self.mock_word = MagicMock()
        mock_dispatch.return_value = self.mock_word
        self.converter = Converter()
    
    def test_to_docx_logic(self):
        test_input = "/test_dir/sample.doc"
        expect_output = "/test_dir/sample.docx"
        expect_format = 12

        self.converter.to_docx(test_input)

        # 验证文件打开参数正确
        self.mock_word.Documents.Open.assert_called_once_with(test_input)
        # 验证保存参数正确
        self.mock_word.ActiveDocument.SaveAs.assert_called_once_with(
            expect_output, FileFormat=expect_format
        )

    def test_to_pdf_logic(self):
        test_input = "/test_dir/sample.doc"
        expect_output = "/test_dir/sample.pdf"
        expect_format = 17

        self.converter.to_pdf(test_input)

        self.mock_word.Documents.Open.assert_called_once_with(test_input)
        self.mock_word.ActiveDocument.SaveAs.assert_called_once_with(
            expect_output, FileFormat=expect_format
        )

内容的提问来源于stack exchange，提问作者adamlepkowski