144c7ca5eSPaolo Bonzini#!/usr/bin/env python 244c7ca5eSPaolo Bonzini# 344c7ca5eSPaolo Bonzini# Tests for image mirroring. 444c7ca5eSPaolo Bonzini# 544c7ca5eSPaolo Bonzini# Copyright (C) 2012 Red Hat, Inc. 644c7ca5eSPaolo Bonzini# 744c7ca5eSPaolo Bonzini# This program is free software; you can redistribute it and/or modify 844c7ca5eSPaolo Bonzini# it under the terms of the GNU General Public License as published by 944c7ca5eSPaolo Bonzini# the Free Software Foundation; either version 2 of the License, or 1044c7ca5eSPaolo Bonzini# (at your option) any later version. 1144c7ca5eSPaolo Bonzini# 1244c7ca5eSPaolo Bonzini# This program is distributed in the hope that it will be useful, 1344c7ca5eSPaolo Bonzini# but WITHOUT ANY WARRANTY; without even the implied warranty of 1444c7ca5eSPaolo Bonzini# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 1544c7ca5eSPaolo Bonzini# GNU General Public License for more details. 1644c7ca5eSPaolo Bonzini# 1744c7ca5eSPaolo Bonzini# You should have received a copy of the GNU General Public License 1844c7ca5eSPaolo Bonzini# along with this program. If not, see <http://www.gnu.org/licenses/>. 1944c7ca5eSPaolo Bonzini# 2044c7ca5eSPaolo Bonzini 2144c7ca5eSPaolo Bonziniimport time 2244c7ca5eSPaolo Bonziniimport os 2344c7ca5eSPaolo Bonziniimport iotests 2444c7ca5eSPaolo Bonzinifrom iotests import qemu_img, qemu_io 2544c7ca5eSPaolo Bonzini 2644c7ca5eSPaolo Bonzinibacking_img = os.path.join(iotests.test_dir, 'backing.img') 2744c7ca5eSPaolo Bonzinitarget_backing_img = os.path.join(iotests.test_dir, 'target-backing.img') 2844c7ca5eSPaolo Bonzinitest_img = os.path.join(iotests.test_dir, 'test.img') 2944c7ca5eSPaolo Bonzinitarget_img = os.path.join(iotests.test_dir, 'target.img') 3044c7ca5eSPaolo Bonzini 3144c7ca5eSPaolo Bonziniclass ImageMirroringTestCase(iotests.QMPTestCase): 3244c7ca5eSPaolo Bonzini '''Abstract base class for image mirroring test cases''' 3344c7ca5eSPaolo Bonzini 342575fe16SStefan Hajnoczi def wait_ready(self, drive='drive0'): 352575fe16SStefan Hajnoczi '''Wait until a block job BLOCK_JOB_READY event''' 3644c7ca5eSPaolo Bonzini ready = False 3744c7ca5eSPaolo Bonzini while not ready: 3844c7ca5eSPaolo Bonzini for event in self.vm.get_qmp_events(wait=True): 3944c7ca5eSPaolo Bonzini if event['event'] == 'BLOCK_JOB_READY': 4044c7ca5eSPaolo Bonzini self.assert_qmp(event, 'data/type', 'mirror') 4144c7ca5eSPaolo Bonzini self.assert_qmp(event, 'data/device', drive) 4244c7ca5eSPaolo Bonzini ready = True 4344c7ca5eSPaolo Bonzini 442575fe16SStefan Hajnoczi def wait_ready_and_cancel(self, drive='drive0'): 452575fe16SStefan Hajnoczi self.wait_ready(drive) 462575fe16SStefan Hajnoczi event = self.cancel_and_wait() 4744c7ca5eSPaolo Bonzini self.assertEquals(event['event'], 'BLOCK_JOB_COMPLETED') 482575fe16SStefan Hajnoczi self.assert_qmp(event, 'data/type', 'mirror') 4944c7ca5eSPaolo Bonzini self.assert_qmp(event, 'data/offset', self.image_len) 5044c7ca5eSPaolo Bonzini self.assert_qmp(event, 'data/len', self.image_len) 5144c7ca5eSPaolo Bonzini 5244c7ca5eSPaolo Bonzini def complete_and_wait(self, drive='drive0', wait_ready=True): 5344c7ca5eSPaolo Bonzini '''Complete a block job and wait for it to finish''' 5444c7ca5eSPaolo Bonzini if wait_ready: 552575fe16SStefan Hajnoczi self.wait_ready() 5644c7ca5eSPaolo Bonzini 5744c7ca5eSPaolo Bonzini result = self.vm.qmp('block-job-complete', device=drive) 5844c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return', {}) 5944c7ca5eSPaolo Bonzini 600dbe8a1bSStefan Hajnoczi event = self.wait_until_completed() 6144c7ca5eSPaolo Bonzini self.assert_qmp(event, 'data/type', 'mirror') 6244c7ca5eSPaolo Bonzini 6344c7ca5eSPaolo Bonziniclass TestSingleDrive(ImageMirroringTestCase): 6444c7ca5eSPaolo Bonzini image_len = 1 * 1024 * 1024 # MB 6544c7ca5eSPaolo Bonzini 6644c7ca5eSPaolo Bonzini def setUp(self): 672499a096SStefan Hajnoczi iotests.create_image(backing_img, TestSingleDrive.image_len) 6844c7ca5eSPaolo Bonzini qemu_img('create', '-f', iotests.imgfmt, '-o', 'backing_file=%s' % backing_img, test_img) 6944c7ca5eSPaolo Bonzini self.vm = iotests.VM().add_drive(test_img) 7044c7ca5eSPaolo Bonzini self.vm.launch() 7144c7ca5eSPaolo Bonzini 7244c7ca5eSPaolo Bonzini def tearDown(self): 7344c7ca5eSPaolo Bonzini self.vm.shutdown() 7444c7ca5eSPaolo Bonzini os.remove(test_img) 7544c7ca5eSPaolo Bonzini os.remove(backing_img) 7644c7ca5eSPaolo Bonzini try: 7744c7ca5eSPaolo Bonzini os.remove(target_img) 7844c7ca5eSPaolo Bonzini except OSError: 7944c7ca5eSPaolo Bonzini pass 8044c7ca5eSPaolo Bonzini 8144c7ca5eSPaolo Bonzini def test_complete(self): 82ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 8344c7ca5eSPaolo Bonzini 8444c7ca5eSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 8544c7ca5eSPaolo Bonzini target=target_img) 8644c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return', {}) 8744c7ca5eSPaolo Bonzini 8844c7ca5eSPaolo Bonzini self.complete_and_wait() 8944c7ca5eSPaolo Bonzini result = self.vm.qmp('query-block') 9044c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return[0]/inserted/file', target_img) 9144c7ca5eSPaolo Bonzini self.vm.shutdown() 923a3918c3SStefan Hajnoczi self.assertTrue(iotests.compare_images(test_img, target_img), 9344c7ca5eSPaolo Bonzini 'target image does not match source after mirroring') 9444c7ca5eSPaolo Bonzini 9544c7ca5eSPaolo Bonzini def test_cancel(self): 96ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 9744c7ca5eSPaolo Bonzini 9844c7ca5eSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 9944c7ca5eSPaolo Bonzini target=target_img) 10044c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return', {}) 10144c7ca5eSPaolo Bonzini 1022575fe16SStefan Hajnoczi self.cancel_and_wait(force=True) 10344c7ca5eSPaolo Bonzini result = self.vm.qmp('query-block') 10444c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return[0]/inserted/file', test_img) 10544c7ca5eSPaolo Bonzini self.vm.shutdown() 10644c7ca5eSPaolo Bonzini 10744c7ca5eSPaolo Bonzini def test_cancel_after_ready(self): 108ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 10944c7ca5eSPaolo Bonzini 11044c7ca5eSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 11144c7ca5eSPaolo Bonzini target=target_img) 11244c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return', {}) 11344c7ca5eSPaolo Bonzini 1142575fe16SStefan Hajnoczi self.wait_ready_and_cancel() 11544c7ca5eSPaolo Bonzini result = self.vm.qmp('query-block') 11644c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return[0]/inserted/file', test_img) 11744c7ca5eSPaolo Bonzini self.vm.shutdown() 1183a3918c3SStefan Hajnoczi self.assertTrue(iotests.compare_images(test_img, target_img), 11944c7ca5eSPaolo Bonzini 'target image does not match source after mirroring') 12044c7ca5eSPaolo Bonzini 12144c7ca5eSPaolo Bonzini def test_pause(self): 122ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 12344c7ca5eSPaolo Bonzini 12444c7ca5eSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 12544c7ca5eSPaolo Bonzini target=target_img) 12644c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return', {}) 12744c7ca5eSPaolo Bonzini 12844c7ca5eSPaolo Bonzini result = self.vm.qmp('block-job-pause', device='drive0') 12944c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return', {}) 13044c7ca5eSPaolo Bonzini 13144c7ca5eSPaolo Bonzini time.sleep(1) 13244c7ca5eSPaolo Bonzini result = self.vm.qmp('query-block-jobs') 13344c7ca5eSPaolo Bonzini offset = self.dictpath(result, 'return[0]/offset') 13444c7ca5eSPaolo Bonzini 13544c7ca5eSPaolo Bonzini time.sleep(1) 13644c7ca5eSPaolo Bonzini result = self.vm.qmp('query-block-jobs') 13744c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return[0]/offset', offset) 13844c7ca5eSPaolo Bonzini 13944c7ca5eSPaolo Bonzini result = self.vm.qmp('block-job-resume', device='drive0') 14044c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return', {}) 14144c7ca5eSPaolo Bonzini 14244c7ca5eSPaolo Bonzini self.complete_and_wait() 14344c7ca5eSPaolo Bonzini self.vm.shutdown() 1443a3918c3SStefan Hajnoczi self.assertTrue(iotests.compare_images(test_img, target_img), 14544c7ca5eSPaolo Bonzini 'target image does not match source after mirroring') 14644c7ca5eSPaolo Bonzini 14708e4ed6cSPaolo Bonzini def test_small_buffer(self): 148ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 14908e4ed6cSPaolo Bonzini 15008e4ed6cSPaolo Bonzini # A small buffer is rounded up automatically 15108e4ed6cSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 15208e4ed6cSPaolo Bonzini buf_size=4096, target=target_img) 15308e4ed6cSPaolo Bonzini self.assert_qmp(result, 'return', {}) 15408e4ed6cSPaolo Bonzini 15508e4ed6cSPaolo Bonzini self.complete_and_wait() 15608e4ed6cSPaolo Bonzini result = self.vm.qmp('query-block') 15708e4ed6cSPaolo Bonzini self.assert_qmp(result, 'return[0]/inserted/file', target_img) 15808e4ed6cSPaolo Bonzini self.vm.shutdown() 1593a3918c3SStefan Hajnoczi self.assertTrue(iotests.compare_images(test_img, target_img), 16008e4ed6cSPaolo Bonzini 'target image does not match source after mirroring') 16108e4ed6cSPaolo Bonzini 16208e4ed6cSPaolo Bonzini def test_small_buffer2(self): 163ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 16408e4ed6cSPaolo Bonzini 16508e4ed6cSPaolo Bonzini qemu_img('create', '-f', iotests.imgfmt, '-o', 'cluster_size=%d,size=%d' 16608e4ed6cSPaolo Bonzini % (TestSingleDrive.image_len, TestSingleDrive.image_len), target_img) 16708e4ed6cSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 16808e4ed6cSPaolo Bonzini buf_size=65536, mode='existing', target=target_img) 16908e4ed6cSPaolo Bonzini self.assert_qmp(result, 'return', {}) 17008e4ed6cSPaolo Bonzini 17108e4ed6cSPaolo Bonzini self.complete_and_wait() 17208e4ed6cSPaolo Bonzini result = self.vm.qmp('query-block') 17308e4ed6cSPaolo Bonzini self.assert_qmp(result, 'return[0]/inserted/file', target_img) 17408e4ed6cSPaolo Bonzini self.vm.shutdown() 1753a3918c3SStefan Hajnoczi self.assertTrue(iotests.compare_images(test_img, target_img), 17608e4ed6cSPaolo Bonzini 'target image does not match source after mirroring') 17708e4ed6cSPaolo Bonzini 17844c7ca5eSPaolo Bonzini def test_large_cluster(self): 179ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 18044c7ca5eSPaolo Bonzini 18144c7ca5eSPaolo Bonzini qemu_img('create', '-f', iotests.imgfmt, '-o', 'cluster_size=%d,backing_file=%s' 18244c7ca5eSPaolo Bonzini % (TestSingleDrive.image_len, backing_img), target_img) 18344c7ca5eSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 18444c7ca5eSPaolo Bonzini mode='existing', target=target_img) 18544c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return', {}) 18644c7ca5eSPaolo Bonzini 18744c7ca5eSPaolo Bonzini self.complete_and_wait() 18844c7ca5eSPaolo Bonzini result = self.vm.qmp('query-block') 18944c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return[0]/inserted/file', target_img) 19044c7ca5eSPaolo Bonzini self.vm.shutdown() 1913a3918c3SStefan Hajnoczi self.assertTrue(iotests.compare_images(test_img, target_img), 19244c7ca5eSPaolo Bonzini 'target image does not match source after mirroring') 19344c7ca5eSPaolo Bonzini 19444c7ca5eSPaolo Bonzini def test_medium_not_found(self): 19544c7ca5eSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='ide1-cd0', sync='full', 19644c7ca5eSPaolo Bonzini target=target_img) 19744c7ca5eSPaolo Bonzini self.assert_qmp(result, 'error/class', 'GenericError') 19844c7ca5eSPaolo Bonzini 19944c7ca5eSPaolo Bonzini def test_image_not_found(self): 20044c7ca5eSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 20144c7ca5eSPaolo Bonzini mode='existing', target=target_img) 20244c7ca5eSPaolo Bonzini self.assert_qmp(result, 'error/class', 'GenericError') 20344c7ca5eSPaolo Bonzini 20444c7ca5eSPaolo Bonzini def test_device_not_found(self): 20544c7ca5eSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='nonexistent', sync='full', 20644c7ca5eSPaolo Bonzini target=target_img) 20744c7ca5eSPaolo Bonzini self.assert_qmp(result, 'error/class', 'DeviceNotFound') 20844c7ca5eSPaolo Bonzini 20944c7ca5eSPaolo Bonziniclass TestMirrorNoBacking(ImageMirroringTestCase): 21044c7ca5eSPaolo Bonzini image_len = 2 * 1024 * 1024 # MB 21144c7ca5eSPaolo Bonzini 21244c7ca5eSPaolo Bonzini def complete_and_wait(self, drive='drive0', wait_ready=True): 2132499a096SStefan Hajnoczi iotests.create_image(target_backing_img, TestMirrorNoBacking.image_len) 21444c7ca5eSPaolo Bonzini return ImageMirroringTestCase.complete_and_wait(self, drive, wait_ready) 21544c7ca5eSPaolo Bonzini 21644c7ca5eSPaolo Bonzini def compare_images(self, img1, img2): 2172499a096SStefan Hajnoczi iotests.create_image(target_backing_img, TestMirrorNoBacking.image_len) 2183a3918c3SStefan Hajnoczi return iotests.compare_images(img1, img2) 21944c7ca5eSPaolo Bonzini 22044c7ca5eSPaolo Bonzini def setUp(self): 2212499a096SStefan Hajnoczi iotests.create_image(backing_img, TestMirrorNoBacking.image_len) 22244c7ca5eSPaolo Bonzini qemu_img('create', '-f', iotests.imgfmt, '-o', 'backing_file=%s' % backing_img, test_img) 22344c7ca5eSPaolo Bonzini self.vm = iotests.VM().add_drive(test_img) 22444c7ca5eSPaolo Bonzini self.vm.launch() 22544c7ca5eSPaolo Bonzini 22644c7ca5eSPaolo Bonzini def tearDown(self): 22744c7ca5eSPaolo Bonzini self.vm.shutdown() 22844c7ca5eSPaolo Bonzini os.remove(test_img) 22944c7ca5eSPaolo Bonzini os.remove(backing_img) 23044c7ca5eSPaolo Bonzini os.remove(target_backing_img) 23144c7ca5eSPaolo Bonzini os.remove(target_img) 23244c7ca5eSPaolo Bonzini 23344c7ca5eSPaolo Bonzini def test_complete(self): 234ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 23544c7ca5eSPaolo Bonzini 23644c7ca5eSPaolo Bonzini qemu_img('create', '-f', iotests.imgfmt, '-o', 'backing_file=%s' % backing_img, target_img) 23744c7ca5eSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 23844c7ca5eSPaolo Bonzini mode='existing', target=target_img) 23944c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return', {}) 24044c7ca5eSPaolo Bonzini 24144c7ca5eSPaolo Bonzini self.complete_and_wait() 24244c7ca5eSPaolo Bonzini result = self.vm.qmp('query-block') 24344c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return[0]/inserted/file', target_img) 24444c7ca5eSPaolo Bonzini self.vm.shutdown() 24544c7ca5eSPaolo Bonzini self.assertTrue(self.compare_images(test_img, target_img), 24644c7ca5eSPaolo Bonzini 'target image does not match source after mirroring') 24744c7ca5eSPaolo Bonzini 24844c7ca5eSPaolo Bonzini def test_cancel(self): 249ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 25044c7ca5eSPaolo Bonzini 25144c7ca5eSPaolo Bonzini qemu_img('create', '-f', iotests.imgfmt, '-o', 'backing_file=%s' % backing_img, target_img) 25244c7ca5eSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 25344c7ca5eSPaolo Bonzini mode='existing', target=target_img) 25444c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return', {}) 25544c7ca5eSPaolo Bonzini 2562575fe16SStefan Hajnoczi self.wait_ready_and_cancel() 25744c7ca5eSPaolo Bonzini result = self.vm.qmp('query-block') 25844c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return[0]/inserted/file', test_img) 25944c7ca5eSPaolo Bonzini self.vm.shutdown() 26044c7ca5eSPaolo Bonzini self.assertTrue(self.compare_images(test_img, target_img), 26144c7ca5eSPaolo Bonzini 'target image does not match source after mirroring') 26244c7ca5eSPaolo Bonzini 263b812f671SPaolo Bonzini def test_large_cluster(self): 264ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 265b812f671SPaolo Bonzini 266b812f671SPaolo Bonzini # qemu-img create fails if the image is not there 267b812f671SPaolo Bonzini qemu_img('create', '-f', iotests.imgfmt, '-o', 'size=%d' 268b812f671SPaolo Bonzini %(TestMirrorNoBacking.image_len), target_backing_img) 269b812f671SPaolo Bonzini qemu_img('create', '-f', iotests.imgfmt, '-o', 'cluster_size=%d,backing_file=%s' 270b812f671SPaolo Bonzini % (TestMirrorNoBacking.image_len, target_backing_img), target_img) 271b812f671SPaolo Bonzini os.remove(target_backing_img) 272b812f671SPaolo Bonzini 273b812f671SPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 274b812f671SPaolo Bonzini mode='existing', target=target_img) 275b812f671SPaolo Bonzini self.assert_qmp(result, 'return', {}) 276b812f671SPaolo Bonzini 277b812f671SPaolo Bonzini self.complete_and_wait() 278b812f671SPaolo Bonzini result = self.vm.qmp('query-block') 279b812f671SPaolo Bonzini self.assert_qmp(result, 'return[0]/inserted/file', target_img) 280b812f671SPaolo Bonzini self.vm.shutdown() 281b812f671SPaolo Bonzini self.assertTrue(self.compare_images(test_img, target_img), 282b812f671SPaolo Bonzini 'target image does not match source after mirroring') 283b812f671SPaolo Bonzini 284a04eca10SVishvananda Ishayaclass TestMirrorResized(ImageMirroringTestCase): 285a04eca10SVishvananda Ishaya backing_len = 1 * 1024 * 1024 # MB 286a04eca10SVishvananda Ishaya image_len = 2 * 1024 * 1024 # MB 287a04eca10SVishvananda Ishaya 288a04eca10SVishvananda Ishaya def setUp(self): 2892499a096SStefan Hajnoczi iotests.create_image(backing_img, TestMirrorResized.backing_len) 290a04eca10SVishvananda Ishaya qemu_img('create', '-f', iotests.imgfmt, '-o', 'backing_file=%s' % backing_img, test_img) 291a04eca10SVishvananda Ishaya qemu_img('resize', test_img, '2M') 292a04eca10SVishvananda Ishaya self.vm = iotests.VM().add_drive(test_img) 293a04eca10SVishvananda Ishaya self.vm.launch() 294a04eca10SVishvananda Ishaya 295a04eca10SVishvananda Ishaya def tearDown(self): 296a04eca10SVishvananda Ishaya self.vm.shutdown() 297a04eca10SVishvananda Ishaya os.remove(test_img) 298a04eca10SVishvananda Ishaya os.remove(backing_img) 299a04eca10SVishvananda Ishaya try: 300a04eca10SVishvananda Ishaya os.remove(target_img) 301a04eca10SVishvananda Ishaya except OSError: 302a04eca10SVishvananda Ishaya pass 303a04eca10SVishvananda Ishaya 304a04eca10SVishvananda Ishaya def test_complete_top(self): 305ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 306a04eca10SVishvananda Ishaya 307a04eca10SVishvananda Ishaya result = self.vm.qmp('drive-mirror', device='drive0', sync='top', 308a04eca10SVishvananda Ishaya target=target_img) 309a04eca10SVishvananda Ishaya self.assert_qmp(result, 'return', {}) 310a04eca10SVishvananda Ishaya 311a04eca10SVishvananda Ishaya self.complete_and_wait() 312a04eca10SVishvananda Ishaya result = self.vm.qmp('query-block') 313a04eca10SVishvananda Ishaya self.assert_qmp(result, 'return[0]/inserted/file', target_img) 314a04eca10SVishvananda Ishaya self.vm.shutdown() 3153a3918c3SStefan Hajnoczi self.assertTrue(iotests.compare_images(test_img, target_img), 316a04eca10SVishvananda Ishaya 'target image does not match source after mirroring') 317a04eca10SVishvananda Ishaya 318a04eca10SVishvananda Ishaya def test_complete_full(self): 319ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 320a04eca10SVishvananda Ishaya 321a04eca10SVishvananda Ishaya result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 322a04eca10SVishvananda Ishaya target=target_img) 323a04eca10SVishvananda Ishaya self.assert_qmp(result, 'return', {}) 324a04eca10SVishvananda Ishaya 325a04eca10SVishvananda Ishaya self.complete_and_wait() 326a04eca10SVishvananda Ishaya result = self.vm.qmp('query-block') 327a04eca10SVishvananda Ishaya self.assert_qmp(result, 'return[0]/inserted/file', target_img) 328a04eca10SVishvananda Ishaya self.vm.shutdown() 3293a3918c3SStefan Hajnoczi self.assertTrue(iotests.compare_images(test_img, target_img), 330a04eca10SVishvananda Ishaya 'target image does not match source after mirroring') 331a04eca10SVishvananda Ishaya 3329dfa9f59SPaolo Bonziniclass TestReadErrors(ImageMirroringTestCase): 3339dfa9f59SPaolo Bonzini image_len = 2 * 1024 * 1024 # MB 3349dfa9f59SPaolo Bonzini 3359dfa9f59SPaolo Bonzini # this should be a multiple of twice the default granularity 3369dfa9f59SPaolo Bonzini # so that we hit this offset first in state 1 3379dfa9f59SPaolo Bonzini MIRROR_GRANULARITY = 1024 * 1024 3389dfa9f59SPaolo Bonzini 3399dfa9f59SPaolo Bonzini def create_blkdebug_file(self, name, event, errno): 3409dfa9f59SPaolo Bonzini file = open(name, 'w') 3419dfa9f59SPaolo Bonzini file.write(''' 3429dfa9f59SPaolo Bonzini[inject-error] 3439dfa9f59SPaolo Bonzinistate = "1" 3449dfa9f59SPaolo Bonzinievent = "%s" 3459dfa9f59SPaolo Bonzinierrno = "%d" 3469dfa9f59SPaolo Bonziniimmediately = "off" 3479dfa9f59SPaolo Bonzinionce = "on" 3489dfa9f59SPaolo Bonzinisector = "%d" 3499dfa9f59SPaolo Bonzini 3509dfa9f59SPaolo Bonzini[set-state] 3519dfa9f59SPaolo Bonzinistate = "1" 3529dfa9f59SPaolo Bonzinievent = "%s" 3539dfa9f59SPaolo Bonzininew_state = "2" 3549dfa9f59SPaolo Bonzini 3559dfa9f59SPaolo Bonzini[set-state] 3569dfa9f59SPaolo Bonzinistate = "2" 3579dfa9f59SPaolo Bonzinievent = "%s" 3589dfa9f59SPaolo Bonzininew_state = "1" 3599dfa9f59SPaolo Bonzini''' % (event, errno, self.MIRROR_GRANULARITY / 512, event, event)) 3609dfa9f59SPaolo Bonzini file.close() 3619dfa9f59SPaolo Bonzini 3629dfa9f59SPaolo Bonzini def setUp(self): 3639dfa9f59SPaolo Bonzini self.blkdebug_file = backing_img + ".blkdebug" 3642499a096SStefan Hajnoczi iotests.create_image(backing_img, TestReadErrors.image_len) 3659dfa9f59SPaolo Bonzini self.create_blkdebug_file(self.blkdebug_file, "read_aio", 5) 3669dfa9f59SPaolo Bonzini qemu_img('create', '-f', iotests.imgfmt, 3679dfa9f59SPaolo Bonzini '-o', 'backing_file=blkdebug:%s:%s,backing_fmt=raw' 3689dfa9f59SPaolo Bonzini % (self.blkdebug_file, backing_img), 3699dfa9f59SPaolo Bonzini test_img) 370b812f671SPaolo Bonzini # Write something for tests that use sync='top' 371b812f671SPaolo Bonzini qemu_io('-c', 'write %d 512' % (self.MIRROR_GRANULARITY + 65536), 372b812f671SPaolo Bonzini test_img) 3739dfa9f59SPaolo Bonzini self.vm = iotests.VM().add_drive(test_img) 3749dfa9f59SPaolo Bonzini self.vm.launch() 3759dfa9f59SPaolo Bonzini 3769dfa9f59SPaolo Bonzini def tearDown(self): 3779dfa9f59SPaolo Bonzini self.vm.shutdown() 3789dfa9f59SPaolo Bonzini os.remove(test_img) 3799dfa9f59SPaolo Bonzini os.remove(backing_img) 3809dfa9f59SPaolo Bonzini os.remove(self.blkdebug_file) 3819dfa9f59SPaolo Bonzini 3829dfa9f59SPaolo Bonzini def test_report_read(self): 383ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 3849dfa9f59SPaolo Bonzini 3859dfa9f59SPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 3869dfa9f59SPaolo Bonzini target=target_img) 3879dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return', {}) 3889dfa9f59SPaolo Bonzini 3899dfa9f59SPaolo Bonzini completed = False 3909dfa9f59SPaolo Bonzini error = False 3919dfa9f59SPaolo Bonzini while not completed: 3929dfa9f59SPaolo Bonzini for event in self.vm.get_qmp_events(wait=True): 3939dfa9f59SPaolo Bonzini if event['event'] == 'BLOCK_JOB_ERROR': 3949dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/device', 'drive0') 3959dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/operation', 'read') 3969dfa9f59SPaolo Bonzini error = True 3979dfa9f59SPaolo Bonzini elif event['event'] == 'BLOCK_JOB_READY': 3989dfa9f59SPaolo Bonzini self.assertTrue(False, 'job completed unexpectedly') 3999dfa9f59SPaolo Bonzini elif event['event'] == 'BLOCK_JOB_COMPLETED': 4009dfa9f59SPaolo Bonzini self.assertTrue(error, 'job completed unexpectedly') 4019dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/type', 'mirror') 4029dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/device', 'drive0') 4039dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/error', 'Input/output error') 4049dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/len', self.image_len) 4059dfa9f59SPaolo Bonzini completed = True 4069dfa9f59SPaolo Bonzini 407ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 4089dfa9f59SPaolo Bonzini self.vm.shutdown() 4099dfa9f59SPaolo Bonzini 4109dfa9f59SPaolo Bonzini def test_ignore_read(self): 411ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 4129dfa9f59SPaolo Bonzini 4139dfa9f59SPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 4149dfa9f59SPaolo Bonzini target=target_img, on_source_error='ignore') 4159dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return', {}) 4169dfa9f59SPaolo Bonzini 4179dfa9f59SPaolo Bonzini event = self.vm.get_qmp_event(wait=True) 4189dfa9f59SPaolo Bonzini self.assertEquals(event['event'], 'BLOCK_JOB_ERROR') 4199dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/device', 'drive0') 4209dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/operation', 'read') 4219dfa9f59SPaolo Bonzini result = self.vm.qmp('query-block-jobs') 4229dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return[0]/paused', False) 4239dfa9f59SPaolo Bonzini self.complete_and_wait() 4249dfa9f59SPaolo Bonzini self.vm.shutdown() 4259dfa9f59SPaolo Bonzini 426b812f671SPaolo Bonzini def test_large_cluster(self): 427ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 428b812f671SPaolo Bonzini 429b812f671SPaolo Bonzini # Test COW into the target image. The first half of the 430b812f671SPaolo Bonzini # cluster at MIRROR_GRANULARITY has to be copied from 431b812f671SPaolo Bonzini # backing_img, even though sync='top'. 432b812f671SPaolo Bonzini qemu_img('create', '-f', iotests.imgfmt, '-ocluster_size=131072,backing_file=%s' %(backing_img), target_img) 433b812f671SPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='top', 434b812f671SPaolo Bonzini on_source_error='ignore', 435b812f671SPaolo Bonzini mode='existing', target=target_img) 436b812f671SPaolo Bonzini self.assert_qmp(result, 'return', {}) 437b812f671SPaolo Bonzini 438b812f671SPaolo Bonzini event = self.vm.get_qmp_event(wait=True) 439b812f671SPaolo Bonzini self.assertEquals(event['event'], 'BLOCK_JOB_ERROR') 440b812f671SPaolo Bonzini self.assert_qmp(event, 'data/device', 'drive0') 441b812f671SPaolo Bonzini self.assert_qmp(event, 'data/operation', 'read') 442b812f671SPaolo Bonzini result = self.vm.qmp('query-block-jobs') 443b812f671SPaolo Bonzini self.assert_qmp(result, 'return[0]/paused', False) 444b812f671SPaolo Bonzini self.complete_and_wait() 445b812f671SPaolo Bonzini self.vm.shutdown() 446b812f671SPaolo Bonzini 447b812f671SPaolo Bonzini # Detach blkdebug to compare images successfully 448b812f671SPaolo Bonzini qemu_img('rebase', '-f', iotests.imgfmt, '-u', '-b', backing_img, test_img) 4493a3918c3SStefan Hajnoczi self.assertTrue(iotests.compare_images(test_img, target_img), 450b812f671SPaolo Bonzini 'target image does not match source after mirroring') 451b812f671SPaolo Bonzini 4529dfa9f59SPaolo Bonzini def test_stop_read(self): 453ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 4549dfa9f59SPaolo Bonzini 4559dfa9f59SPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 4569dfa9f59SPaolo Bonzini target=target_img, on_source_error='stop') 4579dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return', {}) 4589dfa9f59SPaolo Bonzini 4599dfa9f59SPaolo Bonzini error = False 4609dfa9f59SPaolo Bonzini ready = False 4619dfa9f59SPaolo Bonzini while not ready: 4629dfa9f59SPaolo Bonzini for event in self.vm.get_qmp_events(wait=True): 4639dfa9f59SPaolo Bonzini if event['event'] == 'BLOCK_JOB_ERROR': 4649dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/device', 'drive0') 4659dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/operation', 'read') 4669dfa9f59SPaolo Bonzini 4679dfa9f59SPaolo Bonzini result = self.vm.qmp('query-block-jobs') 4689dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return[0]/paused', True) 4699dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return[0]/io-status', 'failed') 4709dfa9f59SPaolo Bonzini 4719dfa9f59SPaolo Bonzini result = self.vm.qmp('block-job-resume', device='drive0') 4729dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return', {}) 4739dfa9f59SPaolo Bonzini error = True 4749dfa9f59SPaolo Bonzini elif event['event'] == 'BLOCK_JOB_READY': 4759dfa9f59SPaolo Bonzini self.assertTrue(error, 'job completed unexpectedly') 4769dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/device', 'drive0') 4779dfa9f59SPaolo Bonzini ready = True 4789dfa9f59SPaolo Bonzini 4799dfa9f59SPaolo Bonzini result = self.vm.qmp('query-block-jobs') 4809dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return[0]/paused', False) 4819dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return[0]/io-status', 'ok') 4829dfa9f59SPaolo Bonzini 4839dfa9f59SPaolo Bonzini self.complete_and_wait(wait_ready=False) 484ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 4859dfa9f59SPaolo Bonzini self.vm.shutdown() 4869dfa9f59SPaolo Bonzini 4879dfa9f59SPaolo Bonziniclass TestWriteErrors(ImageMirroringTestCase): 4889dfa9f59SPaolo Bonzini image_len = 2 * 1024 * 1024 # MB 4899dfa9f59SPaolo Bonzini 4909dfa9f59SPaolo Bonzini # this should be a multiple of twice the default granularity 4919dfa9f59SPaolo Bonzini # so that we hit this offset first in state 1 4929dfa9f59SPaolo Bonzini MIRROR_GRANULARITY = 1024 * 1024 4939dfa9f59SPaolo Bonzini 4949dfa9f59SPaolo Bonzini def create_blkdebug_file(self, name, event, errno): 4959dfa9f59SPaolo Bonzini file = open(name, 'w') 4969dfa9f59SPaolo Bonzini file.write(''' 4979dfa9f59SPaolo Bonzini[inject-error] 4989dfa9f59SPaolo Bonzinistate = "1" 4999dfa9f59SPaolo Bonzinievent = "%s" 5009dfa9f59SPaolo Bonzinierrno = "%d" 5019dfa9f59SPaolo Bonziniimmediately = "off" 5029dfa9f59SPaolo Bonzinionce = "on" 5039dfa9f59SPaolo Bonzinisector = "%d" 5049dfa9f59SPaolo Bonzini 5059dfa9f59SPaolo Bonzini[set-state] 5069dfa9f59SPaolo Bonzinistate = "1" 5079dfa9f59SPaolo Bonzinievent = "%s" 5089dfa9f59SPaolo Bonzininew_state = "2" 5099dfa9f59SPaolo Bonzini 5109dfa9f59SPaolo Bonzini[set-state] 5119dfa9f59SPaolo Bonzinistate = "2" 5129dfa9f59SPaolo Bonzinievent = "%s" 5139dfa9f59SPaolo Bonzininew_state = "1" 5149dfa9f59SPaolo Bonzini''' % (event, errno, self.MIRROR_GRANULARITY / 512, event, event)) 5159dfa9f59SPaolo Bonzini file.close() 5169dfa9f59SPaolo Bonzini 5179dfa9f59SPaolo Bonzini def setUp(self): 5189dfa9f59SPaolo Bonzini self.blkdebug_file = target_img + ".blkdebug" 5192499a096SStefan Hajnoczi iotests.create_image(backing_img, TestWriteErrors.image_len) 5209dfa9f59SPaolo Bonzini self.create_blkdebug_file(self.blkdebug_file, "write_aio", 5) 5219dfa9f59SPaolo Bonzini qemu_img('create', '-f', iotests.imgfmt, '-obacking_file=%s' %(backing_img), test_img) 5229dfa9f59SPaolo Bonzini self.vm = iotests.VM().add_drive(test_img) 5239dfa9f59SPaolo Bonzini self.target_img = 'blkdebug:%s:%s' % (self.blkdebug_file, target_img) 5249dfa9f59SPaolo Bonzini qemu_img('create', '-f', iotests.imgfmt, '-osize=%d' %(TestWriteErrors.image_len), target_img) 5259dfa9f59SPaolo Bonzini self.vm.launch() 5269dfa9f59SPaolo Bonzini 5279dfa9f59SPaolo Bonzini def tearDown(self): 5289dfa9f59SPaolo Bonzini self.vm.shutdown() 5299dfa9f59SPaolo Bonzini os.remove(test_img) 5309dfa9f59SPaolo Bonzini os.remove(backing_img) 5319dfa9f59SPaolo Bonzini os.remove(self.blkdebug_file) 5329dfa9f59SPaolo Bonzini 5339dfa9f59SPaolo Bonzini def test_report_write(self): 534ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 5359dfa9f59SPaolo Bonzini 5369dfa9f59SPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 5379dfa9f59SPaolo Bonzini mode='existing', target=self.target_img) 5389dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return', {}) 5399dfa9f59SPaolo Bonzini 5409dfa9f59SPaolo Bonzini completed = False 5419dfa9f59SPaolo Bonzini error = False 5429dfa9f59SPaolo Bonzini while not completed: 5439dfa9f59SPaolo Bonzini for event in self.vm.get_qmp_events(wait=True): 5449dfa9f59SPaolo Bonzini if event['event'] == 'BLOCK_JOB_ERROR': 5459dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/device', 'drive0') 5469dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/operation', 'write') 5479dfa9f59SPaolo Bonzini error = True 5489dfa9f59SPaolo Bonzini elif event['event'] == 'BLOCK_JOB_READY': 5499dfa9f59SPaolo Bonzini self.assertTrue(False, 'job completed unexpectedly') 5509dfa9f59SPaolo Bonzini elif event['event'] == 'BLOCK_JOB_COMPLETED': 5519dfa9f59SPaolo Bonzini self.assertTrue(error, 'job completed unexpectedly') 5529dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/type', 'mirror') 5539dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/device', 'drive0') 5549dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/error', 'Input/output error') 5559dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/len', self.image_len) 5569dfa9f59SPaolo Bonzini completed = True 5579dfa9f59SPaolo Bonzini 558ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 5599dfa9f59SPaolo Bonzini self.vm.shutdown() 5609dfa9f59SPaolo Bonzini 5619dfa9f59SPaolo Bonzini def test_ignore_write(self): 562ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 5639dfa9f59SPaolo Bonzini 5649dfa9f59SPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 5659dfa9f59SPaolo Bonzini mode='existing', target=self.target_img, 5669dfa9f59SPaolo Bonzini on_target_error='ignore') 5679dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return', {}) 5689dfa9f59SPaolo Bonzini 5699dfa9f59SPaolo Bonzini event = self.vm.get_qmp_event(wait=True) 5709dfa9f59SPaolo Bonzini self.assertEquals(event['event'], 'BLOCK_JOB_ERROR') 5719dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/device', 'drive0') 5729dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/operation', 'write') 5739dfa9f59SPaolo Bonzini result = self.vm.qmp('query-block-jobs') 5749dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return[0]/paused', False) 5759dfa9f59SPaolo Bonzini self.complete_and_wait() 5769dfa9f59SPaolo Bonzini self.vm.shutdown() 5779dfa9f59SPaolo Bonzini 5789dfa9f59SPaolo Bonzini def test_stop_write(self): 579ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 5809dfa9f59SPaolo Bonzini 5819dfa9f59SPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 5829dfa9f59SPaolo Bonzini mode='existing', target=self.target_img, 5839dfa9f59SPaolo Bonzini on_target_error='stop') 5849dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return', {}) 5859dfa9f59SPaolo Bonzini 5869dfa9f59SPaolo Bonzini error = False 5879dfa9f59SPaolo Bonzini ready = False 5889dfa9f59SPaolo Bonzini while not ready: 5899dfa9f59SPaolo Bonzini for event in self.vm.get_qmp_events(wait=True): 5909dfa9f59SPaolo Bonzini if event['event'] == 'BLOCK_JOB_ERROR': 5919dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/device', 'drive0') 5929dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/operation', 'write') 5939dfa9f59SPaolo Bonzini 5949dfa9f59SPaolo Bonzini result = self.vm.qmp('query-block-jobs') 5959dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return[0]/paused', True) 5969dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return[0]/io-status', 'failed') 5979dfa9f59SPaolo Bonzini 5989dfa9f59SPaolo Bonzini result = self.vm.qmp('block-job-resume', device='drive0') 5999dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return', {}) 6009dfa9f59SPaolo Bonzini 6019dfa9f59SPaolo Bonzini result = self.vm.qmp('query-block-jobs') 6029dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return[0]/paused', False) 6039dfa9f59SPaolo Bonzini self.assert_qmp(result, 'return[0]/io-status', 'ok') 6049dfa9f59SPaolo Bonzini error = True 6059dfa9f59SPaolo Bonzini elif event['event'] == 'BLOCK_JOB_READY': 6069dfa9f59SPaolo Bonzini self.assertTrue(error, 'job completed unexpectedly') 6079dfa9f59SPaolo Bonzini self.assert_qmp(event, 'data/device', 'drive0') 6089dfa9f59SPaolo Bonzini ready = True 6099dfa9f59SPaolo Bonzini 6109dfa9f59SPaolo Bonzini self.complete_and_wait(wait_ready=False) 611ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 6129dfa9f59SPaolo Bonzini self.vm.shutdown() 6139dfa9f59SPaolo Bonzini 61444c7ca5eSPaolo Bonziniclass TestSetSpeed(ImageMirroringTestCase): 61544c7ca5eSPaolo Bonzini image_len = 80 * 1024 * 1024 # MB 61644c7ca5eSPaolo Bonzini 61744c7ca5eSPaolo Bonzini def setUp(self): 61844c7ca5eSPaolo Bonzini qemu_img('create', backing_img, str(TestSetSpeed.image_len)) 61944c7ca5eSPaolo Bonzini qemu_img('create', '-f', iotests.imgfmt, '-o', 'backing_file=%s' % backing_img, test_img) 62044c7ca5eSPaolo Bonzini self.vm = iotests.VM().add_drive(test_img) 62144c7ca5eSPaolo Bonzini self.vm.launch() 62244c7ca5eSPaolo Bonzini 62344c7ca5eSPaolo Bonzini def tearDown(self): 62444c7ca5eSPaolo Bonzini self.vm.shutdown() 62544c7ca5eSPaolo Bonzini os.remove(test_img) 62644c7ca5eSPaolo Bonzini os.remove(backing_img) 62744c7ca5eSPaolo Bonzini os.remove(target_img) 62844c7ca5eSPaolo Bonzini 62944c7ca5eSPaolo Bonzini def test_set_speed(self): 630ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 63144c7ca5eSPaolo Bonzini 63244c7ca5eSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 63344c7ca5eSPaolo Bonzini target=target_img) 63444c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return', {}) 63544c7ca5eSPaolo Bonzini 63644c7ca5eSPaolo Bonzini # Default speed is 0 63744c7ca5eSPaolo Bonzini result = self.vm.qmp('query-block-jobs') 63844c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return[0]/device', 'drive0') 63944c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return[0]/speed', 0) 64044c7ca5eSPaolo Bonzini 64144c7ca5eSPaolo Bonzini result = self.vm.qmp('block-job-set-speed', device='drive0', speed=8 * 1024 * 1024) 64244c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return', {}) 64344c7ca5eSPaolo Bonzini 64444c7ca5eSPaolo Bonzini # Ensure the speed we set was accepted 64544c7ca5eSPaolo Bonzini result = self.vm.qmp('query-block-jobs') 64644c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return[0]/device', 'drive0') 64744c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return[0]/speed', 8 * 1024 * 1024) 64844c7ca5eSPaolo Bonzini 6492575fe16SStefan Hajnoczi self.wait_ready_and_cancel() 65044c7ca5eSPaolo Bonzini 65144c7ca5eSPaolo Bonzini # Check setting speed in drive-mirror works 65244c7ca5eSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 65344c7ca5eSPaolo Bonzini target=target_img, speed=4*1024*1024) 65444c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return', {}) 65544c7ca5eSPaolo Bonzini 65644c7ca5eSPaolo Bonzini result = self.vm.qmp('query-block-jobs') 65744c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return[0]/device', 'drive0') 65844c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return[0]/speed', 4 * 1024 * 1024) 65944c7ca5eSPaolo Bonzini 6602575fe16SStefan Hajnoczi self.wait_ready_and_cancel() 66144c7ca5eSPaolo Bonzini 66244c7ca5eSPaolo Bonzini def test_set_speed_invalid(self): 663ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 66444c7ca5eSPaolo Bonzini 66544c7ca5eSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 66644c7ca5eSPaolo Bonzini target=target_img, speed=-1) 66744c7ca5eSPaolo Bonzini self.assert_qmp(result, 'error/class', 'GenericError') 66844c7ca5eSPaolo Bonzini 669ecc1c88eSStefan Hajnoczi self.assert_no_active_block_jobs() 67044c7ca5eSPaolo Bonzini 67144c7ca5eSPaolo Bonzini result = self.vm.qmp('drive-mirror', device='drive0', sync='full', 67244c7ca5eSPaolo Bonzini target=target_img) 67344c7ca5eSPaolo Bonzini self.assert_qmp(result, 'return', {}) 67444c7ca5eSPaolo Bonzini 67544c7ca5eSPaolo Bonzini result = self.vm.qmp('block-job-set-speed', device='drive0', speed=-1) 67644c7ca5eSPaolo Bonzini self.assert_qmp(result, 'error/class', 'GenericError') 67744c7ca5eSPaolo Bonzini 6782575fe16SStefan Hajnoczi self.wait_ready_and_cancel() 67944c7ca5eSPaolo Bonzini 680*c15badeeSMax Reitzclass TestUnbackedSource(ImageMirroringTestCase): 681*c15badeeSMax Reitz image_len = 2 * 1024 * 1024 # MB 682*c15badeeSMax Reitz 683*c15badeeSMax Reitz def setUp(self): 684*c15badeeSMax Reitz qemu_img('create', '-f', iotests.imgfmt, test_img, 685*c15badeeSMax Reitz str(TestUnbackedSource.image_len)) 686*c15badeeSMax Reitz self.vm = iotests.VM().add_drive(test_img) 687*c15badeeSMax Reitz self.vm.launch() 688*c15badeeSMax Reitz 689*c15badeeSMax Reitz def tearDown(self): 690*c15badeeSMax Reitz self.vm.shutdown() 691*c15badeeSMax Reitz os.remove(test_img) 692*c15badeeSMax Reitz os.remove(target_img) 693*c15badeeSMax Reitz 694*c15badeeSMax Reitz def test_absolute_paths(self): 695*c15badeeSMax Reitz self.assert_no_active_block_jobs() 696*c15badeeSMax Reitz 697*c15badeeSMax Reitz for sync_mode in ['full', 'top', 'none']: 698*c15badeeSMax Reitz result = self.vm.qmp('drive-mirror', device='drive0', 699*c15badeeSMax Reitz sync=sync_mode, target=target_img, 700*c15badeeSMax Reitz mode='absolute-paths') 701*c15badeeSMax Reitz self.assert_qmp(result, 'return', {}) 702*c15badeeSMax Reitz self.complete_and_wait() 703*c15badeeSMax Reitz self.assert_no_active_block_jobs() 704*c15badeeSMax Reitz 70544c7ca5eSPaolo Bonziniif __name__ == '__main__': 70644c7ca5eSPaolo Bonzini iotests.main(supported_fmts=['qcow2', 'qed']) 707