borgmatic/tests/integration/hooks/monitoring/test_healthchecks.py

import logging

from flexmock import flexmock

from borgmatic.hooks.monitoring import healthchecks as module


def test_destroy_monitor_removes_healthchecks_handler():
    logger = logging.getLogger()
    original_handlers = list(logger.handlers)

    # Don't mess with the actual log level because it can impact downstream tests.
    flexmock(logger).should_receive('setLevel')
    module.borgmatic.hooks.monitoring.logs.add_handler(
        module.borgmatic.hooks.monitoring.logs.Forgetful_buffering_handler(
            identifier=module.HANDLER_IDENTIFIER, byte_capacity=100, log_level=1
        )
    )

    module.destroy_monitor(
        hook_config=flexmock(), config=flexmock(), monitoring_log_level=1, dry_run=False
    )

    assert logger.handlers == original_handlers


def test_destroy_monitor_without_healthchecks_handler_does_not_raise():
    logger = logging.getLogger()
    original_handlers = list(logger.handlers)

    # Don't mess with the actual log level because it can impact downstream tests.
    flexmock(logger).should_receive('setLevel')
    module.destroy_monitor(
        hook_config=flexmock(), config=flexmock(), monitoring_log_level=1, dry_run=False
    )

    assert logger.handlers == original_handlers