example_decorator_python.py

import unittest
from functools import wraps
from unittest import mock


def log(*args, **kwargs):
    """ Logging function """


def print_debug(*args, **kwargs):
    """ Logging function """
    log(*args, **kwargs)
    print(f"  args = {args}\n  kwargs = {kwargs}")


def logging_decorator(f):
    """ Implement decorator here """

    @wraps(f)
    def inner(*args, **kwargs):
        print(f"\n{f.__name__}:")
        print_debug(*args, **kwargs)
        return f(*args, **kwargs)

    return inner


class TestDecorator(unittest.TestCase):
    @mock.patch(f'{__name__}.log')
    def test_success(self, mocked_login):
        @logging_decorator
        def dummy_function(a, b):
            return a + b

        result = dummy_function(1, b=2)
        self.assertEqual(result, 3)
        mocked_login.assert_called_once()