3 # Copyright (c) 2019 Red Hat, Inc.
6 # Cleber Rosa <crosa@redhat.com>
7 # Caio Carrara <ccarrara@redhat.com>
9 # This work is licensed under the terms of the GNU GPL, version 2 or
10 # later. See the COPYING file in the top-level directory.
14 from avocado_qemu
import Test
15 from avocado
import skipUnless
17 from avocado
.utils
import network
18 from avocado
.utils
import wait
19 from avocado
.utils
.path
import find_command
22 class Migration(Test
):
24 :avocado: tags=migration
30 def migration_finished(vm
):
31 return vm
.command('query-migrate')['status'] in ('completed', 'failed')
33 def assert_migration(self
, src_vm
, dst_vm
):
34 wait
.wait_for(self
.migration_finished
,
38 self
.assertEqual(src_vm
.command('query-migrate')['status'], 'completed')
39 self
.assertEqual(dst_vm
.command('query-migrate')['status'], 'completed')
40 self
.assertEqual(dst_vm
.command('query-status')['status'], 'running')
41 self
.assertEqual(src_vm
.command('query-status')['status'],'postmigrate')
43 def do_migrate(self
, dest_uri
, src_uri
=None):
44 dest_vm
= self
.get_vm('-incoming', dest_uri
)
45 dest_vm
.add_args('-nodefaults')
49 source_vm
= self
.get_vm()
50 source_vm
.add_args('-nodefaults')
52 source_vm
.qmp('migrate', uri
=src_uri
)
53 self
.assert_migration(source_vm
, dest_vm
)
55 def _get_free_port(self
):
56 port
= network
.find_free_port()
58 self
.cancel('Failed to find a free port')
62 def test_migration_with_tcp_localhost(self
):
63 dest_uri
= 'tcp:localhost:%u' % self
._get
_free
_port
()
64 self
.do_migrate(dest_uri
)
66 def test_migration_with_unix(self
):
67 with tempfile
.TemporaryDirectory(prefix
='socket_') as socket_path
:
68 dest_uri
= 'unix:%s/qemu-test.sock' % socket_path
69 self
.do_migrate(dest_uri
)
71 @skipUnless(find_command('nc', default
=False), "'nc' command not found")
72 def test_migration_with_exec(self
):
73 """The test works for both netcat-traditional and netcat-openbsd packages."""
74 free_port
= self
._get
_free
_port
()
75 dest_uri
= 'exec:nc -l localhost %u' % free_port
76 src_uri
= 'exec:nc localhost %u' % free_port
77 self
.do_migrate(dest_uri
, src_uri
)