python - Django 模型中的单元测试外键约束

Question

我定义了 2 个模型，其中一个通过外键关系引用另一个模型。我想编写单元测试来确保这种关系。

class X(models.Model):
    name = models.CharField(unique = True)

class Y(models.Model):
    event = models.ForeignKey(X)

在测试中我有

class TestY(TestCase):
    x = X.objects.create(name="test1")
    x.save()
    y = Y(event=X.objects.create(name="test2"))
    with self.assertRaises(ValidationError):
        if y.full_clean()
            y.save()

    self.assert(0,Y.objects.filter(event__name="test2").count)

这表示测试失败，未引发ValidationError。

此外，如果字段不允许为空，我应该如何测试 ValueError 。 self.assertRaises(ValueError) 不起作用。

Answer 1

你想要这样的东西吗？

class TestY(TestCase):

    def test_model_relation(self):
        x = X.objects.create(name="test1")
        y = Y(event=X.objects.create(name="test2"))
        y.full_clean()  # `event` correctly set. This should pass
        y.save()
        self.assertEqual(Y.objects.filter(event__name="test2").count(), 1)

    def test_model_relation__event_missing(self):
        x = X.objects.create(name="test1")
        y = Y()  # Y without `event` set
        with self.assertRaises(ValidationError):
            y.full_clean()
            y.save()
        self.assertEqual(Y.objects.filter(event__name="test2").count(), 0)

顺便说一句，您应该在测试方法(名称以 test 开头的方法)中指定 test，而不是在类主体中。