django - 如何在 Django 中正确使用 assertRaises

Question

我的模型中有以下验证功能:

@classmethod
def validate_kind(cls, kind):
    if kind == 'test':
        raise ValidationError("Invalid question kind")

我正在尝试如下测试:

w = Model.objects.get(id=1) 

self.assertRaises(ValidationError, w.validate_kind('test'),msg='Invalid question kind')

我也试过:

self.assertRaisesRegex(w.validate_kind('test'),'Invalid question kind')

这两个都不能正常工作。我究竟做错了什么？

Answer 1

您打电话的方式 assertRaises错了 - 你需要 pass a callable而不是调用函数本身，并将任何参数传递给函数作为参数给 assertRaises .将其更改为:

self.assertRaises(ValidationError, w.validate_kind, 'test')

请注意，您无法使用此构造测试异常消息 - 如果要测试消息，则需要使用 assertRaises作为上下文管理器:

with self.assertRaises(ValidationError, msg='Invalid question kind'):
    w.validate_kind('test')