我正在使用 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
是从数据库中检索的。
最佳答案
您在测试中所做的一切都很好。但是,我认为您可以稍微简化/改进它。
我认为你应该使用工厂(你可以使用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)
关于python - 在 Django 测试中,我应该如何保存数据库对象然后从数据库中检索它?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/31898557/