def test_current_hour_with_pending_backup(self, get_worker_name_mock,
task_register, save_backup_group,
make_instance_snapshot_backup):
infra_mock = InfraHelper.create(name='pending_backup_test',
backup_hour=self.backup_hour - 1,
plan__has_persistence=True,
environment=self.dev_env,
plan=self.plan)
DatabaseHelper.create(databaseinfra=infra_mock,
environment=self.dev_env)
instance_mock = mommy.make('Instance', databaseinfra=infra_mock)
get_worker_name_mock.return_value = 'test'
group = BackupGroup()
save_backup_group.return_value = group
make_instance_snapshot_backup.return_value.status.return_value = (
Snapshot.SUCCESS)
make_databases_backup()
calls = [
call(current_hour=self.backup_hour,
instance=self.instance,
error={},
group=group),
call(current_hour=self.backup_hour,
instance=instance_mock,
error={},
group=group)
]
make_instance_snapshot_backup.assert_has_calls(calls, any_order=True)
def test_current_hour_without_pending_backup(
self, get_worker_name_mock, task_register, save_backup_group,
make_instance_snapshot_backup):
infra_mock = InfraHelper.create(
name='backup_test',
backup_hour=self.backup_hour - 1,
plan__has_persistence=True,
environment=self.dev_env,
plan=self.plan,
)
DatabaseHelper.create(databaseinfra=infra_mock,
environment=self.dev_env)
instance_mock = mommy.make('Instance', databaseinfra=infra_mock)
get_worker_name_mock.return_value = 'test'
group = BackupGroup()
save_backup_group.return_value = group
snapshot = mommy.make('Snapshot',
instance=instance_mock,
group=group,
status=Snapshot.SUCCESS,
end_at=FAKE_NOW - timedelta(hours=1))
make_instance_snapshot_backup.return_value = snapshot
make_databases_backup()
make_instance_snapshot_backup.assert_called_once_with(
current_hour=self.backup_hour,
instance=self.instance,
error={},
group=group)
def test_backup_current_hour(self, get_worker_name_mock, task_register,
save_backup_group,
make_instance_snapshot_backup):
get_worker_name_mock.return_value = 'test'
group = BackupGroup()
save_backup_group.return_value = group
make_instance_snapshot_backup.return_value.status.return_value = (
Snapshot.SUCCESS)
make_databases_backup()
make_instance_snapshot_backup.assert_called_with(
current_hour=self.backup_hour,
instance=self.instance,
error={},
group=group)
def test_snapshot_with_error(self, get_worker_name_mock, task_register,
save_backup_group,
make_instance_snapshot_backup):
get_worker_name_mock.return_value = 'test'
group = BackupGroup()
save_backup_group.return_value = group
snapshot = mommy.make('Snapshot',
instance=self.instance,
group=group,
status=Snapshot.ERROR)
make_instance_snapshot_backup.return_value = snapshot
make_databases_backup()
make_instance_snapshot_backup.assertEqual(snapshot.status,
Snapshot.ERROR)