b6aed193e5
In many cases we just want an effect of qmp command and want to raise
on failure. Use vm.cmd() method which does exactly this.
The commit is generated by command
git grep -l '\.qmp(' | xargs ./scripts/python_qmp_updater.py
And then, fix self.assertRaises to expect ExecuteError exception in
tests/qemu-iotests/124
Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@yandex-team.ru>
Reviewed-by: Eric Blake <eblake@redhat.com>
Message-id: 20231006154125.1068348-16-vsementsov@yandex-team.ru
Signed-off-by: John Snow <jsnow@redhat.com>
117 lines
4.0 KiB
Python
Executable File
117 lines
4.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# group: rw
|
|
#
|
|
# Test nbd reconnect
|
|
#
|
|
# Copyright (c) 2019 Virtuozzo International GmbH.
|
|
#
|
|
# 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/>.
|
|
#
|
|
|
|
import time
|
|
import os
|
|
|
|
import iotests
|
|
from iotests import qemu_img_create, file_path, qemu_nbd_popen
|
|
|
|
disk_a, disk_b, nbd_sock = file_path('disk_a', 'disk_b', 'nbd-sock')
|
|
nbd_uri = 'nbd+unix:///?socket=' + nbd_sock
|
|
wait_limit = 3.0
|
|
wait_step = 0.2
|
|
|
|
|
|
class TestNbdReconnect(iotests.QMPTestCase):
|
|
def init_vm(self, disk_size):
|
|
qemu_img_create('-f', iotests.imgfmt, disk_a, str(disk_size))
|
|
qemu_img_create('-f', iotests.imgfmt, disk_b, str(disk_size))
|
|
self.vm = iotests.VM().add_drive(disk_a)
|
|
self.vm.launch()
|
|
self.vm.hmp_qemu_io('drive0', 'write 0 {}'.format(disk_size))
|
|
|
|
def tearDown(self):
|
|
self.vm.shutdown()
|
|
os.remove(disk_a)
|
|
os.remove(disk_b)
|
|
|
|
def start_job(self, job):
|
|
"""Stat job with nbd target and kill the server"""
|
|
assert job in ('blockdev-backup', 'blockdev-mirror')
|
|
with qemu_nbd_popen('-k', nbd_sock, '-f', iotests.imgfmt, disk_b):
|
|
self.vm.cmd('blockdev-add',
|
|
{'node-name': 'backup0',
|
|
'driver': 'raw',
|
|
'file': {'driver': 'nbd',
|
|
'server': {'type': 'unix',
|
|
'path': nbd_sock},
|
|
'reconnect-delay': 10}})
|
|
self.vm.cmd(job, device='drive0',
|
|
sync='full', target='backup0',
|
|
speed=(1 * 1024 * 1024))
|
|
|
|
# Wait for some progress
|
|
t = 0.0
|
|
while t < wait_limit:
|
|
jobs = self.vm.qmp('query-block-jobs')['return']
|
|
if jobs and jobs[0]['offset'] > 0:
|
|
break
|
|
time.sleep(wait_step)
|
|
t += wait_step
|
|
|
|
self.assertTrue(jobs and jobs[0]['offset'] > 0) # job started
|
|
|
|
jobs = self.vm.qmp('query-block-jobs')['return']
|
|
# Check that job is still in progress
|
|
self.assertTrue(jobs)
|
|
self.assertTrue(jobs[0]['offset'] < jobs[0]['len'])
|
|
|
|
self.vm.cmd('block-job-set-speed', device='drive0', speed=0)
|
|
|
|
# Emulate server down time for 1 second
|
|
time.sleep(1)
|
|
|
|
def test_backup(self):
|
|
size = 5 * 1024 * 1024
|
|
self.init_vm(size)
|
|
self.start_job('blockdev-backup')
|
|
|
|
with qemu_nbd_popen('-k', nbd_sock, '-f', iotests.imgfmt, disk_b):
|
|
e = self.vm.event_wait('BLOCK_JOB_COMPLETED')
|
|
self.assertEqual(e['data']['offset'], size)
|
|
self.vm.cmd('blockdev-del', node_name='backup0')
|
|
|
|
def cancel_job(self):
|
|
self.vm.cmd('block-job-cancel', device='drive0', force=True)
|
|
|
|
start_t = time.time()
|
|
self.vm.event_wait('BLOCK_JOB_CANCELLED')
|
|
delta_t = time.time() - start_t
|
|
self.assertTrue(delta_t < 5.0)
|
|
|
|
def test_mirror_cancel(self):
|
|
# Mirror speed limit doesn't work well enough, it seems that mirror
|
|
# will run many parallel requests anyway. MAX_IN_FLIGHT is 16 and
|
|
# MAX_IO_BYTES is 1M in mirror.c, so let's use 20M disk.
|
|
self.init_vm(20 * 1024 * 1024)
|
|
self.start_job('blockdev-mirror')
|
|
self.cancel_job()
|
|
|
|
def test_backup_cancel(self):
|
|
self.init_vm(5 * 1024 * 1024)
|
|
self.start_job('blockdev-backup')
|
|
self.cancel_job()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
iotests.main(supported_fmts=['qcow2'])
|