unit - python 2.7 mock

Python''s assert_called_with, ¿hay un carácter comodín?

Question

Supongamos que tengo una clase en Python configurada así.

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

Como puede ver, me he dado la opción de parametrizar qué función uso para self.sendmail

Ahora en el archivo de prueba.

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

Por alguna razón recibo el mensaje de error

AssertionError: Expected call: mock(''Hello World'')
                Actual Call : mock(''defaultmt'', ''defaultmf'', ''Hello World'', ''default_body'', format= ''html'')

Así que, básicamente, la afirmación espera que se llame a sendmail con una sola variable, cuando termina siendo llamado con todos 5. El problema es que no me importan las otras 4 variables. Solo quiero asegurarme de que se llame con el tema correcto.

Probé el titular de lugar simulado CUALQUIERA, y obtuve lo mismo

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

Realmente no estoy seguro de cómo proceder con este. ¿Alguien tiene algún consejo si solo nos importa una de las variables y queremos ignorar el resto?

Answer 1

Si llama a sendmail con un subject parámetro nombrado, entonces es mejor verificar si el argumento con nombre coincide con lo que espera:

args, kwargs = self.myclass.sendmail.call_args
self.assertEqual(kwargs[''subject''], "Hello World")

Esto supone que ambas implementaciones de sendmail tienen un parámetro llamado subject . Si ese no es el caso, puede hacer lo mismo con un parámetro posicional:

args, kwargs = self.myclass.sendmail.call_args
self.assertTrue("Hello World" in args)

Puede ser explícito sobre la posición del argumento (es decir, el primer argumento o el tercer argumento que se pasa a sendmail pero eso depende de la implementación de sendmail está probando).