test_systemd.py 2.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. import io
  2. import sys
  3. import pytest
  4. from flexmock import flexmock
  5. from borgmatic.hooks.credential import systemd as module
  6. @pytest.mark.parametrize('credential_parameters', ((), ('foo', 'bar')))
  7. def test_load_credential_with_invalid_credential_parameters_raises(credential_parameters):
  8. flexmock(module.os.environ).should_receive('get').never()
  9. with pytest.raises(ValueError):
  10. module.load_credential(
  11. hook_config={},
  12. config={},
  13. credential_parameters=credential_parameters,
  14. )
  15. def test_load_credential_without_credentials_directory_raises():
  16. flexmock(module.os.environ).should_receive('get').with_args('CREDENTIALS_DIRECTORY').and_return(
  17. None,
  18. )
  19. with pytest.raises(ValueError):
  20. module.load_credential(hook_config={}, config={}, credential_parameters=('mycredential',))
  21. def test_load_credential_with_invalid_credential_name_raises():
  22. flexmock(module.os.environ).should_receive('get').with_args('CREDENTIALS_DIRECTORY').and_return(
  23. '/var',
  24. )
  25. with pytest.raises(ValueError):
  26. module.load_credential(
  27. hook_config={},
  28. config={},
  29. credential_parameters=('../../my!@#$credential',),
  30. )
  31. def test_load_credential_reads_named_credential_from_file():
  32. flexmock(module.os.environ).should_receive('get').with_args('CREDENTIALS_DIRECTORY').and_return(
  33. '/var',
  34. )
  35. credential_stream = io.StringIO('password')
  36. credential_stream.name = '/var/borgmatic.pw'
  37. builtins = flexmock(sys.modules['builtins'])
  38. builtins.should_receive('open').with_args('/var/borgmatic.pw', encoding='utf-8').and_return(
  39. credential_stream
  40. )
  41. assert (
  42. module.load_credential(hook_config={}, config={}, credential_parameters=('borgmatic.pw',))
  43. == 'password'
  44. )
  45. def test_load_credential_with_file_not_found_error_raises():
  46. flexmock(module.os.environ).should_receive('get').with_args('CREDENTIALS_DIRECTORY').and_return(
  47. '/var',
  48. )
  49. builtins = flexmock(sys.modules['builtins'])
  50. builtins.should_receive('open').with_args('/var/mycredential', encoding='utf-8').and_raise(
  51. FileNotFoundError
  52. )
  53. with pytest.raises(ValueError):
  54. module.load_credential(hook_config={}, config={}, credential_parameters=('mycredential',))