def test_dump_databases_runs_mysqldump_with_hostname_and_port():
databases = [{
'name': 'foo',
'hostname': 'database.example.org',
'port': 5433
}]
output_file = flexmock()
flexmock(module).should_receive('make_dump_path').and_return('')
flexmock(
module.dump).should_receive('make_database_dump_filename').and_return(
'databases/database.example.org/foo')
flexmock(module.os).should_receive('makedirs')
flexmock(
sys.modules['builtins']).should_receive('open').and_return(output_file)
flexmock(module).should_receive('execute_command').with_args(
(
'mysqldump',
'--add-drop-database',
'--host',
'database.example.org',
'--port',
'5433',
'--protocol',
'tcp',
'--databases',
'foo',
),
output_file=output_file,
extra_environment=None,
).once()
module.dump_databases(databases, 'test.yaml', {}, dry_run=False)
def test_dump_databases_with_dry_run_skips_mysqldump():
databases = [{'name': 'foo'}, {'name': 'bar'}]
flexmock(module).should_receive('make_dump_path').and_return('')
flexmock(
module.dump).should_receive('make_database_dump_filename').and_return(
'databases/localhost/foo').and_return('databases/localhost/bar')
flexmock(module.os).should_receive('makedirs').never()
flexmock(module).should_receive('execute_command').never()
module.dump_databases(databases, 'test.yaml', {}, dry_run=True)
def test_dump_databases_runs_mysqldump_for_each_database():
databases = [{'name': 'foo'}, {'name': 'bar'}]
processes = [flexmock(), flexmock()]
flexmock(module).should_receive('make_dump_path').and_return('')
flexmock(
module.dump).should_receive('make_database_dump_filename').and_return(
'databases/localhost/foo').and_return('databases/localhost/bar')
flexmock(module).should_receive('database_names_to_dump').and_return(
('foo', )).and_return(('bar', ))
flexmock(module.dump).should_receive('create_named_pipe_for_dump')
for name, process in zip(('foo', 'bar'), processes):
flexmock(module).should_receive('execute_command').with_args(
(
'mysqldump',
'--add-drop-database',
'--databases',
name,
'>',
'databases/localhost/{}'.format(name),
),
shell=True,
extra_environment=None,
run_to_completion=False,
).and_return(process).once()
assert module.dump_databases(databases, 'test.yaml', {},
dry_run=False) == processes
def test_dump_databases_runs_mysqldump_for_all_databases():
databases = [{'name': 'all'}]
process = flexmock()
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(
('foo', 'bar'))
flexmock(module.dump).should_receive('create_named_pipe_for_dump')
flexmock(module).should_receive('execute_command').with_args(
(
'mysqldump',
'--add-drop-database',
'--databases',
'foo',
'bar',
'>',
'databases/localhost/all',
),
shell=True,
extra_environment=None,
run_to_completion=False,
).and_return(process).once()
assert module.dump_databases(databases, 'test.yaml', {},
dry_run=False) == [process]
def test_dump_databases_runs_mysqldump_with_username_and_password():
databases = [{'name': 'foo', 'username': '******', 'password': '******'}]
process = flexmock()
flexmock(module).should_receive('make_dump_path').and_return('')
flexmock(module.dump).should_receive(
'make_database_dump_filename').and_return('databases/localhost/foo')
flexmock(module).should_receive('database_names_to_dump').and_return(
('foo', ))
flexmock(module.dump).should_receive('create_named_pipe_for_dump')
flexmock(module).should_receive('execute_command').with_args(
(
'mysqldump',
'--add-drop-database',
'--user',
'root',
'--databases',
'foo',
'>',
'databases/localhost/foo',
),
shell=True,
extra_environment={
'MYSQL_PWD': 'trustsome1'
},
run_to_completion=False,
).and_return(process).once()
assert module.dump_databases(databases, 'test.yaml', {},
dry_run=False) == [process]
def test_dump_databases_runs_mysqldump_for_all_databases():
databases = [{'name': 'all'}]
output_file = flexmock()
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.os).should_receive('makedirs')
flexmock(
sys.modules['builtins']).should_receive('open').and_return(output_file)
flexmock(module).should_receive('execute_command').with_args(
('mysqldump', '--add-drop-database', '--all-databases'),
output_file=output_file,
extra_environment=None,
).once()
module.dump_databases(databases, 'test.yaml', {}, dry_run=False)
def test_dump_databases_runs_mysqldump_with_username_and_password():
databases = [{'name': 'foo', 'username': '******', 'password': '******'}]
output_file = flexmock()
flexmock(module).should_receive('make_dump_path').and_return('')
flexmock(module.dump).should_receive(
'make_database_dump_filename').and_return('databases/localhost/foo')
flexmock(module.os).should_receive('makedirs')
flexmock(
sys.modules['builtins']).should_receive('open').and_return(output_file)
flexmock(module).should_receive('execute_command').with_args(
('mysqldump', '--add-drop-database', '--user', 'root', '--databases',
'foo'),
output_file=output_file,
extra_environment={
'MYSQL_PWD': 'trustsome1'
},
).once()
module.dump_databases(databases, 'test.yaml', {}, dry_run=False)
def test_dump_databases_errors_for_missing_all_databases():
databases = [{'name': 'all'}]
process = flexmock()
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) == [process]
def test_dump_databases_runs_mysqldump_with_hostname_and_port():
databases = [{
'name': 'foo',
'hostname': 'database.example.org',
'port': 5433
}]
process = flexmock()
flexmock(module).should_receive('make_dump_path').and_return('')
flexmock(
module.dump).should_receive('make_database_dump_filename').and_return(
'databases/database.example.org/foo')
flexmock(module).should_receive('database_names_to_dump').and_return(
('foo', ))
flexmock(module.dump).should_receive('create_named_pipe_for_dump')
flexmock(module).should_receive('execute_command').with_args(
(
'mysqldump',
'--add-drop-database',
'--host',
'database.example.org',
'--port',
'5433',
'--protocol',
'tcp',
'--databases',
'foo',
'>',
'databases/database.example.org/foo',
),
shell=True,
extra_environment=None,
run_to_completion=False,
).and_return(process).once()
assert module.dump_databases(databases, 'test.yaml', {},
dry_run=False) == [process]