Checking arguments of called mocked objects

functions.py

# functions.py


class CustomPessoa:
    def __init__(self, *args, **kwargs):
        for (k, v) in kwargs.items():
            setattr(self, k, v)


def target_function(pessoa, outro_arg):
    return


def caller():
    c = CustomPessoa(nome="Vinicius")
    target_function(c, False)
    return

test_functions.py

# test_functions.py
# run it with: $ python -m unittest test_functions.py

import unittest
from unittest import mock

import functions


class TestFunctions(unittest.TestCase):
    @mock.patch("functions.target_function")
    def test_called(self, target_mocked):
        functions.caller()
        self.assertEqual(target_mocked.called, True)

    @unittest.expectedFailure
    @mock.patch("functions.target_function")
    def test_called_with_fail(self, target_mocked):
        functions.caller()

        expected_pessoa = functions.CustomPessoa(nome="Vinicius")
        target_mocked.assert_called_with(expected_pessoa)
        # O teste falha porque o argumento passado para `target_function()`
        # nao eh o mesmo objeto `expected_pessoa`, apesar do atributo `nome`
        # ter o mesmo conteudo.

    @mock.patch("functions.target_function")
    def test_inspect_call_args_success(self, target_mocked):
        expected_pessoa = functions.CustomPessoa(nome="Vinicius")

        functions.caller()

        (arg_pessoa, arg_outro_arg) = target_mocked.call_args[0]
        self.assertEqual(arg_pessoa.nome, expected_pessoa.nome)