如何在 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是错误的 - 您需要传递一个可调用对象而不是调用函数本身，并将任何参数作为参数传递给assertRaises. 将其更改为：

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

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

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

Answer 2

接受的答案不正确。self.assertRaises()不检查异常消息。

正确答案

如果你想断言异常消息，你应该使用self.assertRaisesRegex()。

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

或者

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