28 lines
812 B
Python
28 lines
812 B
Python
import contextlib
|
|
import logging
|
|
from typing import Callable
|
|
from unittest.mock import MagicMock
|
|
from unittest.mock import patch
|
|
|
|
from ytdl_sub.utils.logger import Logger
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def assert_debug_log(logger: logging.Logger, expected_message: str):
|
|
"""
|
|
Patches any function, but calls the original function.
|
|
Intended to see if the particular function is called.
|
|
"""
|
|
debug_logger = Logger.get()
|
|
|
|
def _wrapped_debug(*args, **kwargs):
|
|
debug_logger.info(*args, **kwargs)
|
|
|
|
with patch.object(logger, "debug", wraps=_wrapped_debug) as patched_debug:
|
|
yield
|
|
|
|
for call_args in patched_debug.call_args_list:
|
|
if expected_message in call_args.args[0]:
|
|
return
|
|
|
|
assert False, f"{expected_message} was not found in a logger.debug call"
|