模拟 - 测试是否在不指定参数的情况下调用方法

时间:2014-01-21 17:28:17

标签: python django unit-testing mocking

我正在使用Mock库嘲笑Django的send_mail函数:

from django.core.mail import send_mail
send_mail = Mock()

当我想断言send_mail被调用时,我的问题出现了。我不能使用assert_called_with或相关的方法,因为他们期望参数,我不会完全知道/不想在测试环境中输入(它涉及模板)。我只是想知道已经调用的方法。打印语句清楚地表明它已被调用,但如果我尝试这样做:

self.assertEqual(send_mail.called, True) 

我收到错误,因为被叫属性仍为False。如何在不必指定属性的情况下测试方法是否已被测试。

1 个答案:

答案 0 :(得分:6)

您尚未正确模拟send_mail方法。您只能在自己的模块中反弹名称,而不是其他地方。

这将是一个正确的重新绑定:

from django.core import mail
mail.send_mail = Mock()

但使用patch() context manager / decorator

更容易
with patch('django.core.mail.send_mail') as mocked_send_mail:
    # run your tests
    self.assertTrue(mocked_send_mail.called)