tests/qemu-iotests/234

   1 #!/usr/bin/env python
   2 #
   3 # Copyright (C) 2018 Red Hat, Inc.
   4 #
   5 # This program is free software; you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation; either version 2 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 #
  18 # Creator/Owner: Kevin Wolf <kwolf@redhat.com>
  19 #
  20 # Check that block node activation and inactivation works with a block graph
  21 # that is built with individually created nodes
  22
  23 import iotests
  24 import os
  25
  26 iotests.verify_image_format(supported_fmts=['qcow2'])
  27 iotests.verify_platform(['linux'])
  28
  29 def enable_migration_events(vm, name):
  30     iotests.log('Enabling migration QMP events on %s...' % name)
  31     iotests.log(vm.qmp('migrate-set-capabilities', capabilities=[
  32         {
  33             'capability': 'events',
  34             'state': True
  35         }
  36     ]))
  37
  38 def wait_migration(vm):
  39     while True:
  40         event = vm.event_wait('MIGRATION')
  41         iotests.log(event, filters=[iotests.filter_qmp_event])
  42         if event['data']['status'] == 'completed':
  43             break
  44
  45 with iotests.FilePath('img') as img_path, \
  46      iotests.FilePath('backing') as backing_path, \
  47      iotests.FilePath('mig_fifo_a') as fifo_a, \
  48      iotests.FilePath('mig_fifo_b') as fifo_b, \
  49      iotests.VM(path_suffix='a') as vm_a, \
  50      iotests.VM(path_suffix='b') as vm_b:
  51
  52     iotests.qemu_img_pipe('create', '-f', iotests.imgfmt, backing_path, '64M')
  53     iotests.qemu_img_pipe('create', '-f', iotests.imgfmt, img_path, '64M')
  54
  55     os.mkfifo(fifo_a)
  56     os.mkfifo(fifo_b)
  57
  58     iotests.log('Launching source VM...')
  59     (vm_a.add_blockdev('file,filename=%s,node-name=drive0-file' % (img_path))
  60          .add_blockdev('%s,file=drive0-file,node-name=drive0' % (iotests.imgfmt))
  61          .add_blockdev('file,filename=%s,node-name=drive0-backing-file' % (backing_path))
  62          .add_blockdev('%s,file=drive0-backing-file,node-name=drive0-backing' % (iotests.imgfmt))
  63          .launch())
  64
  65     enable_migration_events(vm_a, 'A')
  66
  67     iotests.log('Launching destination VM...')
  68     (vm_b.add_blockdev('file,filename=%s,node-name=drive0-file' % (img_path))
  69          .add_blockdev('%s,file=drive0-file,node-name=drive0' % (iotests.imgfmt))
  70          .add_blockdev('file,filename=%s,node-name=drive0-backing-file' % (backing_path))
  71          .add_blockdev('%s,file=drive0-backing-file,node-name=drive0-backing' % (iotests.imgfmt))
  72          .add_incoming("exec: cat '%s'" % (fifo_a))
  73          .launch())
  74
  75     enable_migration_events(vm_b, 'B')
  76
  77     # Add a child node that was created after the parent node. The reverse case
  78     # is covered by the -blockdev options above.
  79     iotests.log(vm_a.qmp('blockdev-snapshot', node='drive0-backing',
  80                          overlay='drive0'))
  81     iotests.log(vm_b.qmp('blockdev-snapshot', node='drive0-backing',
  82                          overlay='drive0'))
  83
  84     iotests.log('Starting migration to B...')
  85     iotests.log(vm_a.qmp('migrate', uri='exec:cat >%s' % (fifo_a)))
  86     with iotests.Timeout(3, 'Migration does not complete'):
  87         # Wait for the source first (which includes setup=setup)
  88         wait_migration(vm_a)
  89         # Wait for the destination second (which does not)
  90         wait_migration(vm_b)
  91
  92     iotests.log(vm_a.qmp('query-migrate')['return']['status'])
  93     iotests.log(vm_b.qmp('query-migrate')['return']['status'])
  94
  95     iotests.log(vm_a.qmp('query-status'))
  96     iotests.log(vm_b.qmp('query-status'))
  97
  98     iotests.log('Add a second parent to drive0-file...')
  99     iotests.log(vm_b.qmp('blockdev-add', driver='raw', file='drive0-file',
 100                          node_name='drive0-raw'))
 101
 102     iotests.log('Restart A with -incoming and second parent...')
 103     vm_a.shutdown()
 104     (vm_a.add_blockdev('raw,file=drive0-file,node-name=drive0-raw')
 105          .add_incoming("exec: cat '%s'" % (fifo_b))
 106          .launch())
 107
 108     enable_migration_events(vm_a, 'A')
 109
 110     iotests.log(vm_a.qmp('blockdev-snapshot', node='drive0-backing',
 111                          overlay='drive0'))
 112
 113     iotests.log('Starting migration back to A...')
 114     iotests.log(vm_b.qmp('migrate', uri='exec:cat >%s' % (fifo_b)))
 115     with iotests.Timeout(3, 'Migration does not complete'):
 116         # Wait for the source first (which includes setup=setup)
 117         wait_migration(vm_b)
 118         # Wait for the destination second (which does not)
 119         wait_migration(vm_a)
 120
 121     iotests.log(vm_a.qmp('query-migrate')['return']['status'])
 122     iotests.log(vm_b.qmp('query-migrate')['return']['status'])
 123
 124     iotests.log(vm_a.qmp('query-status'))
 125     iotests.log(vm_b.qmp('query-status'))