假设我在 python 中有一个这样设置的类。
from somewhere import sendmail
class MyClass:
def __init__(self, **kargs):
self.sendmail = kwargs.get("sendmail", sendmail) #if we can't find it, use imported def
def publish():
#lots of irrelevant code
#and then
self.sendmail(mail_to, mail_from, subject, body, format= 'html')
如你所见,我已经给自己一个选项来参数化我用于 self.sendmail 的函数
现在在测试文件中。
Class Tester():
kwargs = {"sendmail": MagicMock(mail_from= None, mail_to= None, subject= None, body= None, format= None)}
self.myclass = MyClass(**kwargs)
##later on
def testDefaultEmailHeader():
default_subject = "Hello World"
self.myclass.publish()
self.myclass.sendmail.assert_called() #this is doing just fine
self.myclass.sendmail.assert_called_with(default_subject) #this is having issues
由于某种原因,我收到了错误消息
AssertionError: Expected call: mock('Hello World')
Actual Call : mock('defaultmt', 'defaultmf', 'Hello World', 'default_body', format= 'html')
所以基本上,assert 期望 sendmail 只用一个变量调用,当它最终用所有 5 个变量调用时。问题是,我不在乎其他 4 个变量是什么!我只是想确保使用正确的主题调用它。
我尝试了模拟占位符 ANY,得到了同样的结果
self.myclass.sendmail.assert_called_with(ANY, ANY, 'Hello World', ANY, ANY)
AssertionError: Expected call: mock(<ANY>, <ANY>, 'Hello World', <ANY>, <ANY>)
Actual Call : mock('defaultmt', 'defaultmf', 'Hello World', 'default_body, 'format= 'html')
真的不确定如何继续这个。如果我们只关心其中一个变量而想忽略其余变量,有人有什么建议吗?
最佳答案
如果您使用命名参数 subject
调用 sendmail
,那么最好检查命名参数是否符合您的预期:
args, kwargs = self.myclass.sendmail.call_args
self.assertEqual(kwargs['subject'], "Hello World")
这确实假设 sendmail
的两个实现都有一个名为 subject
的命名参数。如果不是这种情况,您可以使用位置参数执行相同操作:
args, kwargs = self.myclass.sendmail.call_args
self.assertTrue("Hello World" in args)
您可以明确参数的位置(即,传递给 sendmail
的第一个参数或第三个参数,但这取决于 sendmail
的实现测试)。
关于Python assert_call_with,有通配符吗?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/20428750/