mirror of
https://github.com/qemu/qemu.git
synced 2026-07-08 17:46:10 +00:00
cb8f285ce9
Introduce configure_machine, launch_source_vm and assert_dest_vm methods to allow child classes to override some pieces of source/dest VMs creation, start and check logic. Reviewed-by: Peter Xu <peterx@redhat.com> Acked-by: Stefan Hajnoczi <stefanha@redhat.com> Signed-off-by: Alexander Mikhalitsyn <aleksandr.mikhalitsyn@futurfusion.io> Signed-off-by: Klaus Jensen <k.jensen@samsung.com>
107 lines
3.5 KiB
Python
107 lines
3.5 KiB
Python
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
#
|
|
# Migration test base class
|
|
#
|
|
# Copyright (c) 2019 Red Hat, Inc.
|
|
#
|
|
# Authors:
|
|
# Cleber Rosa <crosa@redhat.com>
|
|
# Caio Carrara <ccarrara@redhat.com>
|
|
#
|
|
# This work is licensed under the terms of the GNU GPL, version 2 or
|
|
# later. See the COPYING file in the top-level directory.
|
|
|
|
import time
|
|
|
|
from qemu_test import QemuSystemTest, which
|
|
from qemu_test.ports import Ports
|
|
|
|
|
|
class MigrationTest(QemuSystemTest):
|
|
|
|
timeout = 10
|
|
|
|
@staticmethod
|
|
def migration_finished(vm):
|
|
return vm.cmd('query-migrate')['status'] in ('completed', 'failed')
|
|
|
|
def assert_migration(self, src_vm, dst_vm):
|
|
|
|
end = time.monotonic() + self.timeout
|
|
while time.monotonic() < end and not self.migration_finished(src_vm):
|
|
time.sleep(0.1)
|
|
|
|
end = time.monotonic() + self.timeout
|
|
while time.monotonic() < end and not self.migration_finished(dst_vm):
|
|
time.sleep(0.1)
|
|
|
|
self.assertEqual(src_vm.cmd('query-migrate')['status'], 'completed')
|
|
self.assertEqual(dst_vm.cmd('query-migrate')['status'], 'completed')
|
|
self.assertEqual(dst_vm.cmd('query-status')['status'], 'running')
|
|
self.assertEqual(src_vm.cmd('query-status')['status'],'postmigrate')
|
|
|
|
# Can be overridden by subclasses to configure both source/dest VMs.
|
|
def configure_machine(self, vm):
|
|
vm.add_args('-nodefaults')
|
|
|
|
# Can be overridden by subclasses to prepare the source VM before
|
|
# migration, e.g. by running some workload inside the source VM
|
|
# to see if it continues to run properly after migration.
|
|
def launch_source_vm(self, vm):
|
|
vm.launch()
|
|
|
|
# Can be overridden by subclasses to check the destination VM after
|
|
# migration, e.g. by checking if the workload is still running after
|
|
# migration.
|
|
def assert_dest_vm(self, vm):
|
|
pass
|
|
|
|
def migrate_vms(self, dst_uri, src_uri, dst_vm, src_vm):
|
|
dst_vm.qmp('migrate-incoming', uri=dst_uri)
|
|
src_vm.qmp('migrate', uri=src_uri)
|
|
self.assert_migration(src_vm, dst_vm)
|
|
self.assert_dest_vm(dst_vm)
|
|
|
|
def migrate(self, dst_uri, src_uri=None):
|
|
dst_vm = self.get_vm('-incoming', 'defer', name="dst-qemu")
|
|
self.configure_machine(dst_vm)
|
|
dst_vm.launch()
|
|
|
|
src_vm = self.get_vm(name="src-qemu")
|
|
self.configure_machine(src_vm)
|
|
self.launch_source_vm(src_vm)
|
|
|
|
if src_uri is None:
|
|
src_uri = dst_uri
|
|
|
|
self.migrate_vms(dst_uri, src_uri, dst_vm, src_vm)
|
|
|
|
def _get_free_port(self, ports):
|
|
port = ports.find_free_port()
|
|
if port is None:
|
|
self.skipTest('Failed to find a free port')
|
|
return port
|
|
|
|
def migration_with_tcp_localhost_vms(self, dst_vm, src_vm):
|
|
with Ports() as ports:
|
|
uri = 'tcp:localhost:%u' % self._get_free_port(ports)
|
|
self.migrate_vms(uri, uri, dst_vm, src_vm)
|
|
|
|
def migration_with_tcp_localhost(self):
|
|
with Ports() as ports:
|
|
dst_uri = 'tcp:localhost:%u' % self._get_free_port(ports)
|
|
self.migrate(dst_uri)
|
|
|
|
def migration_with_unix(self):
|
|
dst_uri = 'unix:%s/migration.sock' % self.socket_dir().name
|
|
self.migrate(dst_uri)
|
|
|
|
def migration_with_exec(self):
|
|
if not which('socat'):
|
|
self.skipTest('socat is not available')
|
|
with Ports() as ports:
|
|
free_port = self._get_free_port(ports)
|
|
dst_uri = 'exec:socat TCP-LISTEN:%u -' % free_port
|
|
src_uri = 'exec:socat - TCP:localhost:%u,forever' % free_port
|
|
self.migrate(dst_uri, src_uri)
|