borgmatic/tests/integration/test_execute.py

import logging
import subprocess

import pytest
from flexmock import flexmock

from borgmatic import execute as module


def test_execute_and_log_output_logs_each_line_separately():
    flexmock(module.logger).should_receive('log').with_args(logging.INFO, 'hi').once()
    flexmock(module.logger).should_receive('log').with_args(logging.INFO, 'there').once()

    module.execute_and_log_output(['echo', 'hi'], output_log_level=logging.INFO, shell=False)
    module.execute_and_log_output(['echo', 'there'], output_log_level=logging.INFO, shell=False)


def test_execute_and_log_output_with_borg_warning_does_not_raise():
    flexmock(module.logger).should_receive('log')

    # Borg's exit code 1 is a warning, not an error.
    module.execute_and_log_output(['false'], output_log_level=logging.INFO, shell=False)


def test_execute_and_log_output_includes_borg_error_output_in_exception():
    flexmock(module.logger).should_receive('log')

    with pytest.raises(subprocess.CalledProcessError) as error:
        module.execute_and_log_output(['grep'], output_log_level=logging.INFO, shell=False)

    assert error.value.returncode == 2
    assert error.value.output


def test_execute_and_log_output_with_shell_error_raises():
    flexmock(module.logger).should_receive('log')

    with pytest.raises(subprocess.CalledProcessError) as error:
        module.execute_and_log_output(['false'], output_log_level=logging.INFO, shell=True)

    assert error.value.returncode == 1


def test_execute_and_log_output_truncates_long_borg_error_output():
    flexmock(module).ERROR_OUTPUT_MAX_LINE_COUNT = 0
    flexmock(module.logger).should_receive('log')

    with pytest.raises(subprocess.CalledProcessError) as error:
        module.execute_and_log_output(['grep'], output_log_level=logging.INFO, shell=False)

    assert error.value.returncode == 2
    assert error.value.output.startswith('...')


def test_execute_and_log_output_with_no_output_logs_nothing():
    flexmock(module.logger).should_receive('log').never()

    module.execute_and_log_output(['true'], output_log_level=logging.INFO, shell=False)


def test_execute_and_log_output_with_error_exit_status_raises():
    flexmock(module.logger).should_receive('log')

    with pytest.raises(subprocess.CalledProcessError):
        module.execute_and_log_output(['grep'], output_log_level=logging.INFO, shell=False)