82 lines
2.7 KiB
Python
82 lines
2.7 KiB
Python
|
# Migration test
|
||
|
|
#
|
||
|
|
# 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 tempfile
|
||
|
|
from avocado_qemu import Test
|
||
|
|
from avocado import skipUnless
|
||
|
|
|
||
|
|
from avocado.utils import network
|
||
|
|
from avocado.utils import wait
|
||
|
|
from avocado.utils.path import find_command
|
||
|
|
|
||
|
|
|
||
|
|
class Migration(Test):
|
||
|
|
"""
|
||
|
|
:avocado: tags=migration
|
||
|
|
"""
|
||
|
|
|
||
|
|
timeout = 10
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def migration_finished(vm):
|
||
|
|
return vm.command('query-migrate')['status'] in ('completed', 'failed')
|
||
|
|
|
||
|
|
def assert_migration(self, src_vm, dst_vm):
|
||
|
|
wait.wait_for(self.migration_finished,
|
||
|
|
timeout=self.timeout,
|
||
|
|
step=0.1,
|
||
|
|
args=(src_vm,))
|
||
|
|
wait.wait_for(self.migration_finished,
|
||
|
|
timeout=self.timeout,
|
||
|
|
step=0.1,
|
||
|
|
args=(dst_vm,))
|
||
|
|
self.assertEqual(src_vm.command('query-migrate')['status'], 'completed')
|
||
|
|
self.assertEqual(dst_vm.command('query-migrate')['status'], 'completed')
|
||
|
|
self.assertEqual(dst_vm.command('query-status')['status'], 'running')
|
||
|
|
self.assertEqual(src_vm.command('query-status')['status'],'postmigrate')
|
||
|
|
|
||
|
|
def do_migrate(self, dest_uri, src_uri=None):
|
||
|
|
dest_vm = self.get_vm('-incoming', dest_uri)
|
||
|
|
dest_vm.add_args('-nodefaults')
|
||
|
|
dest_vm.launch()
|
||
|
|
if src_uri is None:
|
||
|
|
src_uri = dest_uri
|
||
|
|
source_vm = self.get_vm()
|
||
|
|
source_vm.add_args('-nodefaults')
|
||
|
|
source_vm.launch()
|
||
|
|
source_vm.qmp('migrate', uri=src_uri)
|
||
|
|
self.assert_migration(source_vm, dest_vm)
|
||
|
|
|
||
|
|
def _get_free_port(self):
|
||
|
|
port = network.find_free_port()
|
||
|
|
if port is None:
|
||
|
|
self.cancel('Failed to find a free port')
|
||
|
|
return port
|
||
|
|
|
||
|
|
|
||
|
|
def test_migration_with_tcp_localhost(self):
|
||
|
|
dest_uri = 'tcp:localhost:%u' % self._get_free_port()
|
||
|
|
self.do_migrate(dest_uri)
|
||
|
|
|
||
|
|
def test_migration_with_unix(self):
|
||
|
|
with tempfile.TemporaryDirectory(prefix='socket_') as socket_path:
|
||
|
|
dest_uri = 'unix:%s/qemu-test.sock' % socket_path
|
||
|
|
self.do_migrate(dest_uri)
|
||
|
|
|
||
|
|
@skipUnless(find_command('nc', default=False), "'nc' command not found")
|
||
|
|
def test_migration_with_exec(self):
|
||
|
|
"""The test works for both netcat-traditional and netcat-openbsd packages."""
|
||
|
|
free_port = self._get_free_port()
|
||
|
|
dest_uri = 'exec:nc -l localhost %u' % free_port
|
||
|
|
src_uri = 'exec:nc localhost %u' % free_port
|
||
|
|
self.do_migrate(dest_uri, src_uri)
|