|
@@ -0,0 +1,644 @@
|
|
|
+import logging
|
|
|
+
|
|
|
+import pytest
|
|
|
+from flexmock import flexmock
|
|
|
+
|
|
|
+from borgmatic.hooks import mariadb as module
|
|
|
+
|
|
|
+
|
|
|
+def test_database_names_to_dump_passes_through_name():
|
|
|
+ extra_environment = flexmock()
|
|
|
+ log_prefix = ''
|
|
|
+
|
|
|
+ names = module.database_names_to_dump(
|
|
|
+ {'name': 'foo'}, extra_environment, log_prefix, dry_run=False
|
|
|
+ )
|
|
|
+
|
|
|
+ assert names == ('foo',)
|
|
|
+
|
|
|
+
|
|
|
+def test_database_names_to_dump_bails_for_dry_run():
|
|
|
+ extra_environment = flexmock()
|
|
|
+ log_prefix = ''
|
|
|
+ flexmock(module).should_receive('execute_command_and_capture_output').never()
|
|
|
+
|
|
|
+ names = module.database_names_to_dump(
|
|
|
+ {'name': 'all'}, extra_environment, log_prefix, dry_run=True
|
|
|
+ )
|
|
|
+
|
|
|
+ assert names == ()
|
|
|
+
|
|
|
+
|
|
|
+def test_database_names_to_dump_queries_mariadb_for_database_names():
|
|
|
+ extra_environment = flexmock()
|
|
|
+ log_prefix = ''
|
|
|
+ flexmock(module).should_receive('execute_command_and_capture_output').with_args(
|
|
|
+ ('mariadb', '--skip-column-names', '--batch', '--execute', 'show schemas'),
|
|
|
+ extra_environment=extra_environment,
|
|
|
+ ).and_return('foo\nbar\nmysql\n').once()
|
|
|
+
|
|
|
+ names = module.database_names_to_dump(
|
|
|
+ {'name': 'all'}, extra_environment, log_prefix, dry_run=False
|
|
|
+ )
|
|
|
+
|
|
|
+ assert names == ('foo', 'bar')
|
|
|
+
|
|
|
+
|
|
|
+def test_dump_databases_dumps_each_database():
|
|
|
+ databases = [{'name': 'foo'}, {'name': 'bar'}]
|
|
|
+ processes = [flexmock(), flexmock()]
|
|
|
+ flexmock(module).should_receive('make_dump_path').and_return('')
|
|
|
+ flexmock(module).should_receive('database_names_to_dump').and_return(('foo',)).and_return(
|
|
|
+ ('bar',)
|
|
|
+ )
|
|
|
+
|
|
|
+ for name, process in zip(('foo', 'bar'), processes):
|
|
|
+ flexmock(module).should_receive('execute_dump_command').with_args(
|
|
|
+ database={'name': name},
|
|
|
+ log_prefix=object,
|
|
|
+ dump_path=object,
|
|
|
+ database_names=(name,),
|
|
|
+ extra_environment=object,
|
|
|
+ dry_run=object,
|
|
|
+ dry_run_label=object,
|
|
|
+ ).and_return(process).once()
|
|
|
+
|
|
|
+ assert module.dump_databases(databases, {}, 'test.yaml', dry_run=False) == processes
|
|
|
+
|
|
|
+
|
|
|
+def test_dump_databases_dumps_with_password():
|
|
|
+ database = {'name': 'foo', 'username': 'root', 'password': 'trustsome1'}
|
|
|
+ process = flexmock()
|
|
|
+ flexmock(module).should_receive('make_dump_path').and_return('')
|
|
|
+ flexmock(module).should_receive('database_names_to_dump').and_return(('foo',)).and_return(
|
|
|
+ ('bar',)
|
|
|
+ )
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_dump_command').with_args(
|
|
|
+ database=database,
|
|
|
+ log_prefix=object,
|
|
|
+ dump_path=object,
|
|
|
+ database_names=('foo',),
|
|
|
+ extra_environment={'MYSQL_PWD': 'trustsome1'},
|
|
|
+ dry_run=object,
|
|
|
+ dry_run_label=object,
|
|
|
+ ).and_return(process).once()
|
|
|
+
|
|
|
+ assert module.dump_databases([database], {}, 'test.yaml', dry_run=False) == [process]
|
|
|
+
|
|
|
+
|
|
|
+def test_dump_databases_dumps_all_databases_at_once():
|
|
|
+ databases = [{'name': 'all'}]
|
|
|
+ process = flexmock()
|
|
|
+ flexmock(module).should_receive('make_dump_path').and_return('')
|
|
|
+ flexmock(module).should_receive('database_names_to_dump').and_return(('foo', 'bar'))
|
|
|
+ flexmock(module).should_receive('execute_dump_command').with_args(
|
|
|
+ database={'name': 'all'},
|
|
|
+ log_prefix=object,
|
|
|
+ dump_path=object,
|
|
|
+ database_names=('foo', 'bar'),
|
|
|
+ extra_environment=object,
|
|
|
+ dry_run=object,
|
|
|
+ dry_run_label=object,
|
|
|
+ ).and_return(process).once()
|
|
|
+
|
|
|
+ assert module.dump_databases(databases, {}, 'test.yaml', dry_run=False) == [process]
|
|
|
+
|
|
|
+
|
|
|
+def test_dump_databases_dumps_all_databases_separately_when_format_configured():
|
|
|
+ databases = [{'name': 'all', 'format': 'sql'}]
|
|
|
+ processes = [flexmock(), flexmock()]
|
|
|
+ flexmock(module).should_receive('make_dump_path').and_return('')
|
|
|
+ flexmock(module).should_receive('database_names_to_dump').and_return(('foo', 'bar'))
|
|
|
+
|
|
|
+ for name, process in zip(('foo', 'bar'), processes):
|
|
|
+ flexmock(module).should_receive('execute_dump_command').with_args(
|
|
|
+ database={'name': name, 'format': 'sql'},
|
|
|
+ log_prefix=object,
|
|
|
+ dump_path=object,
|
|
|
+ database_names=(name,),
|
|
|
+ extra_environment=object,
|
|
|
+ dry_run=object,
|
|
|
+ dry_run_label=object,
|
|
|
+ ).and_return(process).once()
|
|
|
+
|
|
|
+ assert module.dump_databases(databases, {}, 'test.yaml', dry_run=False) == processes
|
|
|
+
|
|
|
+
|
|
|
+def test_database_names_to_dump_runs_mariadb_with_list_options():
|
|
|
+ database = {'name': 'all', 'list_options': '--defaults-extra-file=mariadb.cnf'}
|
|
|
+ flexmock(module).should_receive('execute_command_and_capture_output').with_args(
|
|
|
+ (
|
|
|
+ 'mariadb',
|
|
|
+ '--defaults-extra-file=mariadb.cnf',
|
|
|
+ '--skip-column-names',
|
|
|
+ '--batch',
|
|
|
+ '--execute',
|
|
|
+ 'show schemas',
|
|
|
+ ),
|
|
|
+ extra_environment=None,
|
|
|
+ ).and_return(('foo\nbar')).once()
|
|
|
+
|
|
|
+ assert module.database_names_to_dump(database, None, 'test.yaml', '') == ('foo', 'bar')
|
|
|
+
|
|
|
+
|
|
|
+def test_execute_dump_command_runs_mariadb_dump():
|
|
|
+ process = flexmock()
|
|
|
+ flexmock(module.dump).should_receive('make_database_dump_filename').and_return('dump')
|
|
|
+ flexmock(module.os.path).should_receive('exists').and_return(False)
|
|
|
+ flexmock(module.dump).should_receive('create_named_pipe_for_dump')
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_command').with_args(
|
|
|
+ (
|
|
|
+ 'mariadb-dump',
|
|
|
+ '--add-drop-database',
|
|
|
+ '--databases',
|
|
|
+ 'foo',
|
|
|
+ '--result-file',
|
|
|
+ 'dump',
|
|
|
+ ),
|
|
|
+ extra_environment=None,
|
|
|
+ run_to_completion=False,
|
|
|
+ ).and_return(process).once()
|
|
|
+
|
|
|
+ assert (
|
|
|
+ module.execute_dump_command(
|
|
|
+ database={'name': 'foo'},
|
|
|
+ log_prefix='log',
|
|
|
+ dump_path=flexmock(),
|
|
|
+ database_names=('foo',),
|
|
|
+ extra_environment=None,
|
|
|
+ dry_run=False,
|
|
|
+ dry_run_label='',
|
|
|
+ )
|
|
|
+ == process
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_execute_dump_command_runs_mariadb_dump_without_add_drop_database():
|
|
|
+ process = flexmock()
|
|
|
+ flexmock(module.dump).should_receive('make_database_dump_filename').and_return('dump')
|
|
|
+ flexmock(module.os.path).should_receive('exists').and_return(False)
|
|
|
+ flexmock(module.dump).should_receive('create_named_pipe_for_dump')
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_command').with_args(
|
|
|
+ (
|
|
|
+ 'mariadb-dump',
|
|
|
+ '--databases',
|
|
|
+ 'foo',
|
|
|
+ '--result-file',
|
|
|
+ 'dump',
|
|
|
+ ),
|
|
|
+ extra_environment=None,
|
|
|
+ run_to_completion=False,
|
|
|
+ ).and_return(process).once()
|
|
|
+
|
|
|
+ assert (
|
|
|
+ module.execute_dump_command(
|
|
|
+ database={'name': 'foo', 'add_drop_database': False},
|
|
|
+ log_prefix='log',
|
|
|
+ dump_path=flexmock(),
|
|
|
+ database_names=('foo',),
|
|
|
+ extra_environment=None,
|
|
|
+ dry_run=False,
|
|
|
+ dry_run_label='',
|
|
|
+ )
|
|
|
+ == process
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_execute_dump_command_runs_mariadb_dump_with_hostname_and_port():
|
|
|
+ process = flexmock()
|
|
|
+ flexmock(module.dump).should_receive('make_database_dump_filename').and_return('dump')
|
|
|
+ flexmock(module.os.path).should_receive('exists').and_return(False)
|
|
|
+ flexmock(module.dump).should_receive('create_named_pipe_for_dump')
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_command').with_args(
|
|
|
+ (
|
|
|
+ 'mariadb-dump',
|
|
|
+ '--add-drop-database',
|
|
|
+ '--host',
|
|
|
+ 'database.example.org',
|
|
|
+ '--port',
|
|
|
+ '5433',
|
|
|
+ '--protocol',
|
|
|
+ 'tcp',
|
|
|
+ '--databases',
|
|
|
+ 'foo',
|
|
|
+ '--result-file',
|
|
|
+ 'dump',
|
|
|
+ ),
|
|
|
+ extra_environment=None,
|
|
|
+ run_to_completion=False,
|
|
|
+ ).and_return(process).once()
|
|
|
+
|
|
|
+ assert (
|
|
|
+ module.execute_dump_command(
|
|
|
+ database={'name': 'foo', 'hostname': 'database.example.org', 'port': 5433},
|
|
|
+ log_prefix='log',
|
|
|
+ dump_path=flexmock(),
|
|
|
+ database_names=('foo',),
|
|
|
+ extra_environment=None,
|
|
|
+ dry_run=False,
|
|
|
+ dry_run_label='',
|
|
|
+ )
|
|
|
+ == process
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_execute_dump_command_runs_mariadb_dump_with_username_and_password():
|
|
|
+ process = flexmock()
|
|
|
+ flexmock(module.dump).should_receive('make_database_dump_filename').and_return('dump')
|
|
|
+ flexmock(module.os.path).should_receive('exists').and_return(False)
|
|
|
+ flexmock(module.dump).should_receive('create_named_pipe_for_dump')
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_command').with_args(
|
|
|
+ (
|
|
|
+ 'mariadb-dump',
|
|
|
+ '--add-drop-database',
|
|
|
+ '--user',
|
|
|
+ 'root',
|
|
|
+ '--databases',
|
|
|
+ 'foo',
|
|
|
+ '--result-file',
|
|
|
+ 'dump',
|
|
|
+ ),
|
|
|
+ extra_environment={'MYSQL_PWD': 'trustsome1'},
|
|
|
+ run_to_completion=False,
|
|
|
+ ).and_return(process).once()
|
|
|
+
|
|
|
+ assert (
|
|
|
+ module.execute_dump_command(
|
|
|
+ database={'name': 'foo', 'username': 'root', 'password': 'trustsome1'},
|
|
|
+ log_prefix='log',
|
|
|
+ dump_path=flexmock(),
|
|
|
+ database_names=('foo',),
|
|
|
+ extra_environment={'MYSQL_PWD': 'trustsome1'},
|
|
|
+ dry_run=False,
|
|
|
+ dry_run_label='',
|
|
|
+ )
|
|
|
+ == process
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_execute_dump_command_runs_mariadb_dump_with_options():
|
|
|
+ process = flexmock()
|
|
|
+ flexmock(module.dump).should_receive('make_database_dump_filename').and_return('dump')
|
|
|
+ flexmock(module.os.path).should_receive('exists').and_return(False)
|
|
|
+ flexmock(module.dump).should_receive('create_named_pipe_for_dump')
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_command').with_args(
|
|
|
+ (
|
|
|
+ 'mariadb-dump',
|
|
|
+ '--stuff=such',
|
|
|
+ '--add-drop-database',
|
|
|
+ '--databases',
|
|
|
+ 'foo',
|
|
|
+ '--result-file',
|
|
|
+ 'dump',
|
|
|
+ ),
|
|
|
+ extra_environment=None,
|
|
|
+ run_to_completion=False,
|
|
|
+ ).and_return(process).once()
|
|
|
+
|
|
|
+ assert (
|
|
|
+ module.execute_dump_command(
|
|
|
+ database={'name': 'foo', 'options': '--stuff=such'},
|
|
|
+ log_prefix='log',
|
|
|
+ dump_path=flexmock(),
|
|
|
+ database_names=('foo',),
|
|
|
+ extra_environment=None,
|
|
|
+ dry_run=False,
|
|
|
+ dry_run_label='',
|
|
|
+ )
|
|
|
+ == process
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_execute_dump_command_with_duplicate_dump_skips_mariadb_dump():
|
|
|
+ flexmock(module.dump).should_receive('make_database_dump_filename').and_return('dump')
|
|
|
+ flexmock(module.os.path).should_receive('exists').and_return(True)
|
|
|
+ flexmock(module.dump).should_receive('create_named_pipe_for_dump').never()
|
|
|
+ flexmock(module).should_receive('execute_command').never()
|
|
|
+
|
|
|
+ assert (
|
|
|
+ module.execute_dump_command(
|
|
|
+ database={'name': 'foo'},
|
|
|
+ log_prefix='log',
|
|
|
+ dump_path=flexmock(),
|
|
|
+ database_names=('foo',),
|
|
|
+ extra_environment=None,
|
|
|
+ dry_run=True,
|
|
|
+ dry_run_label='SO DRY',
|
|
|
+ )
|
|
|
+ is None
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_execute_dump_command_with_dry_run_skips_mariadb_dump():
|
|
|
+ flexmock(module.dump).should_receive('make_database_dump_filename').and_return('dump')
|
|
|
+ flexmock(module.os.path).should_receive('exists').and_return(False)
|
|
|
+ flexmock(module.dump).should_receive('create_named_pipe_for_dump')
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_command').never()
|
|
|
+
|
|
|
+ assert (
|
|
|
+ module.execute_dump_command(
|
|
|
+ database={'name': 'foo'},
|
|
|
+ log_prefix='log',
|
|
|
+ dump_path=flexmock(),
|
|
|
+ database_names=('foo',),
|
|
|
+ extra_environment=None,
|
|
|
+ dry_run=True,
|
|
|
+ dry_run_label='SO DRY',
|
|
|
+ )
|
|
|
+ is None
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_dump_databases_errors_for_missing_all_databases():
|
|
|
+ databases = [{'name': 'all'}]
|
|
|
+ flexmock(module).should_receive('make_dump_path').and_return('')
|
|
|
+ flexmock(module.dump).should_receive('make_database_dump_filename').and_return(
|
|
|
+ 'databases/localhost/all'
|
|
|
+ )
|
|
|
+ flexmock(module).should_receive('database_names_to_dump').and_return(())
|
|
|
+
|
|
|
+ with pytest.raises(ValueError):
|
|
|
+ assert module.dump_databases(databases, {}, 'test.yaml', dry_run=False)
|
|
|
+
|
|
|
+
|
|
|
+def test_dump_databases_does_not_error_for_missing_all_databases_with_dry_run():
|
|
|
+ databases = [{'name': 'all'}]
|
|
|
+ flexmock(module).should_receive('make_dump_path').and_return('')
|
|
|
+ flexmock(module.dump).should_receive('make_database_dump_filename').and_return(
|
|
|
+ 'databases/localhost/all'
|
|
|
+ )
|
|
|
+ flexmock(module).should_receive('database_names_to_dump').and_return(())
|
|
|
+
|
|
|
+ assert module.dump_databases(databases, {}, 'test.yaml', dry_run=True) == []
|
|
|
+
|
|
|
+
|
|
|
+def test_restore_database_dump_runs_mariadb_to_restore():
|
|
|
+ databases_config = [{'name': 'foo'}, {'name': 'bar'}]
|
|
|
+ extract_process = flexmock(stdout=flexmock())
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_command_with_processes').with_args(
|
|
|
+ ('mariadb', '--batch'),
|
|
|
+ processes=[extract_process],
|
|
|
+ output_log_level=logging.DEBUG,
|
|
|
+ input_file=extract_process.stdout,
|
|
|
+ extra_environment=None,
|
|
|
+ ).once()
|
|
|
+
|
|
|
+ module.restore_database_dump(
|
|
|
+ databases_config,
|
|
|
+ {},
|
|
|
+ 'test.yaml',
|
|
|
+ database_name='foo',
|
|
|
+ dry_run=False,
|
|
|
+ extract_process=extract_process,
|
|
|
+ connection_params={
|
|
|
+ 'hostname': None,
|
|
|
+ 'port': None,
|
|
|
+ 'username': None,
|
|
|
+ 'password': None,
|
|
|
+ },
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_restore_database_dump_errors_when_database_missing_from_configuration():
|
|
|
+ databases_config = [{'name': 'foo'}, {'name': 'bar'}]
|
|
|
+ extract_process = flexmock(stdout=flexmock())
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_command_with_processes').never()
|
|
|
+
|
|
|
+ with pytest.raises(ValueError):
|
|
|
+ module.restore_database_dump(
|
|
|
+ databases_config,
|
|
|
+ {},
|
|
|
+ 'test.yaml',
|
|
|
+ database_name='other',
|
|
|
+ dry_run=False,
|
|
|
+ extract_process=extract_process,
|
|
|
+ connection_params={
|
|
|
+ 'hostname': None,
|
|
|
+ 'port': None,
|
|
|
+ 'username': None,
|
|
|
+ 'password': None,
|
|
|
+ },
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_restore_database_dump_runs_mariadb_with_options():
|
|
|
+ databases_config = [{'name': 'foo', 'restore_options': '--harder'}]
|
|
|
+ extract_process = flexmock(stdout=flexmock())
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_command_with_processes').with_args(
|
|
|
+ ('mariadb', '--batch', '--harder'),
|
|
|
+ processes=[extract_process],
|
|
|
+ output_log_level=logging.DEBUG,
|
|
|
+ input_file=extract_process.stdout,
|
|
|
+ extra_environment=None,
|
|
|
+ ).once()
|
|
|
+
|
|
|
+ module.restore_database_dump(
|
|
|
+ databases_config,
|
|
|
+ {},
|
|
|
+ 'test.yaml',
|
|
|
+ database_name='foo',
|
|
|
+ dry_run=False,
|
|
|
+ extract_process=extract_process,
|
|
|
+ connection_params={
|
|
|
+ 'hostname': None,
|
|
|
+ 'port': None,
|
|
|
+ 'username': None,
|
|
|
+ 'password': None,
|
|
|
+ },
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_restore_database_dump_runs_mariadb_with_hostname_and_port():
|
|
|
+ databases_config = [{'name': 'foo', 'hostname': 'database.example.org', 'port': 5433}]
|
|
|
+ extract_process = flexmock(stdout=flexmock())
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_command_with_processes').with_args(
|
|
|
+ (
|
|
|
+ 'mariadb',
|
|
|
+ '--batch',
|
|
|
+ '--host',
|
|
|
+ 'database.example.org',
|
|
|
+ '--port',
|
|
|
+ '5433',
|
|
|
+ '--protocol',
|
|
|
+ 'tcp',
|
|
|
+ ),
|
|
|
+ processes=[extract_process],
|
|
|
+ output_log_level=logging.DEBUG,
|
|
|
+ input_file=extract_process.stdout,
|
|
|
+ extra_environment=None,
|
|
|
+ ).once()
|
|
|
+
|
|
|
+ module.restore_database_dump(
|
|
|
+ databases_config,
|
|
|
+ {},
|
|
|
+ 'test.yaml',
|
|
|
+ database_name='foo',
|
|
|
+ dry_run=False,
|
|
|
+ extract_process=extract_process,
|
|
|
+ connection_params={
|
|
|
+ 'hostname': None,
|
|
|
+ 'port': None,
|
|
|
+ 'username': None,
|
|
|
+ 'password': None,
|
|
|
+ },
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_restore_database_dump_runs_mariadb_with_username_and_password():
|
|
|
+ databases_config = [{'name': 'foo', 'username': 'root', 'password': 'trustsome1'}]
|
|
|
+ extract_process = flexmock(stdout=flexmock())
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_command_with_processes').with_args(
|
|
|
+ ('mariadb', '--batch', '--user', 'root'),
|
|
|
+ processes=[extract_process],
|
|
|
+ output_log_level=logging.DEBUG,
|
|
|
+ input_file=extract_process.stdout,
|
|
|
+ extra_environment={'MYSQL_PWD': 'trustsome1'},
|
|
|
+ ).once()
|
|
|
+
|
|
|
+ module.restore_database_dump(
|
|
|
+ databases_config,
|
|
|
+ {},
|
|
|
+ 'test.yaml',
|
|
|
+ database_name='foo',
|
|
|
+ dry_run=False,
|
|
|
+ extract_process=extract_process,
|
|
|
+ connection_params={
|
|
|
+ 'hostname': None,
|
|
|
+ 'port': None,
|
|
|
+ 'username': None,
|
|
|
+ 'password': None,
|
|
|
+ },
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_restore_database_dump_with_connection_params_uses_connection_params_for_restore():
|
|
|
+ databases_config = [
|
|
|
+ {
|
|
|
+ 'name': 'foo',
|
|
|
+ 'username': 'root',
|
|
|
+ 'password': 'trustsome1',
|
|
|
+ 'restore_hostname': 'restorehost',
|
|
|
+ 'restore_port': 'restoreport',
|
|
|
+ 'restore_username': 'restoreusername',
|
|
|
+ 'restore_password': 'restorepassword',
|
|
|
+ }
|
|
|
+ ]
|
|
|
+ extract_process = flexmock(stdout=flexmock())
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_command_with_processes').with_args(
|
|
|
+ (
|
|
|
+ 'mariadb',
|
|
|
+ '--batch',
|
|
|
+ '--host',
|
|
|
+ 'clihost',
|
|
|
+ '--port',
|
|
|
+ 'cliport',
|
|
|
+ '--protocol',
|
|
|
+ 'tcp',
|
|
|
+ '--user',
|
|
|
+ 'cliusername',
|
|
|
+ ),
|
|
|
+ processes=[extract_process],
|
|
|
+ output_log_level=logging.DEBUG,
|
|
|
+ input_file=extract_process.stdout,
|
|
|
+ extra_environment={'MYSQL_PWD': 'clipassword'},
|
|
|
+ ).once()
|
|
|
+
|
|
|
+ module.restore_database_dump(
|
|
|
+ databases_config,
|
|
|
+ {},
|
|
|
+ 'test.yaml',
|
|
|
+ database_name='foo',
|
|
|
+ dry_run=False,
|
|
|
+ extract_process=extract_process,
|
|
|
+ connection_params={
|
|
|
+ 'hostname': 'clihost',
|
|
|
+ 'port': 'cliport',
|
|
|
+ 'username': 'cliusername',
|
|
|
+ 'password': 'clipassword',
|
|
|
+ },
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_restore_database_dump_without_connection_params_uses_restore_params_in_config_for_restore():
|
|
|
+ databases_config = [
|
|
|
+ {
|
|
|
+ 'name': 'foo',
|
|
|
+ 'username': 'root',
|
|
|
+ 'password': 'trustsome1',
|
|
|
+ 'hostname': 'dbhost',
|
|
|
+ 'port': 'dbport',
|
|
|
+ 'restore_username': 'restoreuser',
|
|
|
+ 'restore_password': 'restorepass',
|
|
|
+ 'restore_hostname': 'restorehost',
|
|
|
+ 'restore_port': 'restoreport',
|
|
|
+ }
|
|
|
+ ]
|
|
|
+ extract_process = flexmock(stdout=flexmock())
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_command_with_processes').with_args(
|
|
|
+ (
|
|
|
+ 'mariadb',
|
|
|
+ '--batch',
|
|
|
+ '--host',
|
|
|
+ 'restorehost',
|
|
|
+ '--port',
|
|
|
+ 'restoreport',
|
|
|
+ '--protocol',
|
|
|
+ 'tcp',
|
|
|
+ '--user',
|
|
|
+ 'restoreuser',
|
|
|
+ ),
|
|
|
+ processes=[extract_process],
|
|
|
+ output_log_level=logging.DEBUG,
|
|
|
+ input_file=extract_process.stdout,
|
|
|
+ extra_environment={'MYSQL_PWD': 'restorepass'},
|
|
|
+ ).once()
|
|
|
+
|
|
|
+ module.restore_database_dump(
|
|
|
+ databases_config,
|
|
|
+ {},
|
|
|
+ 'test.yaml',
|
|
|
+ database_name='foo',
|
|
|
+ dry_run=False,
|
|
|
+ extract_process=extract_process,
|
|
|
+ connection_params={
|
|
|
+ 'hostname': None,
|
|
|
+ 'port': None,
|
|
|
+ 'username': None,
|
|
|
+ 'password': None,
|
|
|
+ },
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_restore_database_dump_with_dry_run_skips_restore():
|
|
|
+ databases_config = [{'name': 'foo'}]
|
|
|
+
|
|
|
+ flexmock(module).should_receive('execute_command_with_processes').never()
|
|
|
+
|
|
|
+ module.restore_database_dump(
|
|
|
+ databases_config,
|
|
|
+ {},
|
|
|
+ 'test.yaml',
|
|
|
+ database_name='foo',
|
|
|
+ dry_run=True,
|
|
|
+ extract_process=flexmock(),
|
|
|
+ connection_params={
|
|
|
+ 'hostname': None,
|
|
|
+ 'port': None,
|
|
|
+ 'username': None,
|
|
|
+ 'password': None,
|
|
|
+ },
|
|
|
+ )
|