Django 单元测试删除在测试中创建的文件

Question

我正在使用 DRF 视图测试。 在我的测试中，我创建了要在测试运行后删除的文件。 我在我的setUp方法中使用addCleanup来删除目录。 它基本上可以工作，但是当我运行完整的单元测试集时，不会删除一个目录（我认为这是创建的第一个目录）。

以下是相关代码：

class BaseViewTest(APITestCase):
    client = APIClient()
    base_media_root = settings.MEDIA_ROOT

    def setUp(self):
        settings.MEDIA_ROOT = tempfile.mkdtemp(dir=self.base_media_root)

        # add test data
  
        self.addCleanup(shutil.rmtree, settings.MEDIA_ROOT)

Answer 1

我最终编写了自己的测试运行程序

class MyTestRunner(DiscoverRunner):

    def setup_test_environment(self, **kwargs):
        settings.MEDIA_ROOT = tempfile.mkdtemp(dir=settings.MEDIA_ROOT)
        super(MyTestRunner, self).setup_test_environment(**kwargs)

    def teardown_test_environment(self, **kwargs):
        shutil.rmtree(settings.MEDIA_ROOT)
        super(MyTestRunner, self).teardown_test_environment(**kwargs)

并在 settings.py

TEST_RUNNER = 'path.to.MyTestRunner'