Assert_called_once_with besoin de vérifier si les instances de l'appel sont avec la même info

Question 1

Je suis en train d'utiliser assert_called_once_with de unittest.maquette https://docs.python.org/3/library/unittest.mock.html#unittest.mock.Mock.assert_called_once_with

Mais je veux vérifier si 2 instances d'objets ont été passés avec te attributs corrects.

Donc, à l'intérieur de la classe I surdéfini l' __eq__ func:

    def __eq__(self, other):
        return (
            self.id == other.id
            and self.users == other.users
        )

Et dans le code que j'utilise comme ceci

        mock_add_user.assert_called_once_with(context_fix, expected_user_obj1, expected_user_obj2, users)

Mais je reçois un message d'erreur et la teste garde en comparant les repr de l'instance, à l'instar de ce

<app.domain.model.load_md.Load object at 0x10cb7d7d0>

E           AssertionError: Expected call: add_user(<app.infrastructure.context.Context object at 0x10cb7da50>, <app.domain.model.load_md.Load object at 0x10cb7d7d0>, <app.domain.model.load_md.Load object at 0x10cb7d550>, [1, 2, 3, 2])
E           Actual call: add_user(<app.infrastructure.context.Context object at 0x10cb7da50>, <app.domain.model.load_md.Load object at 0x10ca7f5d0>, <app.domain.model.load_md.Load object at 0x10cd1e650>, [1, 2, 3, 2])

J'ai vraiment besoin d'un moyen d'affirmer si une instance d'un objet est passé en paramètre de la fonction avec les attributs corrects rempli.

Question 2

Vous souhaitez utiliser le call_args attribut de l'objet fantaisie:

from unittest.mock import ANY, Mock

class Foo:
    pass

def call(add_user):
    f = Foo()
    add_user(f)

def test_is_instance():
    mock_add_user = Mock()

    call(mock_add_user)

    # expected = Foo()
    # mock_add_user.assert_called_once_with(expected) # This would fail
    mock_add_user.assert_called_once_with(ANY)
    assert isinstance(mock_add_user.call_args.args[0], Foo)

Voici un exemple légèrement plus complexe avec args et kwargs affirmations:

from unittest.mock import ANY, Mock

class Foo:
    pass

class Bar:
    pass

def call(add_user):
    f = Foo()
    b = Bar()
    add_user(f, bar=b)

def test_is_instance():
    mock_add_user = Mock()

    call(mock_add_user)

    mock_add_user.assert_called_once_with(ANY, bar=ANY)
    assert isinstance(mock_add_user.call_args.args[0], Foo)
    assert isinstance(mock_add_user.call_args.kwargs["bar"], Bar)

Tibor Takács · Answer 1 · 2021-11-26T15:33:54

Vous souhaitez utiliser le call_args attribut de l'objet fantaisie:

from unittest.mock import ANY, Mock

class Foo:
    pass

def call(add_user):
    f = Foo()
    add_user(f)

def test_is_instance():
    mock_add_user = Mock()

    call(mock_add_user)

    # expected = Foo()
    # mock_add_user.assert_called_once_with(expected) # This would fail
    mock_add_user.assert_called_once_with(ANY)
    assert isinstance(mock_add_user.call_args.args[0], Foo)

Voici un exemple légèrement plus complexe avec args et kwargs affirmations:

from unittest.mock import ANY, Mock

class Foo:
    pass

class Bar:
    pass

def call(add_user):
    f = Foo()
    b = Bar()
    add_user(f, bar=b)

def test_is_instance():
    mock_add_user = Mock()

    call(mock_add_user)

    mock_add_user.assert_called_once_with(ANY, bar=ANY)
    assert isinstance(mock_add_user.call_args.args[0], Foo)
    assert isinstance(mock_add_user.call_args.kwargs["bar"], Bar)

Assert_called_once_with besoin de vérifier si les instances de l'appel sont avec la même info

La question

La meilleure réponse

Dans d'autres langues

Cette page est dans d'autres langues

Populaires dans cette catégorie

Questions populaires dans cette catégorie