如何测试创建目录的函数？

Question

如何在 Python 中测试创建目录的函数？可以举个例子make_directory_test()函数吗？

def make_directory(directory):
    if not os.path.exists(directory):
        os.makedirs(directory)

Answer 1

只需测试该目录是否存在

def test_make_directory(self):
    directory_path = '...' # somepath

    make_directory(directory_path)

    self.assertTrue(os.path.exists(directory_path))

并确保在tearDown或setUp中的每个测试后删除单元测试中创建的所有目录，以确保测试的独立性

Answer 2

这里的一个好的做法是学习如何使用测试框架，如 mock 或 flexmock

此外，您应该使用 Python 的 unittest 框架。

您的目标不一定是确保创建某些东西，而是确保调用它。所以你想要做的是模拟外部调用来帮助测试你的方法的正确流程。所以在这种情况下你应该模拟存在和makedirs。然后确保它被调用。那将是一个可接受的单元测试。你会做这样的事情：

让我们假设您的主要代码位于一个名为 my_module.py 的模块中。因此，您可能想创建一个测试文件，我们将其命名为 test.py 以使其简单，并在您的单元测试中执行以下操作：

from mock import patch
import my_module
import unittest


class MyTest(unittest.TestCase):
    @patch('my_module.exists')
    @patch('my_module.makedirs')
    def test_create_dir(self, mock_make_dirs, mock_exists):
        mock_exists.return_value = True

        make_directory('thing_to_create')

        mock_make_dirs.assert_called_with('thing_to_create')

所以，这里发生的事情是你告诉模拟你想模拟出 makedirs 并且存在。您正在使用 mock_exists.return_value 指定的 True 返回存在。然后您进行实际调用，您的模拟将在该方法调用中生效。最后一行mock_make_dirs.assert_called_with 将确保您的创建目录的方法将被实际调用。

如果您仍想测试是否确实创建了某些东西

在这种情况下，您可以做的可能是冒险使用上下文管理器并创建一个临时文件夹进行测试，完成您的工作以测试您的方法以及您必须做的任何其他事情，一旦您完成完成你的工作后，上下文管理器会自行销毁。

有关上下文管理器的信息，请查看contextlib