def rebuild_by_snapshot(context, server):
server_id = server.id
snapshot_ensure = "snapshot-{}-ensure".format(server_id)
snapshot_flow = snapshot_tasks.migrate_snapshot(context, server)
return [], [], [snapshot_flow], snapshot_ensure
def test_migrate_snapshot(self, flow_mock,
ensure_snapshot_mock,
ensure_image_mock):
flow = snapshot.migrate_snapshot(
self.context,
self.test_server_info,
)
flow_mock.assert_called_once_with("migrate-ephemeral-"
"storage-server-{}"
.format(self.test_server_id))
self.assertEqual(flow.add.call_args_list,
[call(ensure_snapshot_mock()),
call(ensure_image_mock())])