summaryrefslogtreecommitdiff
path: root/tests/qemu-iotests/194
blob: a3e3bad664b9d6aada7fbcbe7db907408a7c06de (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#!/usr/bin/env python
#
# Copyright (C) 2017 Red Hat, Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# Creator/Owner: Stefan Hajnoczi <stefanha@redhat.com>
#
# Non-shared storage migration test using NBD server and drive-mirror

import os
import atexit
import iotests

iotests.verify_platform(['linux'])

img_size = '1G'
source_img_path = os.path.join(iotests.test_dir, 'source.img')
dest_img_path = os.path.join(iotests.test_dir, 'dest.img')
iotests.qemu_img_pipe('create', '-f', iotests.imgfmt, source_img_path, img_size)
iotests.qemu_img_pipe('create', '-f', iotests.imgfmt, dest_img_path, img_size)

iotests.log('Launching VMs...')
migration_sock_path = os.path.join(iotests.test_dir, 'migration.sock')
nbd_sock_path = os.path.join(iotests.test_dir, 'nbd.sock')
source_vm = iotests.VM('source').add_drive(source_img_path)
dest_vm = (iotests.VM('dest').add_drive(dest_img_path)
                             .add_incoming('unix:{0}'.format(migration_sock_path)))
source_vm.launch()
atexit.register(source_vm.shutdown)
dest_vm.launch()
atexit.register(dest_vm.shutdown)

iotests.log('Launching NBD server on destination...')
iotests.log(dest_vm.qmp('nbd-server-start', addr={'type': 'unix', 'data': {'path': nbd_sock_path}}))
iotests.log(dest_vm.qmp('nbd-server-add', device='drive0', writable=True))

iotests.log('Starting `drive-mirror` on source...')
iotests.log(source_vm.qmp(
              'drive-mirror',
              device='drive0',
              target='nbd+unix:///drive0?socket={0}'.format(nbd_sock_path),
              sync='full',
              format='raw', # always raw, the server handles the format
              mode='existing',
              job_id='mirror-job0'))

iotests.log('Waiting for `drive-mirror` to complete...')
iotests.log(source_vm.event_wait('BLOCK_JOB_READY'),
            filters=[iotests.filter_qmp_event])

iotests.log('Starting migration...')
source_vm.qmp('migrate-set-capabilities',
              capabilities=[{'capability': 'events', 'state': True}])
dest_vm.qmp('migrate-set-capabilities',
            capabilities=[{'capability': 'events', 'state': True}])
iotests.log(source_vm.qmp('migrate', uri='unix:{0}'.format(migration_sock_path)))

while True:
    event1 = source_vm.event_wait('MIGRATION')
    iotests.log(event1, filters=[iotests.filter_qmp_event])
    if event1['data']['status'] in ('completed', 'failed'):
        iotests.log('Gracefully ending the `drive-mirror` job on source...')
        iotests.log(source_vm.qmp('block-job-cancel', device='mirror-job0'))
        break

while True:
    event2 = source_vm.event_wait('BLOCK_JOB_COMPLETED')
    iotests.log(event2, filters=[iotests.filter_qmp_event])
    if event2['event'] == 'BLOCK_JOB_COMPLETED':
        iotests.log('Stopping the NBD server on destination...')
        iotests.log(dest_vm.qmp('nbd-server-stop'))
        break