25 lines
762 B
Python
25 lines
762 B
Python
from flexmock import flexmock
|
|
from borgmatic.borg import execute as module
|
|
|
|
|
|
def test_execute_command_calls_full_command():
|
|
full_command = ['foo', 'bar']
|
|
subprocess = flexmock(module.subprocess)
|
|
subprocess.should_receive('check_call').with_args(full_command).once()
|
|
|
|
output = module.execute_command(full_command)
|
|
|
|
assert output is None
|
|
|
|
|
|
def test_execute_command_captures_output():
|
|
full_command = ['foo', 'bar']
|
|
expected_output = '[]'
|
|
subprocess = flexmock(module.subprocess)
|
|
subprocess.should_receive('check_output').with_args(full_command).and_return(
|
|
flexmock(decode=lambda: expected_output)
|
|
).once()
|
|
|
|
output = module.execute_command(full_command, capture_output=True)
|
|
|
|
assert output == expected_output
|