Python assert_called_with, есть ли подстановочный знак?

Предположим, что у меня есть класс в 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 переменные! Я просто хочу убедиться, что он вызван с правильным субъектом.

Я попробовал ложный держатель места ЛЮБОЙ, и получил то же самое

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')

Не уверен, как это сделать. У кого-нибудь есть какие-либо советы, если мы только заботимся об одной из переменных и хотим игнорировать остальные?

Ответы

Ответ 1

Если вы вызываете sendmail с именованным параметром subject, тогда лучше проверить, соответствует ли именованный аргумент тому, что вы ожидаете:

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).