web-dev-qa-db-ja.com

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

ご覧のとおり、私は自分自身に使用する関数をパラメーター化するオプションを自分に与えています。

今テストファイルです。

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が1つの変数のみで呼び出されることを期待しており、最終的にすべて5で呼び出されることになります。正しい件名で呼び出されることを確認したいだけです。

私は模擬プレースホルダーを何回か試してみましたが、同じものを得ました

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つだけを気にし、残りを無視したい場合は、誰かアドバイスがありますか？

pythontestingmockingassert
33
Zack

名前付きパラメーター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に渡される最初の引数または3番目の引数ですが、テストされるsendmailの実装によって異なります）。

34
Simeon Visser