819cec0114
Source disk is created and filled with test data before each test case. Instead initialize it once for the whole unit. Test disk filling patterns are merged into one pattern. Also TestSetSpeed used different image_len for source and target (by mistake) - this is automatically fixed here. Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> Message-id: 1470748523-13856-1-git-send-email-vsementsov@virtuozzo.com Reviewed-by: Pavel Butsykin <pbutsykin@virtuozzo.com> Reviewed-by: Max Reitz <mreitz@redhat.com> Signed-off-by: Max Reitz <mreitz@redhat.com>
555 lines
20 KiB
Python
Executable File
555 lines
20 KiB
Python
Executable File
#!/usr/bin/env python
|
|
#
|
|
# Tests for drive-backup and blockdev-backup
|
|
#
|
|
# Copyright (C) 2013, 2014 Red Hat, Inc.
|
|
#
|
|
# Based on 041.
|
|
#
|
|
# 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, qemu_io
|
|
|
|
test_img = os.path.join(iotests.test_dir, 'test.img')
|
|
target_img = os.path.join(iotests.test_dir, 'target.img')
|
|
blockdev_target_img = os.path.join(iotests.test_dir, 'blockdev-target.img')
|
|
|
|
image_len = 64 * 1024 * 1024 # MB
|
|
|
|
def setUpModule():
|
|
qemu_img('create', '-f', iotests.imgfmt, test_img, str(image_len))
|
|
qemu_io('-f', iotests.imgfmt, '-c', 'write -P0x11 0 64k', test_img)
|
|
qemu_io('-f', iotests.imgfmt, '-c', 'write -P0x00 64k 128k', test_img)
|
|
qemu_io('-f', iotests.imgfmt, '-c', 'write -P0x22 162k 32k', test_img)
|
|
qemu_io('-f', iotests.imgfmt, '-c', 'write -P0xd5 1M 32k', test_img)
|
|
qemu_io('-f', iotests.imgfmt, '-c', 'write -P0xdc 32M 124k', test_img)
|
|
qemu_io('-f', iotests.imgfmt, '-c', 'write -P0x33 67043328 64k', test_img)
|
|
|
|
def tearDownModule():
|
|
os.remove(test_img)
|
|
|
|
|
|
class TestSingleDrive(iotests.QMPTestCase):
|
|
def setUp(self):
|
|
qemu_img('create', '-f', iotests.imgfmt, blockdev_target_img, str(image_len))
|
|
|
|
self.vm = iotests.VM().add_drive(test_img).add_drive(blockdev_target_img)
|
|
if iotests.qemu_default_machine == 'pc':
|
|
self.vm.add_drive(None, 'media=cdrom', 'ide')
|
|
self.vm.launch()
|
|
|
|
def tearDown(self):
|
|
self.vm.shutdown()
|
|
os.remove(blockdev_target_img)
|
|
try:
|
|
os.remove(target_img)
|
|
except OSError:
|
|
pass
|
|
|
|
def do_test_cancel(self, cmd, target):
|
|
self.assert_no_active_block_jobs()
|
|
|
|
result = self.vm.qmp(cmd, device='drive0', target=target, sync='full')
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
event = self.cancel_and_wait()
|
|
self.assert_qmp(event, 'data/type', 'backup')
|
|
|
|
def test_cancel_drive_backup(self):
|
|
self.do_test_cancel('drive-backup', target_img)
|
|
|
|
def test_cancel_blockdev_backup(self):
|
|
self.do_test_cancel('blockdev-backup', 'drive1')
|
|
|
|
def do_test_pause(self, cmd, target, image):
|
|
self.assert_no_active_block_jobs()
|
|
|
|
self.vm.pause_drive('drive0')
|
|
result = self.vm.qmp(cmd, device='drive0',
|
|
target=target, sync='full')
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
result = self.vm.qmp('block-job-pause', device='drive0')
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
self.vm.resume_drive('drive0')
|
|
time.sleep(1)
|
|
result = self.vm.qmp('query-block-jobs')
|
|
offset = self.dictpath(result, 'return[0]/offset')
|
|
|
|
time.sleep(1)
|
|
result = self.vm.qmp('query-block-jobs')
|
|
self.assert_qmp(result, 'return[0]/offset', offset)
|
|
|
|
result = self.vm.qmp('block-job-resume', device='drive0')
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
self.wait_until_completed()
|
|
|
|
self.vm.shutdown()
|
|
self.assertTrue(iotests.compare_images(test_img, image),
|
|
'target image does not match source after backup')
|
|
|
|
def test_pause_drive_backup(self):
|
|
self.do_test_pause('drive-backup', target_img, target_img)
|
|
|
|
def test_pause_blockdev_backup(self):
|
|
self.do_test_pause('blockdev-backup', 'drive1', blockdev_target_img)
|
|
|
|
def test_medium_not_found(self):
|
|
if iotests.qemu_default_machine != 'pc':
|
|
return
|
|
|
|
result = self.vm.qmp('drive-backup', device='drive2', # CD-ROM
|
|
target=target_img, sync='full')
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
def test_medium_not_found_blockdev_backup(self):
|
|
if iotests.qemu_default_machine != 'pc':
|
|
return
|
|
|
|
result = self.vm.qmp('blockdev-backup', device='drive2', # CD-ROM
|
|
target='drive1', sync='full')
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
def test_image_not_found(self):
|
|
result = self.vm.qmp('drive-backup', device='drive0',
|
|
target=target_img, sync='full', mode='existing')
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
def test_invalid_format(self):
|
|
result = self.vm.qmp('drive-backup', device='drive0',
|
|
target=target_img, sync='full',
|
|
format='spaghetti-noodles')
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
def do_test_device_not_found(self, cmd, **args):
|
|
result = self.vm.qmp(cmd, **args)
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
def test_device_not_found(self):
|
|
self.do_test_device_not_found('drive-backup', device='nonexistent',
|
|
target=target_img, sync='full')
|
|
|
|
self.do_test_device_not_found('blockdev-backup', device='nonexistent',
|
|
target='drive0', sync='full')
|
|
|
|
self.do_test_device_not_found('blockdev-backup', device='drive0',
|
|
target='nonexistent', sync='full')
|
|
|
|
self.do_test_device_not_found('blockdev-backup', device='nonexistent',
|
|
target='nonexistent', sync='full')
|
|
|
|
def test_target_is_source(self):
|
|
result = self.vm.qmp('blockdev-backup', device='drive0',
|
|
target='drive0', sync='full')
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
class TestSetSpeed(iotests.QMPTestCase):
|
|
def setUp(self):
|
|
qemu_img('create', '-f', iotests.imgfmt, blockdev_target_img, str(image_len))
|
|
|
|
self.vm = iotests.VM().add_drive(test_img).add_drive(blockdev_target_img)
|
|
self.vm.launch()
|
|
|
|
def tearDown(self):
|
|
self.vm.shutdown()
|
|
os.remove(blockdev_target_img)
|
|
try:
|
|
os.remove(target_img)
|
|
except OSError:
|
|
pass
|
|
|
|
def do_test_set_speed(self, cmd, target):
|
|
self.assert_no_active_block_jobs()
|
|
|
|
self.vm.pause_drive('drive0')
|
|
result = self.vm.qmp(cmd, device='drive0', target=target, sync='full')
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
# Default speed is 0
|
|
result = self.vm.qmp('query-block-jobs')
|
|
self.assert_qmp(result, 'return[0]/device', 'drive0')
|
|
self.assert_qmp(result, 'return[0]/speed', 0)
|
|
|
|
result = self.vm.qmp('block-job-set-speed', device='drive0', speed=8 * 1024 * 1024)
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
# Ensure the speed we set was accepted
|
|
result = self.vm.qmp('query-block-jobs')
|
|
self.assert_qmp(result, 'return[0]/device', 'drive0')
|
|
self.assert_qmp(result, 'return[0]/speed', 8 * 1024 * 1024)
|
|
|
|
event = self.cancel_and_wait(resume=True)
|
|
self.assert_qmp(event, 'data/type', 'backup')
|
|
|
|
# Check setting speed option works
|
|
self.vm.pause_drive('drive0')
|
|
result = self.vm.qmp(cmd, device='drive0',
|
|
target=target, sync='full', speed=4*1024*1024)
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
result = self.vm.qmp('query-block-jobs')
|
|
self.assert_qmp(result, 'return[0]/device', 'drive0')
|
|
self.assert_qmp(result, 'return[0]/speed', 4 * 1024 * 1024)
|
|
|
|
event = self.cancel_and_wait(resume=True)
|
|
self.assert_qmp(event, 'data/type', 'backup')
|
|
|
|
def test_set_speed_drive_backup(self):
|
|
self.do_test_set_speed('drive-backup', target_img)
|
|
|
|
def test_set_speed_blockdev_backup(self):
|
|
self.do_test_set_speed('blockdev-backup', 'drive1')
|
|
|
|
def do_test_set_speed_invalid(self, cmd, target):
|
|
self.assert_no_active_block_jobs()
|
|
|
|
result = self.vm.qmp(cmd, device='drive0',
|
|
target=target, sync='full', speed=-1)
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
self.assert_no_active_block_jobs()
|
|
|
|
self.vm.pause_drive('drive0')
|
|
result = self.vm.qmp(cmd, device='drive0',
|
|
target=target, sync='full')
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
result = self.vm.qmp('block-job-set-speed', device='drive0', speed=-1)
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
event = self.cancel_and_wait(resume=True)
|
|
self.assert_qmp(event, 'data/type', 'backup')
|
|
|
|
def test_set_speed_invalid_drive_backup(self):
|
|
self.do_test_set_speed_invalid('drive-backup', target_img)
|
|
|
|
def test_set_speed_invalid_blockdev_backup(self):
|
|
self.do_test_set_speed_invalid('blockdev-backup', 'drive1')
|
|
|
|
class TestSingleTransaction(iotests.QMPTestCase):
|
|
def setUp(self):
|
|
qemu_img('create', '-f', iotests.imgfmt, blockdev_target_img, str(image_len))
|
|
|
|
self.vm = iotests.VM().add_drive(test_img).add_drive(blockdev_target_img)
|
|
if iotests.qemu_default_machine == 'pc':
|
|
self.vm.add_drive(None, 'media=cdrom', 'ide')
|
|
self.vm.launch()
|
|
|
|
def tearDown(self):
|
|
self.vm.shutdown()
|
|
os.remove(blockdev_target_img)
|
|
try:
|
|
os.remove(target_img)
|
|
except OSError:
|
|
pass
|
|
|
|
def do_test_cancel(self, cmd, target):
|
|
self.assert_no_active_block_jobs()
|
|
|
|
result = self.vm.qmp('transaction', actions=[{
|
|
'type': cmd,
|
|
'data': { 'device': 'drive0',
|
|
'target': target,
|
|
'sync': 'full' },
|
|
}
|
|
])
|
|
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
event = self.cancel_and_wait()
|
|
self.assert_qmp(event, 'data/type', 'backup')
|
|
|
|
def test_cancel_drive_backup(self):
|
|
self.do_test_cancel('drive-backup', target_img)
|
|
|
|
def test_cancel_blockdev_backup(self):
|
|
self.do_test_cancel('blockdev-backup', 'drive1')
|
|
|
|
def do_test_pause(self, cmd, target, image):
|
|
self.assert_no_active_block_jobs()
|
|
|
|
self.vm.pause_drive('drive0')
|
|
result = self.vm.qmp('transaction', actions=[{
|
|
'type': cmd,
|
|
'data': { 'device': 'drive0',
|
|
'target': target,
|
|
'sync': 'full' },
|
|
}
|
|
])
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
result = self.vm.qmp('block-job-pause', device='drive0')
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
self.vm.resume_drive('drive0')
|
|
time.sleep(1)
|
|
result = self.vm.qmp('query-block-jobs')
|
|
offset = self.dictpath(result, 'return[0]/offset')
|
|
|
|
time.sleep(1)
|
|
result = self.vm.qmp('query-block-jobs')
|
|
self.assert_qmp(result, 'return[0]/offset', offset)
|
|
|
|
result = self.vm.qmp('block-job-resume', device='drive0')
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
self.wait_until_completed()
|
|
|
|
self.vm.shutdown()
|
|
self.assertTrue(iotests.compare_images(test_img, image),
|
|
'target image does not match source after backup')
|
|
|
|
def test_pause_drive_backup(self):
|
|
self.do_test_pause('drive-backup', target_img, target_img)
|
|
|
|
def test_pause_blockdev_backup(self):
|
|
self.do_test_pause('blockdev-backup', 'drive1', blockdev_target_img)
|
|
|
|
def do_test_medium_not_found(self, cmd, target):
|
|
if iotests.qemu_default_machine != 'pc':
|
|
return
|
|
|
|
result = self.vm.qmp('transaction', actions=[{
|
|
'type': cmd,
|
|
'data': { 'device': 'drive2', # CD-ROM
|
|
'target': target,
|
|
'sync': 'full' },
|
|
}
|
|
])
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
def test_medium_not_found_drive_backup(self):
|
|
self.do_test_medium_not_found('drive-backup', target_img)
|
|
|
|
def test_medium_not_found_blockdev_backup(self):
|
|
self.do_test_medium_not_found('blockdev-backup', 'drive1')
|
|
|
|
def test_image_not_found(self):
|
|
result = self.vm.qmp('transaction', actions=[{
|
|
'type': 'drive-backup',
|
|
'data': { 'device': 'drive0',
|
|
'mode': 'existing',
|
|
'target': target_img,
|
|
'sync': 'full' },
|
|
}
|
|
])
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
def test_device_not_found(self):
|
|
result = self.vm.qmp('transaction', actions=[{
|
|
'type': 'drive-backup',
|
|
'data': { 'device': 'nonexistent',
|
|
'mode': 'existing',
|
|
'target': target_img,
|
|
'sync': 'full' },
|
|
}
|
|
])
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
result = self.vm.qmp('transaction', actions=[{
|
|
'type': 'blockdev-backup',
|
|
'data': { 'device': 'nonexistent',
|
|
'target': 'drive1',
|
|
'sync': 'full' },
|
|
}
|
|
])
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
result = self.vm.qmp('transaction', actions=[{
|
|
'type': 'blockdev-backup',
|
|
'data': { 'device': 'drive0',
|
|
'target': 'nonexistent',
|
|
'sync': 'full' },
|
|
}
|
|
])
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
result = self.vm.qmp('transaction', actions=[{
|
|
'type': 'blockdev-backup',
|
|
'data': { 'device': 'nonexistent',
|
|
'target': 'nonexistent',
|
|
'sync': 'full' },
|
|
}
|
|
])
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
def test_target_is_source(self):
|
|
result = self.vm.qmp('transaction', actions=[{
|
|
'type': 'blockdev-backup',
|
|
'data': { 'device': 'drive0',
|
|
'target': 'drive0',
|
|
'sync': 'full' },
|
|
}
|
|
])
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
|
|
def test_abort(self):
|
|
result = self.vm.qmp('transaction', actions=[{
|
|
'type': 'drive-backup',
|
|
'data': { 'device': 'nonexistent',
|
|
'mode': 'existing',
|
|
'target': target_img,
|
|
'sync': 'full' },
|
|
}, {
|
|
'type': 'Abort',
|
|
'data': {},
|
|
}
|
|
])
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
self.assert_no_active_block_jobs()
|
|
|
|
result = self.vm.qmp('transaction', actions=[{
|
|
'type': 'blockdev-backup',
|
|
'data': { 'device': 'nonexistent',
|
|
'target': 'drive1',
|
|
'sync': 'full' },
|
|
}, {
|
|
'type': 'Abort',
|
|
'data': {},
|
|
}
|
|
])
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
self.assert_no_active_block_jobs()
|
|
|
|
result = self.vm.qmp('transaction', actions=[{
|
|
'type': 'blockdev-backup',
|
|
'data': { 'device': 'drive0',
|
|
'target': 'nonexistent',
|
|
'sync': 'full' },
|
|
}, {
|
|
'type': 'Abort',
|
|
'data': {},
|
|
}
|
|
])
|
|
self.assert_qmp(result, 'error/class', 'GenericError')
|
|
self.assert_no_active_block_jobs()
|
|
|
|
|
|
class TestDriveCompression(iotests.QMPTestCase):
|
|
image_len = 64 * 1024 * 1024 # MB
|
|
fmt_supports_compression = [{'type': 'qcow2', 'args': ()},
|
|
{'type': 'vmdk', 'args': ('-o', 'subformat=streamOptimized')}]
|
|
|
|
def tearDown(self):
|
|
self.vm.shutdown()
|
|
os.remove(blockdev_target_img)
|
|
try:
|
|
os.remove(target_img)
|
|
except OSError:
|
|
pass
|
|
|
|
def do_prepare_drives(self, fmt, args):
|
|
self.vm = iotests.VM().add_drive(test_img)
|
|
|
|
qemu_img('create', '-f', fmt, blockdev_target_img,
|
|
str(TestDriveCompression.image_len), *args)
|
|
self.vm.add_drive(blockdev_target_img, format=fmt)
|
|
|
|
self.vm.launch()
|
|
|
|
def do_test_compress_complete(self, cmd, format, **args):
|
|
self.do_prepare_drives(format['type'], format['args'])
|
|
|
|
self.assert_no_active_block_jobs()
|
|
|
|
result = self.vm.qmp(cmd, device='drive0', sync='full', compress=True, **args)
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
self.wait_until_completed()
|
|
|
|
self.vm.shutdown()
|
|
self.assertTrue(iotests.compare_images(test_img, blockdev_target_img,
|
|
iotests.imgfmt, format['type']),
|
|
'target image does not match source after backup')
|
|
|
|
def test_complete_compress_drive_backup(self):
|
|
for format in TestDriveCompression.fmt_supports_compression:
|
|
self.do_test_compress_complete('drive-backup', format,
|
|
target=blockdev_target_img, mode='existing')
|
|
|
|
def test_complete_compress_blockdev_backup(self):
|
|
for format in TestDriveCompression.fmt_supports_compression:
|
|
self.do_test_compress_complete('blockdev-backup', format, target='drive1')
|
|
|
|
def do_test_compress_cancel(self, cmd, format, **args):
|
|
self.do_prepare_drives(format['type'], format['args'])
|
|
|
|
self.assert_no_active_block_jobs()
|
|
|
|
result = self.vm.qmp(cmd, device='drive0', sync='full', compress=True, **args)
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
event = self.cancel_and_wait()
|
|
self.assert_qmp(event, 'data/type', 'backup')
|
|
|
|
self.vm.shutdown()
|
|
|
|
def test_compress_cancel_drive_backup(self):
|
|
for format in TestDriveCompression.fmt_supports_compression:
|
|
self.do_test_compress_cancel('drive-backup', format,
|
|
target=blockdev_target_img, mode='existing')
|
|
|
|
def test_compress_cancel_blockdev_backup(self):
|
|
for format in TestDriveCompression.fmt_supports_compression:
|
|
self.do_test_compress_cancel('blockdev-backup', format, target='drive1')
|
|
|
|
def do_test_compress_pause(self, cmd, format, **args):
|
|
self.do_prepare_drives(format['type'], format['args'])
|
|
|
|
self.assert_no_active_block_jobs()
|
|
|
|
self.vm.pause_drive('drive0')
|
|
result = self.vm.qmp(cmd, device='drive0', sync='full', compress=True, **args)
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
result = self.vm.qmp('block-job-pause', device='drive0')
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
self.vm.resume_drive('drive0')
|
|
time.sleep(1)
|
|
result = self.vm.qmp('query-block-jobs')
|
|
offset = self.dictpath(result, 'return[0]/offset')
|
|
|
|
time.sleep(1)
|
|
result = self.vm.qmp('query-block-jobs')
|
|
self.assert_qmp(result, 'return[0]/offset', offset)
|
|
|
|
result = self.vm.qmp('block-job-resume', device='drive0')
|
|
self.assert_qmp(result, 'return', {})
|
|
|
|
self.wait_until_completed()
|
|
|
|
self.vm.shutdown()
|
|
self.assertTrue(iotests.compare_images(test_img, blockdev_target_img,
|
|
iotests.imgfmt, format['type']),
|
|
'target image does not match source after backup')
|
|
|
|
def test_compress_pause_drive_backup(self):
|
|
for format in TestDriveCompression.fmt_supports_compression:
|
|
self.do_test_compress_pause('drive-backup', format,
|
|
target=blockdev_target_img, mode='existing')
|
|
|
|
def test_compress_pause_blockdev_backup(self):
|
|
for format in TestDriveCompression.fmt_supports_compression:
|
|
self.do_test_compress_pause('blockdev-backup', format, target='drive1')
|
|
|
|
if __name__ == '__main__':
|
|
iotests.main(supported_fmts=['raw', 'qcow2'])
|