Assert_called_once_with需要检查如果中的实例的呼吁是与同样的信息

0

的问题

我试图使用assert_called_once_with从单元测试.模拟 https://docs.python.org/3/library/unittest.mock.html#unittest.mock.Mock.assert_called_once_with

但我想检查,如果2个实例的对象已经通过与te正确的属性。

所以内部流我重写的 __eq__ func:

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

在代码我曾像这样

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

但是我继续收到错误和睾丸,保持比较repr的实例,这样的

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

我真的需要一种可以断言,如果一个实例的对象已被通过作为一个参数的功能的正确性填补。

mocking pytest python python-unittest
2021-11-23 20:31:11
1

最好的答案

0

你想使用的 call_args 属性模拟对象:

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)

这里是一个稍微更复杂的例子 argskwargs 的断言:

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)
2021-11-26 15:33:54

其他语言

此页面有其他语言版本

Русский
..................................................................................................................
Italiano
..................................................................................................................
Polski
..................................................................................................................
Română
..................................................................................................................
한국어
..................................................................................................................
हिन्दी
..................................................................................................................
Français
..................................................................................................................
Türk
..................................................................................................................
Česk
..................................................................................................................
Português
..................................................................................................................
ไทย
..................................................................................................................
Español
..................................................................................................................
Slovenský
..................................................................................................................