Integration tests for execute.py.
This commit is contained in:
parent
b43ef9d76d
commit
7d7308a80d
2 changed files with 37 additions and 1 deletions
|
@ -23,7 +23,7 @@ def execute_and_log_output(full_command, output_log_level, shell):
|
||||||
|
|
||||||
remaining_output = process.stdout.read().rstrip().decode()
|
remaining_output = process.stdout.read().rstrip().decode()
|
||||||
if remaining_output:
|
if remaining_output:
|
||||||
logger.info(remaining_output)
|
logger.log(output_log_level, remaining_output)
|
||||||
|
|
||||||
exit_code = process.poll()
|
exit_code = process.poll()
|
||||||
if exit_code != 0:
|
if exit_code != 0:
|
||||||
|
|
36
tests/integration/test_execute.py
Normal file
36
tests/integration/test_execute.py
Normal file
|
@ -0,0 +1,36 @@
|
||||||
|
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\nthere'], output_log_level=logging.INFO, shell=False)
|
||||||
|
|
||||||
|
|
||||||
|
def test_execute_and_log_output_logs_borg_error_as_error():
|
||||||
|
flexmock(module.logger).should_receive('error').with_args('borg: error: oopsie').once()
|
||||||
|
flexmock(module.logger).should_receive('log').with_args(logging.INFO, 'daisie').once()
|
||||||
|
|
||||||
|
module.execute_and_log_output(
|
||||||
|
['echo', 'borg: error: oopsie\ndaisie\n'], output_log_level=logging.INFO, shell=False
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
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').never()
|
||||||
|
|
||||||
|
with pytest.raises(subprocess.CalledProcessError):
|
||||||
|
module.execute_and_log_output(['false'], output_log_level=logging.INFO, shell=False)
|
Loading…
Reference in a new issue