tests/qemu-iotests/194

   1 #!/usr/bin/env python3
   2 # group: rw migration quick
   3 #
   4 # Copyright (C) 2017 Red Hat, Inc.
   5 #
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 2 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 #
  19 # Creator/Owner: Stefan Hajnoczi <stefanha@redhat.com>
  20 #
  21 # Non-shared storage migration test using NBD server and drive-mirror
  22
  23 import iotests
  24
  25 iotests.script_initialize(supported_fmts=['qcow2', 'qed', 'raw'],
  26                           supported_platforms=['linux'])
  27
  28 with iotests.FilePath('source.img') as source_img_path, \
  29      iotests.FilePath('dest.img') as dest_img_path, \
  30      iotests.FilePath('migration.sock', 'nbd.sock', base_dir=iotests.sock_dir) \
  31         as (migration_sock_path, nbd_sock_path), \
  32      iotests.VM('source') as source_vm, \
  33      iotests.VM('dest') as dest_vm:
  34
  35     img_size = '1G'
  36     iotests.qemu_img_create('-f', iotests.imgfmt, source_img_path, img_size)
  37     iotests.qemu_img_create('-f', iotests.imgfmt, dest_img_path, img_size)
  38
  39     iotests.log('Launching VMs...')
  40     (source_vm.add_drive(source_img_path)
  41               .launch())
  42     (dest_vm.add_drive(dest_img_path)
  43             .add_incoming('unix:{0}'.format(migration_sock_path))
  44             .launch())
  45
  46     source_vm.qmp_log('block-dirty-bitmap-add', node='drive0', name='bitmap0')
  47
  48     iotests.log('Launching NBD server on destination...')
  49     iotests.log(dest_vm.qmp('nbd-server-start', addr={'type': 'unix', 'data': {'path': nbd_sock_path}}))
  50     iotests.log(dest_vm.qmp('nbd-server-add', device='drive0', writable=True))
  51
  52     iotests.log('Starting `drive-mirror` on source...')
  53     iotests.log(source_vm.qmp(
  54                   'drive-mirror',
  55                   device='drive0',
  56                   target='nbd+unix:///drive0?socket={0}'.format(nbd_sock_path),
  57                   sync='full',
  58                   format='raw', # always raw, the server handles the format
  59                   mode='existing',
  60                   job_id='mirror-job0'))
  61
  62     iotests.log('Waiting for `drive-mirror` to complete...')
  63     iotests.log(source_vm.event_wait('BLOCK_JOB_READY'),
  64                 filters=[iotests.filter_qmp_event])
  65
  66     iotests.log('Starting migration...')
  67     capabilities = [{'capability': 'events', 'state': True},
  68                     {'capability': 'dirty-bitmaps', 'state': True}]
  69     source_vm.qmp('migrate-set-capabilities', capabilities=capabilities)
  70     dest_vm.qmp('migrate-set-capabilities', capabilities=capabilities)
  71     iotests.log(source_vm.qmp('migrate', uri='unix:{0}'.format(migration_sock_path)))
  72
  73     source_vm.qmp_log('migrate-start-postcopy')
  74
  75     while True:
  76         event1 = source_vm.event_wait('MIGRATION')
  77         iotests.log(event1, filters=[iotests.filter_qmp_event])
  78         if event1['data']['status'] in ('completed', 'failed'):
  79             iotests.log('Gracefully ending the `drive-mirror` job on source...')
  80             iotests.log(source_vm.qmp('block-job-cancel', device='mirror-job0'))
  81             break
  82
  83     while True:
  84         event2 = source_vm.event_wait('BLOCK_JOB_COMPLETED')
  85         iotests.log(event2, filters=[iotests.filter_qmp_event])
  86         if event2['event'] == 'BLOCK_JOB_COMPLETED':
  87             iotests.log('Stopping the NBD server on destination...')
  88             iotests.log(dest_vm.qmp('nbd-server-stop'))
  89             break
  90
  91     iotests.log('Wait for migration completion on target...')
  92     migr_events = (('MIGRATION', {'data': {'status': 'completed'}}),
  93                    ('MIGRATION', {'data': {'status': 'failed'}}))
  94     event = dest_vm.events_wait(migr_events)
  95     iotests.log(event, filters=[iotests.filter_qmp_event])
  96
  97     iotests.log('Check bitmaps on source:')
  98     iotests.log(source_vm.qmp('query-block')['return'][0]['inserted']['dirty-bitmaps'])
  99
 100     iotests.log('Check bitmaps on target:')
 101     iotests.log(dest_vm.qmp('query-block')['return'][0]['inserted']['dirty-bitmaps'])