python - 在 Django 测试中，我应该如何保存数据库对象然后从数据库中检索它？

Question

我正在使用 Django 1.8。我写了下面的代码来测试 pre_save 钩子(Hook)是否正常工作，但是这段代码看起来很不优雅。这是编写此类单元测试的“正确方法”吗？

class PreSaveTests(TestCase):
    def test_pre_save_hook(self):
        person = Person(name="Joe")
        person.save()
        person2 = Person.objects.get(pk = person.pk)
        # Confirm that the pre_save hook ran.
        # The hook sets person.is_cool to True.
        self.assertEqual(person2.is_cool, True)

这很好用，但看起来很丑。

真正丑陋的部分是 person 和 person2 是同一个数据库对象。唯一的区别是 person2 是从数据库中检索的。

Answer 1

您在测试中所做的一切都很好。但是，我认为您可以稍微简化/改进它。

我认为你应该使用工厂(你可以使用FactoryBoy)。这样，当您在模型上添加/删除必填字段时，您不必更新测试。此外，您可以从测试中删除不相关的信息。在这种情况下，人名为 Joe 的事实完全无关紧要。

你可以替换:

person = Person(name="Joe")
person.save()

与:

person = PersonFactory.create()

正如 Daniel 提到的，您不需要重新加载 Person 实例。所以你不必这样做:

person2 = Person.objects.get(pk = person.pk)

最后，一个小提示，你可以使用assertTrue代替assertEquals(something, True):

class PreSaveTests(TestCase):

    def test_pre_save_hook(self):
        person = PersonFactory.create()
        self.assertTrue(person.is_cool)