python - 在每次 Django 测试之前初始化 MEDIA_ROOT

Question

我想让我的 Django 测试创建和修改媒体文件。因此，就像 Django 测试对数据库所做的一样，我想在运行每个测试之前设置一个空的 MEDIA_ROOT 文件夹。

我想我会创建一个临时文件夹并将 MEDIA_ROOT 指向它。但是，我无法弄清楚将执行此操作的代码放在哪里。在 this example , 一个特殊的 Runner 被创建。运行者设置媒体根并将其拆除。

不幸的是，setup_test_environment 在第一个测试函数运行之前被调用一次，而不是每次运行测试时都被调用。

我尝试创建一个 FileSystemTestCase 类，该类在其 setUp 函数中设置文件系统，并让我的所有测试用例都派生自它。虽然这可行，但它要求每个编写测试用例的人记住调用我的 setUp 方法，因为它不会自动调用。

通常我不会为此烦恼，但忘记调用父 setUp 方法的成本可能非常高 - 如果有人忘记调用并且测试不小心在实时系统上运行，就会发生不好的事情。

编辑:我找到的临时解决方案是同时实现我自己的运行器和基本TestCase。两者都设置了一个临时的 MEDIA_ROOT，所以如果有人忘记调用我的 setUp 方法，测试将在前一个测试的临时文件夹或运行器设置的文件夹中运行。这可能会导致测试失败，但不会破坏实时数据。

我希望有一个更优雅的解决方案。

Answer 1

在我看来，您正在尝试解决两个不同的问题:

1. 允许测试在测试人员做正确的事情(即从您的测试类继承并调用您的setUp())时独立运行(关于MEDIA_ROOT)。
2. 防止测试人员在不小心做错事情时弄乱真实数据。

鉴于此，我认为双管齐下的方法是有道理的。您的 setUp() 解决了问题 1。但是，在测试运行器中设置 MEDIA_ROOT 隐藏了您的测试人员做错事的事实。相反，我只关注保护数据:例如，您可以将 MEDIA_ROOT 设置为 None。这将屏蔽 MEDIA_ROOT 中的真实数据；如果测试人员不使用您的 setUp()，则更有可能看到错误；并减少代码重复。

一种更稳健的方法是编写您自己的测试运行器，在每次测试之前进行设置(模仿 Django 处理数据库的方式)，但这对您的需求来说可能有点矫枉过正。