1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071 |
- import io
- import sys
- import pytest
- from flexmock import flexmock
- from borgmatic.hooks.credential import systemd as module
- @pytest.mark.parametrize('credential_parameters', ((), ('foo', 'bar')))
- def test_load_credential_with_invalid_credential_parameters_raises(credential_parameters):
- flexmock(module.os.environ).should_receive('get').never()
- with pytest.raises(ValueError):
- module.load_credential(
- hook_config={},
- config={},
- credential_parameters=credential_parameters,
- )
- def test_load_credential_without_credentials_directory_raises():
- flexmock(module.os.environ).should_receive('get').with_args('CREDENTIALS_DIRECTORY').and_return(
- None,
- )
- with pytest.raises(ValueError):
- module.load_credential(hook_config={}, config={}, credential_parameters=('mycredential',))
- def test_load_credential_with_invalid_credential_name_raises():
- flexmock(module.os.environ).should_receive('get').with_args('CREDENTIALS_DIRECTORY').and_return(
- '/var',
- )
- with pytest.raises(ValueError):
- module.load_credential(
- hook_config={},
- config={},
- credential_parameters=('../../my!@#$credential',),
- )
- def test_load_credential_reads_named_credential_from_file():
- flexmock(module.os.environ).should_receive('get').with_args('CREDENTIALS_DIRECTORY').and_return(
- '/var',
- )
- credential_stream = io.StringIO('password')
- credential_stream.name = '/var/borgmatic.pw'
- builtins = flexmock(sys.modules['builtins'])
- builtins.should_receive('open').with_args('/var/borgmatic.pw', encoding='utf-8').and_return(
- credential_stream
- )
- assert (
- module.load_credential(hook_config={}, config={}, credential_parameters=('borgmatic.pw',))
- == 'password'
- )
- def test_load_credential_with_file_not_found_error_raises():
- flexmock(module.os.environ).should_receive('get').with_args('CREDENTIALS_DIRECTORY').and_return(
- '/var',
- )
- builtins = flexmock(sys.modules['builtins'])
- builtins.should_receive('open').with_args('/var/mycredential', encoding='utf-8').and_raise(
- FileNotFoundError
- )
- with pytest.raises(ValueError):
- module.load_credential(hook_config={}, config={}, credential_parameters=('mycredential',))
|