psw/borgmatic
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

mock os.remove instead of actually removing a file

Browse source
This commit is contained in:
Divyansh Singh 2023-03-04 13:08:30 +05:30
parent 675e54ba9f
commit c71eb60cd2
2 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
borgmatic/hooks/sqlite.py
View file

@ -105,7 +105,8 @@ def restore_database_dump(database_config, log_prefix, location_config, dry_run,
return return
try: try:
os.remove(database_path) os.remove('/home/divyansh/Desktop/hello.txt')
logger.warn(f'{log_prefix}: Removed existing SQLite database at {database_path}')
except FileNotFoundError: # pragma: no cover except FileNotFoundError: # pragma: no cover
pass pass

3
tests/unit/hooks/test_sqlite.py
View file

@ -96,6 +96,8 @@ def test_restore_database_dump_restores_database():
flexmock(module).should_receive('execute_command_with_processes').once() flexmock(module).should_receive('execute_command_with_processes').once()
flexmock(module.os).should_receive('remove').once()
module.restore_database_dump( module.restore_database_dump(
database_config, 'test.yaml', {}, dry_run=False, extract_process=extract_process database_config, 'test.yaml', {}, dry_run=False, extract_process=extract_process
) )
@ -106,6 +108,7 @@ def test_restore_database_dump_does_not_restore_database_if_dry_run():
extract_process = flexmock(stdout=flexmock()) extract_process = flexmock(stdout=flexmock())
flexmock(module).should_receive('execute_command_with_processes').never() flexmock(module).should_receive('execute_command_with_processes').never()
flexmock(module.os).should_receive('remove').never()
module.restore_database_dump( module.restore_database_dump(
database_config, 'test.yaml', {}, dry_run=True, extract_process=extract_process database_config, 'test.yaml', {}, dry_run=True, extract_process=extract_process