From 3aa88085ed7310b43380f3e4ec48449fb73f7692 Mon Sep 17 00:00:00 2001 From: Divyansh Singh Date: Fri, 3 Mar 2023 00:01:52 +0530 Subject: [PATCH] formatting fix --- tests/unit/hooks/test_sqlite.py | 32 +++++++++++++++++++++++--------- 1 file changed, 23 insertions(+), 9 deletions(-) diff --git a/tests/unit/hooks/test_sqlite.py b/tests/unit/hooks/test_sqlite.py index 15d5911d..d679cad0 100644 --- a/tests/unit/hooks/test_sqlite.py +++ b/tests/unit/hooks/test_sqlite.py @@ -17,7 +17,8 @@ def test_dump_databases_logs_and_skips_if_dump_already_exists(): flexmock(logging).should_receive('info') flexmock(logging).should_receive('warning') - assert module.dump_databases(databases, 'test.yaml', {}, dry_run = False) == [] + assert module.dump_databases(databases, 'test.yaml', {}, dry_run=False) == [] + def test_dump_databases_dumps_each_database(): databases = [ @@ -35,7 +36,11 @@ def test_dump_databases_dumps_each_database(): flexmock(module).should_receive('dump.create_parent_directory_for_dump') flexmock(module).should_receive('execute_command').and_return('process') - assert module.dump_databases(databases, 'test.yaml', {}, dry_run = False) == ['process', 'process'] + assert module.dump_databases(databases, 'test.yaml', {}, dry_run=False) == [ + 'process', + 'process', + ] + def test_dump_databases_does_not_dump_if_dry_run(): databases = [{'path': '/path/to/database', 'name': 'database'}] @@ -49,27 +54,36 @@ def test_dump_databases_does_not_dump_if_dry_run(): flexmock(logging).should_receive('warning') flexmock(module).should_receive('dump.create_parent_directory_for_dump') - assert module.dump_databases(databases, 'test.yaml', {}, dry_run = True) == [] + assert module.dump_databases(databases, 'test.yaml', {}, dry_run=True) == [] + def test_restore_database_dump_restores_database(): database_config = [{'path': '/path/to/database', 'name': 'database'}] - extract_process = flexmock(stdout = flexmock()) + extract_process = flexmock(stdout=flexmock()) flexmock(module).should_receive('execute_command_with_processes').and_return('process') - module.restore_database_dump(database_config, 'test.yaml', {}, dry_run = False, extract_process = extract_process) + module.restore_database_dump( + database_config, 'test.yaml', {}, dry_run=False, extract_process=extract_process + ) + def test_restore_database_dump_does_not_restore_database_if_dry_run(): database_config = [{'path': '/path/to/database', 'name': 'database'}] - extract_process = flexmock(stdout = flexmock()) + extract_process = flexmock(stdout=flexmock()) flexmock(module).should_receive('execute_command_with_processes').never() - module.restore_database_dump(database_config, 'test.yaml', {}, dry_run = True, extract_process = extract_process) + module.restore_database_dump( + database_config, 'test.yaml', {}, dry_run=True, extract_process=extract_process + ) + def test_restore_database_dump_raises_error_if_database_config_is_invalid(): database_config = [] - extract_process = flexmock(stdout = flexmock()) + extract_process = flexmock(stdout=flexmock()) with pytest.raises(ValueError): - module.restore_database_dump(database_config, 'test.yaml', {}, dry_run = False, extract_process = extract_process) \ No newline at end of file + module.restore_database_dump( + database_config, 'test.yaml', {}, dry_run=False, extract_process=extract_process + )