1903cb1bfSPhilippe Mathieu-Daudé#!/usr/bin/env python3 29dd003a9SVladimir Sementsov-Ogievskiy# group: rw 3e5ca8fddSStefan Hajnoczi# 47c6a4ab8SFam Zheng# Tests for drive-backup and blockdev-backup 5e5ca8fddSStefan Hajnoczi# 67c6a4ab8SFam Zheng# Copyright (C) 2013, 2014 Red Hat, Inc. 7e5ca8fddSStefan Hajnoczi# 8e5ca8fddSStefan Hajnoczi# Based on 041. 9e5ca8fddSStefan Hajnoczi# 10e5ca8fddSStefan Hajnoczi# This program is free software; you can redistribute it and/or modify 11e5ca8fddSStefan Hajnoczi# it under the terms of the GNU General Public License as published by 12e5ca8fddSStefan Hajnoczi# the Free Software Foundation; either version 2 of the License, or 13e5ca8fddSStefan Hajnoczi# (at your option) any later version. 14e5ca8fddSStefan Hajnoczi# 15e5ca8fddSStefan Hajnoczi# This program is distributed in the hope that it will be useful, 16e5ca8fddSStefan Hajnoczi# but WITHOUT ANY WARRANTY; without even the implied warranty of 17e5ca8fddSStefan Hajnoczi# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 18e5ca8fddSStefan Hajnoczi# GNU General Public License for more details. 19e5ca8fddSStefan Hajnoczi# 20e5ca8fddSStefan Hajnoczi# You should have received a copy of the GNU General Public License 21e5ca8fddSStefan Hajnoczi# along with this program. If not, see <http://www.gnu.org/licenses/>. 22e5ca8fddSStefan Hajnoczi# 23e5ca8fddSStefan Hajnoczi 24e5ca8fddSStefan Hajnocziimport time 25e5ca8fddSStefan Hajnocziimport os 26e5ca8fddSStefan Hajnocziimport iotests 27e5ca8fddSStefan Hajnoczifrom iotests import qemu_img, qemu_io 28e5ca8fddSStefan Hajnoczi 29e5ca8fddSStefan Hajnoczitest_img = os.path.join(iotests.test_dir, 'test.img') 30e5ca8fddSStefan Hajnoczitarget_img = os.path.join(iotests.test_dir, 'target.img') 317c6a4ab8SFam Zhengblockdev_target_img = os.path.join(iotests.test_dir, 'blockdev-target.img') 32e5ca8fddSStefan Hajnoczi 33e5ca8fddSStefan Hajnocziimage_len = 64 * 1024 * 1024 # MB 34e5ca8fddSStefan Hajnoczi 35819cec01SVladimir Sementsov-Ogievskiydef setUpModule(): 36819cec01SVladimir Sementsov-Ogievskiy qemu_img('create', '-f', iotests.imgfmt, test_img, str(image_len)) 37819cec01SVladimir Sementsov-Ogievskiy qemu_io('-f', iotests.imgfmt, '-c', 'write -P0x11 0 64k', test_img) 38819cec01SVladimir Sementsov-Ogievskiy qemu_io('-f', iotests.imgfmt, '-c', 'write -P0x00 64k 128k', test_img) 39819cec01SVladimir Sementsov-Ogievskiy qemu_io('-f', iotests.imgfmt, '-c', 'write -P0x22 162k 32k', test_img) 4090c9b167SKevin Wolf qemu_io('-f', iotests.imgfmt, '-c', 'write -P0xd5 1M 32k', test_img) 4190c9b167SKevin Wolf qemu_io('-f', iotests.imgfmt, '-c', 'write -P0xdc 32M 124k', test_img) 42819cec01SVladimir Sementsov-Ogievskiy qemu_io('-f', iotests.imgfmt, '-c', 'write -P0x33 67043328 64k', test_img) 43819cec01SVladimir Sementsov-Ogievskiy 44819cec01SVladimir Sementsov-Ogievskiydef tearDownModule(): 45819cec01SVladimir Sementsov-Ogievskiy os.remove(test_img) 46819cec01SVladimir Sementsov-Ogievskiy 47819cec01SVladimir Sementsov-Ogievskiy 48819cec01SVladimir Sementsov-Ogievskiyclass TestSingleDrive(iotests.QMPTestCase): 49819cec01SVladimir Sementsov-Ogievskiy def setUp(self): 50819cec01SVladimir Sementsov-Ogievskiy qemu_img('create', '-f', iotests.imgfmt, blockdev_target_img, str(image_len)) 51e5ca8fddSStefan Hajnoczi 520a82a927SKevin Wolf self.vm = iotests.VM() 530a82a927SKevin Wolf self.vm.add_drive('blkdebug::' + test_img, 'node-name=source') 540a82a927SKevin Wolf self.vm.add_drive(blockdev_target_img, 'node-name=target', 550a82a927SKevin Wolf interface="none") 560ed82f7aSMax Reitz if iotests.qemu_default_machine == 'pc': 570ed82f7aSMax Reitz self.vm.add_drive(None, 'media=cdrom', 'ide') 58e5ca8fddSStefan Hajnoczi self.vm.launch() 59e5ca8fddSStefan Hajnoczi 60e5ca8fddSStefan Hajnoczi def tearDown(self): 61e5ca8fddSStefan Hajnoczi self.vm.shutdown() 627c6a4ab8SFam Zheng os.remove(blockdev_target_img) 63e5ca8fddSStefan Hajnoczi try: 64e5ca8fddSStefan Hajnoczi os.remove(target_img) 65e5ca8fddSStefan Hajnoczi except OSError: 66e5ca8fddSStefan Hajnoczi pass 67e5ca8fddSStefan Hajnoczi 687c6a4ab8SFam Zheng def do_test_cancel(self, cmd, target): 69e5ca8fddSStefan Hajnoczi self.assert_no_active_block_jobs() 70e5ca8fddSStefan Hajnoczi 71bc11aee2SMax Reitz self.vm.pause_drive('drive0') 72*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd(cmd, device='drive0', target=target, sync='full') 73e5ca8fddSStefan Hajnoczi 74bc11aee2SMax Reitz event = self.cancel_and_wait(resume=True) 75e5ca8fddSStefan Hajnoczi self.assert_qmp(event, 'data/type', 'backup') 76e5ca8fddSStefan Hajnoczi 777c6a4ab8SFam Zheng def test_cancel_drive_backup(self): 787c6a4ab8SFam Zheng self.do_test_cancel('drive-backup', target_img) 797c6a4ab8SFam Zheng 807c6a4ab8SFam Zheng def test_cancel_blockdev_backup(self): 817c6a4ab8SFam Zheng self.do_test_cancel('blockdev-backup', 'drive1') 827c6a4ab8SFam Zheng 837c6a4ab8SFam Zheng def do_test_pause(self, cmd, target, image): 84e5ca8fddSStefan Hajnoczi self.assert_no_active_block_jobs() 85e5ca8fddSStefan Hajnoczi 86b59b3d57SFam Zheng self.vm.pause_drive('drive0') 87*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd(cmd, device='drive0', 887c6a4ab8SFam Zheng target=target, sync='full') 89e5ca8fddSStefan Hajnoczi 90f03d9d24SJohn Snow self.pause_job('drive0', wait=False) 91b59b3d57SFam Zheng self.vm.resume_drive('drive0') 92f03d9d24SJohn Snow self.pause_wait('drive0') 932c93c5cbSKevin Wolf 94e5ca8fddSStefan Hajnoczi result = self.vm.qmp('query-block-jobs') 95e5ca8fddSStefan Hajnoczi offset = self.dictpath(result, 'return[0]/offset') 96e5ca8fddSStefan Hajnoczi 972c93c5cbSKevin Wolf time.sleep(0.5) 98e5ca8fddSStefan Hajnoczi result = self.vm.qmp('query-block-jobs') 99e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'return[0]/offset', offset) 100e5ca8fddSStefan Hajnoczi 101*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd('block-job-resume', device='drive0') 102e5ca8fddSStefan Hajnoczi 103e5ca8fddSStefan Hajnoczi self.wait_until_completed() 104e5ca8fddSStefan Hajnoczi 105e5ca8fddSStefan Hajnoczi self.vm.shutdown() 1067c6a4ab8SFam Zheng self.assertTrue(iotests.compare_images(test_img, image), 107e5ca8fddSStefan Hajnoczi 'target image does not match source after backup') 108e5ca8fddSStefan Hajnoczi 1097c6a4ab8SFam Zheng def test_pause_drive_backup(self): 1107c6a4ab8SFam Zheng self.do_test_pause('drive-backup', target_img, target_img) 1117c6a4ab8SFam Zheng 1127c6a4ab8SFam Zheng def test_pause_blockdev_backup(self): 1137c6a4ab8SFam Zheng self.do_test_pause('blockdev-backup', 'drive1', blockdev_target_img) 1147c6a4ab8SFam Zheng 1150a82a927SKevin Wolf def do_test_resize_blockdev_backup(self, device, node): 1160a82a927SKevin Wolf def pre_finalize(): 1170a82a927SKevin Wolf result = self.vm.qmp('block_resize', device=device, size=65536) 1180a82a927SKevin Wolf self.assert_qmp(result, 'error/class', 'GenericError') 1190a82a927SKevin Wolf 1200a82a927SKevin Wolf result = self.vm.qmp('block_resize', node_name=node, size=65536) 1210a82a927SKevin Wolf self.assert_qmp(result, 'error/class', 'GenericError') 1220a82a927SKevin Wolf 123*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd('blockdev-backup', job_id='job0', device='drive0', 1240a82a927SKevin Wolf target='drive1', sync='full', auto_finalize=False, 1250a82a927SKevin Wolf auto_dismiss=False) 1260a82a927SKevin Wolf 1270a82a927SKevin Wolf self.vm.run_job('job0', auto_finalize=False, pre_finalize=pre_finalize) 1280a82a927SKevin Wolf 1290a82a927SKevin Wolf def test_source_resize_blockdev_backup(self): 1300a82a927SKevin Wolf self.do_test_resize_blockdev_backup('drive0', 'source') 1310a82a927SKevin Wolf 1320a82a927SKevin Wolf def test_target_resize_blockdev_backup(self): 1330a82a927SKevin Wolf self.do_test_resize_blockdev_backup('drive1', 'target') 1340a82a927SKevin Wolf 1350a82a927SKevin Wolf def do_test_target_size(self, size): 136*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd('block_resize', device='drive1', size=size) 1370a82a927SKevin Wolf 1380a82a927SKevin Wolf result = self.vm.qmp('blockdev-backup', job_id='job0', device='drive0', 1390a82a927SKevin Wolf target='drive1', sync='full') 1400a82a927SKevin Wolf self.assert_qmp(result, 'error/class', 'GenericError') 1410a82a927SKevin Wolf 1420a82a927SKevin Wolf def test_small_target(self): 1430a82a927SKevin Wolf self.do_test_target_size(image_len // 2) 1440a82a927SKevin Wolf 1450a82a927SKevin Wolf def test_large_target(self): 1460a82a927SKevin Wolf self.do_test_target_size(image_len * 2) 1470a82a927SKevin Wolf 148e5ca8fddSStefan Hajnoczi def test_medium_not_found(self): 149d8683155SBo Tu if iotests.qemu_default_machine != 'pc': 150d8683155SBo Tu return 151d8683155SBo Tu 1520ed82f7aSMax Reitz result = self.vm.qmp('drive-backup', device='drive2', # CD-ROM 153b53169eaSStefan Hajnoczi target=target_img, sync='full') 154e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'error/class', 'GenericError') 155e5ca8fddSStefan Hajnoczi 1567c6a4ab8SFam Zheng def test_medium_not_found_blockdev_backup(self): 157d8683155SBo Tu if iotests.qemu_default_machine != 'pc': 158d8683155SBo Tu return 159d8683155SBo Tu 1600ed82f7aSMax Reitz result = self.vm.qmp('blockdev-backup', device='drive2', # CD-ROM 1617c6a4ab8SFam Zheng target='drive1', sync='full') 1627c6a4ab8SFam Zheng self.assert_qmp(result, 'error/class', 'GenericError') 1637c6a4ab8SFam Zheng 164e5ca8fddSStefan Hajnoczi def test_image_not_found(self): 165e5ca8fddSStefan Hajnoczi result = self.vm.qmp('drive-backup', device='drive0', 166b53169eaSStefan Hajnoczi target=target_img, sync='full', mode='existing') 167e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'error/class', 'GenericError') 168e5ca8fddSStefan Hajnoczi 169e3409362SIan Main def test_invalid_format(self): 170e3409362SIan Main result = self.vm.qmp('drive-backup', device='drive0', 171e3409362SIan Main target=target_img, sync='full', 172e3409362SIan Main format='spaghetti-noodles') 173e3409362SIan Main self.assert_qmp(result, 'error/class', 'GenericError') 174e3409362SIan Main 1757c6a4ab8SFam Zheng def do_test_device_not_found(self, cmd, **args): 1767c6a4ab8SFam Zheng result = self.vm.qmp(cmd, **args) 1775b347c54SMarkus Armbruster self.assert_qmp(result, 'error/class', 'GenericError') 178e5ca8fddSStefan Hajnoczi 1797c6a4ab8SFam Zheng def test_device_not_found(self): 1807c6a4ab8SFam Zheng self.do_test_device_not_found('drive-backup', device='nonexistent', 1817c6a4ab8SFam Zheng target=target_img, sync='full') 1827c6a4ab8SFam Zheng 1837c6a4ab8SFam Zheng self.do_test_device_not_found('blockdev-backup', device='nonexistent', 1847c6a4ab8SFam Zheng target='drive0', sync='full') 1857c6a4ab8SFam Zheng 1867c6a4ab8SFam Zheng self.do_test_device_not_found('blockdev-backup', device='drive0', 1877c6a4ab8SFam Zheng target='nonexistent', sync='full') 1887c6a4ab8SFam Zheng 1897c6a4ab8SFam Zheng self.do_test_device_not_found('blockdev-backup', device='nonexistent', 1907c6a4ab8SFam Zheng target='nonexistent', sync='full') 1917c6a4ab8SFam Zheng 1927c6a4ab8SFam Zheng def test_target_is_source(self): 1937c6a4ab8SFam Zheng result = self.vm.qmp('blockdev-backup', device='drive0', 1947c6a4ab8SFam Zheng target='drive0', sync='full') 1957c6a4ab8SFam Zheng self.assert_qmp(result, 'error/class', 'GenericError') 1967c6a4ab8SFam Zheng 197e5ca8fddSStefan Hajnocziclass TestSetSpeed(iotests.QMPTestCase): 198e5ca8fddSStefan Hajnoczi def setUp(self): 199819cec01SVladimir Sementsov-Ogievskiy qemu_img('create', '-f', iotests.imgfmt, blockdev_target_img, str(image_len)) 2007c6a4ab8SFam Zheng 201bc11aee2SMax Reitz self.vm = iotests.VM().add_drive('blkdebug::' + test_img) 2024e9e4323SKevin Wolf self.vm.add_drive(blockdev_target_img, interface="none") 203e5ca8fddSStefan Hajnoczi self.vm.launch() 204e5ca8fddSStefan Hajnoczi 205e5ca8fddSStefan Hajnoczi def tearDown(self): 206e5ca8fddSStefan Hajnoczi self.vm.shutdown() 2077c6a4ab8SFam Zheng os.remove(blockdev_target_img) 2087c6a4ab8SFam Zheng try: 209e5ca8fddSStefan Hajnoczi os.remove(target_img) 2107c6a4ab8SFam Zheng except OSError: 2117c6a4ab8SFam Zheng pass 212e5ca8fddSStefan Hajnoczi 2137c6a4ab8SFam Zheng def do_test_set_speed(self, cmd, target): 214e5ca8fddSStefan Hajnoczi self.assert_no_active_block_jobs() 215e5ca8fddSStefan Hajnoczi 216b59b3d57SFam Zheng self.vm.pause_drive('drive0') 217*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd(cmd, device='drive0', target=target, sync='full') 218e5ca8fddSStefan Hajnoczi 219e5ca8fddSStefan Hajnoczi # Default speed is 0 220e5ca8fddSStefan Hajnoczi result = self.vm.qmp('query-block-jobs') 221e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'return[0]/device', 'drive0') 222e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'return[0]/speed', 0) 223e5ca8fddSStefan Hajnoczi 224*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd('block-job-set-speed', device='drive0', speed=8 * 1024 * 1024) 225e5ca8fddSStefan Hajnoczi 226e5ca8fddSStefan Hajnoczi # Ensure the speed we set was accepted 227e5ca8fddSStefan Hajnoczi result = self.vm.qmp('query-block-jobs') 228e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'return[0]/device', 'drive0') 229e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'return[0]/speed', 8 * 1024 * 1024) 230e5ca8fddSStefan Hajnoczi 231b59b3d57SFam Zheng event = self.cancel_and_wait(resume=True) 232e5ca8fddSStefan Hajnoczi self.assert_qmp(event, 'data/type', 'backup') 233e5ca8fddSStefan Hajnoczi 2347c6a4ab8SFam Zheng # Check setting speed option works 235b59b3d57SFam Zheng self.vm.pause_drive('drive0') 236*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd(cmd, device='drive0', 2377c6a4ab8SFam Zheng target=target, sync='full', speed=4*1024*1024) 238e5ca8fddSStefan Hajnoczi 239e5ca8fddSStefan Hajnoczi result = self.vm.qmp('query-block-jobs') 240e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'return[0]/device', 'drive0') 241e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'return[0]/speed', 4 * 1024 * 1024) 242e5ca8fddSStefan Hajnoczi 243b59b3d57SFam Zheng event = self.cancel_and_wait(resume=True) 244e5ca8fddSStefan Hajnoczi self.assert_qmp(event, 'data/type', 'backup') 245e5ca8fddSStefan Hajnoczi 2467c6a4ab8SFam Zheng def test_set_speed_drive_backup(self): 2477c6a4ab8SFam Zheng self.do_test_set_speed('drive-backup', target_img) 2487c6a4ab8SFam Zheng 2497c6a4ab8SFam Zheng def test_set_speed_blockdev_backup(self): 2507c6a4ab8SFam Zheng self.do_test_set_speed('blockdev-backup', 'drive1') 2517c6a4ab8SFam Zheng 2527c6a4ab8SFam Zheng def do_test_set_speed_invalid(self, cmd, target): 253e5ca8fddSStefan Hajnoczi self.assert_no_active_block_jobs() 254e5ca8fddSStefan Hajnoczi 2557c6a4ab8SFam Zheng result = self.vm.qmp(cmd, device='drive0', 2567c6a4ab8SFam Zheng target=target, sync='full', speed=-1) 257e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'error/class', 'GenericError') 258e5ca8fddSStefan Hajnoczi 259e5ca8fddSStefan Hajnoczi self.assert_no_active_block_jobs() 260e5ca8fddSStefan Hajnoczi 261b59b3d57SFam Zheng self.vm.pause_drive('drive0') 262*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd(cmd, device='drive0', 2637c6a4ab8SFam Zheng target=target, sync='full') 264e5ca8fddSStefan Hajnoczi 265e5ca8fddSStefan Hajnoczi result = self.vm.qmp('block-job-set-speed', device='drive0', speed=-1) 266e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'error/class', 'GenericError') 267e5ca8fddSStefan Hajnoczi 268b59b3d57SFam Zheng event = self.cancel_and_wait(resume=True) 269e5ca8fddSStefan Hajnoczi self.assert_qmp(event, 'data/type', 'backup') 270e5ca8fddSStefan Hajnoczi 2717c6a4ab8SFam Zheng def test_set_speed_invalid_drive_backup(self): 2727c6a4ab8SFam Zheng self.do_test_set_speed_invalid('drive-backup', target_img) 2737c6a4ab8SFam Zheng 2747c6a4ab8SFam Zheng def test_set_speed_invalid_blockdev_backup(self): 2757c6a4ab8SFam Zheng self.do_test_set_speed_invalid('blockdev-backup', 'drive1') 2767c6a4ab8SFam Zheng 277bc11aee2SMax Reitz# Note: We cannot use pause_drive() here, or the transaction command 278bc11aee2SMax Reitz# would stall. Instead, we limit the block job speed here. 279e5ca8fddSStefan Hajnocziclass TestSingleTransaction(iotests.QMPTestCase): 280e5ca8fddSStefan Hajnoczi def setUp(self): 281819cec01SVladimir Sementsov-Ogievskiy qemu_img('create', '-f', iotests.imgfmt, blockdev_target_img, str(image_len)) 282e5ca8fddSStefan Hajnoczi 2834e9e4323SKevin Wolf self.vm = iotests.VM().add_drive(test_img) 2844e9e4323SKevin Wolf self.vm.add_drive(blockdev_target_img, interface="none") 2850ed82f7aSMax Reitz if iotests.qemu_default_machine == 'pc': 2860ed82f7aSMax Reitz self.vm.add_drive(None, 'media=cdrom', 'ide') 287e5ca8fddSStefan Hajnoczi self.vm.launch() 288e5ca8fddSStefan Hajnoczi 289e5ca8fddSStefan Hajnoczi def tearDown(self): 290e5ca8fddSStefan Hajnoczi self.vm.shutdown() 2917c6a4ab8SFam Zheng os.remove(blockdev_target_img) 292e5ca8fddSStefan Hajnoczi try: 293e5ca8fddSStefan Hajnoczi os.remove(target_img) 294e5ca8fddSStefan Hajnoczi except OSError: 295e5ca8fddSStefan Hajnoczi pass 296e5ca8fddSStefan Hajnoczi 2977c6a4ab8SFam Zheng def do_test_cancel(self, cmd, target): 298e5ca8fddSStefan Hajnoczi self.assert_no_active_block_jobs() 299e5ca8fddSStefan Hajnoczi 300*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd('transaction', actions=[{ 3017c6a4ab8SFam Zheng 'type': cmd, 302e5ca8fddSStefan Hajnoczi 'data': { 'device': 'drive0', 3037c6a4ab8SFam Zheng 'target': target, 304bc11aee2SMax Reitz 'sync': 'full', 305bc11aee2SMax Reitz 'speed': 64 * 1024 }, 306e5ca8fddSStefan Hajnoczi } 307e5ca8fddSStefan Hajnoczi ]) 3087c6a4ab8SFam Zheng 309e5ca8fddSStefan Hajnoczi event = self.cancel_and_wait() 310e5ca8fddSStefan Hajnoczi self.assert_qmp(event, 'data/type', 'backup') 311e5ca8fddSStefan Hajnoczi 3127c6a4ab8SFam Zheng def test_cancel_drive_backup(self): 3137c6a4ab8SFam Zheng self.do_test_cancel('drive-backup', target_img) 3147c6a4ab8SFam Zheng 3157c6a4ab8SFam Zheng def test_cancel_blockdev_backup(self): 3167c6a4ab8SFam Zheng self.do_test_cancel('blockdev-backup', 'drive1') 3177c6a4ab8SFam Zheng 3187c6a4ab8SFam Zheng def do_test_pause(self, cmd, target, image): 319e5ca8fddSStefan Hajnoczi self.assert_no_active_block_jobs() 320e5ca8fddSStefan Hajnoczi 321*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd('transaction', actions=[{ 3227c6a4ab8SFam Zheng 'type': cmd, 323e5ca8fddSStefan Hajnoczi 'data': { 'device': 'drive0', 3247c6a4ab8SFam Zheng 'target': target, 325bc11aee2SMax Reitz 'sync': 'full', 326bc11aee2SMax Reitz 'speed': 64 * 1024 }, 327e5ca8fddSStefan Hajnoczi } 328e5ca8fddSStefan Hajnoczi ]) 329e5ca8fddSStefan Hajnoczi 330f03d9d24SJohn Snow self.pause_job('drive0', wait=False) 331e5ca8fddSStefan Hajnoczi 332*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd('block-job-set-speed', device='drive0', speed=0) 333bc11aee2SMax Reitz 334f03d9d24SJohn Snow self.pause_wait('drive0') 3352c93c5cbSKevin Wolf 336e5ca8fddSStefan Hajnoczi result = self.vm.qmp('query-block-jobs') 337e5ca8fddSStefan Hajnoczi offset = self.dictpath(result, 'return[0]/offset') 338e5ca8fddSStefan Hajnoczi 3392c93c5cbSKevin Wolf time.sleep(0.5) 340e5ca8fddSStefan Hajnoczi result = self.vm.qmp('query-block-jobs') 341e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'return[0]/offset', offset) 342e5ca8fddSStefan Hajnoczi 343*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd('block-job-resume', device='drive0') 344e5ca8fddSStefan Hajnoczi 345e5ca8fddSStefan Hajnoczi self.wait_until_completed() 346e5ca8fddSStefan Hajnoczi 347e5ca8fddSStefan Hajnoczi self.vm.shutdown() 3487c6a4ab8SFam Zheng self.assertTrue(iotests.compare_images(test_img, image), 349e5ca8fddSStefan Hajnoczi 'target image does not match source after backup') 350e5ca8fddSStefan Hajnoczi 3517c6a4ab8SFam Zheng def test_pause_drive_backup(self): 3527c6a4ab8SFam Zheng self.do_test_pause('drive-backup', target_img, target_img) 3537c6a4ab8SFam Zheng 3547c6a4ab8SFam Zheng def test_pause_blockdev_backup(self): 3557c6a4ab8SFam Zheng self.do_test_pause('blockdev-backup', 'drive1', blockdev_target_img) 3567c6a4ab8SFam Zheng 3577c6a4ab8SFam Zheng def do_test_medium_not_found(self, cmd, target): 358d8683155SBo Tu if iotests.qemu_default_machine != 'pc': 359d8683155SBo Tu return 360d8683155SBo Tu 361e5ca8fddSStefan Hajnoczi result = self.vm.qmp('transaction', actions=[{ 3627c6a4ab8SFam Zheng 'type': cmd, 3630ed82f7aSMax Reitz 'data': { 'device': 'drive2', # CD-ROM 3647c6a4ab8SFam Zheng 'target': target, 365b53169eaSStefan Hajnoczi 'sync': 'full' }, 366e5ca8fddSStefan Hajnoczi } 367e5ca8fddSStefan Hajnoczi ]) 368e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'error/class', 'GenericError') 369e5ca8fddSStefan Hajnoczi 3707c6a4ab8SFam Zheng def test_medium_not_found_drive_backup(self): 3717c6a4ab8SFam Zheng self.do_test_medium_not_found('drive-backup', target_img) 3727c6a4ab8SFam Zheng 3737c6a4ab8SFam Zheng def test_medium_not_found_blockdev_backup(self): 3747c6a4ab8SFam Zheng self.do_test_medium_not_found('blockdev-backup', 'drive1') 3757c6a4ab8SFam Zheng 376e5ca8fddSStefan Hajnoczi def test_image_not_found(self): 377e5ca8fddSStefan Hajnoczi result = self.vm.qmp('transaction', actions=[{ 378e5ca8fddSStefan Hajnoczi 'type': 'drive-backup', 379e5ca8fddSStefan Hajnoczi 'data': { 'device': 'drive0', 380e5ca8fddSStefan Hajnoczi 'mode': 'existing', 381b53169eaSStefan Hajnoczi 'target': target_img, 382b53169eaSStefan Hajnoczi 'sync': 'full' }, 383e5ca8fddSStefan Hajnoczi } 384e5ca8fddSStefan Hajnoczi ]) 385e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'error/class', 'GenericError') 386e5ca8fddSStefan Hajnoczi 387e5ca8fddSStefan Hajnoczi def test_device_not_found(self): 388e5ca8fddSStefan Hajnoczi result = self.vm.qmp('transaction', actions=[{ 389e5ca8fddSStefan Hajnoczi 'type': 'drive-backup', 390e5ca8fddSStefan Hajnoczi 'data': { 'device': 'nonexistent', 391e5ca8fddSStefan Hajnoczi 'mode': 'existing', 392b53169eaSStefan Hajnoczi 'target': target_img, 393b53169eaSStefan Hajnoczi 'sync': 'full' }, 394e5ca8fddSStefan Hajnoczi } 395e5ca8fddSStefan Hajnoczi ]) 396b7e4fa22SKevin Wolf self.assert_qmp(result, 'error/class', 'GenericError') 397e5ca8fddSStefan Hajnoczi 3987c6a4ab8SFam Zheng result = self.vm.qmp('transaction', actions=[{ 3997c6a4ab8SFam Zheng 'type': 'blockdev-backup', 4007c6a4ab8SFam Zheng 'data': { 'device': 'nonexistent', 4017c6a4ab8SFam Zheng 'target': 'drive1', 4027c6a4ab8SFam Zheng 'sync': 'full' }, 4037c6a4ab8SFam Zheng } 4047c6a4ab8SFam Zheng ]) 4055b347c54SMarkus Armbruster self.assert_qmp(result, 'error/class', 'GenericError') 4067c6a4ab8SFam Zheng 4077c6a4ab8SFam Zheng result = self.vm.qmp('transaction', actions=[{ 4087c6a4ab8SFam Zheng 'type': 'blockdev-backup', 4097c6a4ab8SFam Zheng 'data': { 'device': 'drive0', 4107c6a4ab8SFam Zheng 'target': 'nonexistent', 4117c6a4ab8SFam Zheng 'sync': 'full' }, 4127c6a4ab8SFam Zheng } 4137c6a4ab8SFam Zheng ]) 4145b347c54SMarkus Armbruster self.assert_qmp(result, 'error/class', 'GenericError') 4157c6a4ab8SFam Zheng 4167c6a4ab8SFam Zheng result = self.vm.qmp('transaction', actions=[{ 4177c6a4ab8SFam Zheng 'type': 'blockdev-backup', 4187c6a4ab8SFam Zheng 'data': { 'device': 'nonexistent', 4197c6a4ab8SFam Zheng 'target': 'nonexistent', 4207c6a4ab8SFam Zheng 'sync': 'full' }, 4217c6a4ab8SFam Zheng } 4227c6a4ab8SFam Zheng ]) 4235b347c54SMarkus Armbruster self.assert_qmp(result, 'error/class', 'GenericError') 4247c6a4ab8SFam Zheng 4257c6a4ab8SFam Zheng def test_target_is_source(self): 4267c6a4ab8SFam Zheng result = self.vm.qmp('transaction', actions=[{ 4277c6a4ab8SFam Zheng 'type': 'blockdev-backup', 4287c6a4ab8SFam Zheng 'data': { 'device': 'drive0', 4297c6a4ab8SFam Zheng 'target': 'drive0', 4307c6a4ab8SFam Zheng 'sync': 'full' }, 4317c6a4ab8SFam Zheng } 4327c6a4ab8SFam Zheng ]) 4337c6a4ab8SFam Zheng self.assert_qmp(result, 'error/class', 'GenericError') 4347c6a4ab8SFam Zheng 435e5ca8fddSStefan Hajnoczi def test_abort(self): 436e5ca8fddSStefan Hajnoczi result = self.vm.qmp('transaction', actions=[{ 437e5ca8fddSStefan Hajnoczi 'type': 'drive-backup', 438e5ca8fddSStefan Hajnoczi 'data': { 'device': 'nonexistent', 439e5ca8fddSStefan Hajnoczi 'mode': 'existing', 440b53169eaSStefan Hajnoczi 'target': target_img, 441b53169eaSStefan Hajnoczi 'sync': 'full' }, 442e5ca8fddSStefan Hajnoczi }, { 443e5ca8fddSStefan Hajnoczi 'type': 'Abort', 444e5ca8fddSStefan Hajnoczi 'data': {}, 445e5ca8fddSStefan Hajnoczi } 446e5ca8fddSStefan Hajnoczi ]) 447e5ca8fddSStefan Hajnoczi self.assert_qmp(result, 'error/class', 'GenericError') 448e5ca8fddSStefan Hajnoczi self.assert_no_active_block_jobs() 449e5ca8fddSStefan Hajnoczi 4507c6a4ab8SFam Zheng result = self.vm.qmp('transaction', actions=[{ 4517c6a4ab8SFam Zheng 'type': 'blockdev-backup', 4527c6a4ab8SFam Zheng 'data': { 'device': 'nonexistent', 4537c6a4ab8SFam Zheng 'target': 'drive1', 4547c6a4ab8SFam Zheng 'sync': 'full' }, 4557c6a4ab8SFam Zheng }, { 4567c6a4ab8SFam Zheng 'type': 'Abort', 4577c6a4ab8SFam Zheng 'data': {}, 4587c6a4ab8SFam Zheng } 4597c6a4ab8SFam Zheng ]) 4607c6a4ab8SFam Zheng self.assert_qmp(result, 'error/class', 'GenericError') 4617c6a4ab8SFam Zheng self.assert_no_active_block_jobs() 4627c6a4ab8SFam Zheng 4637c6a4ab8SFam Zheng result = self.vm.qmp('transaction', actions=[{ 4647c6a4ab8SFam Zheng 'type': 'blockdev-backup', 4657c6a4ab8SFam Zheng 'data': { 'device': 'drive0', 4667c6a4ab8SFam Zheng 'target': 'nonexistent', 4677c6a4ab8SFam Zheng 'sync': 'full' }, 4687c6a4ab8SFam Zheng }, { 4697c6a4ab8SFam Zheng 'type': 'Abort', 4707c6a4ab8SFam Zheng 'data': {}, 4717c6a4ab8SFam Zheng } 4727c6a4ab8SFam Zheng ]) 4737c6a4ab8SFam Zheng self.assert_qmp(result, 'error/class', 'GenericError') 4747c6a4ab8SFam Zheng self.assert_no_active_block_jobs() 4757c6a4ab8SFam Zheng 476e1b5c51fSPavel Butsykin 4778e837294SVladimir Sementsov-Ogievskiyclass TestCompressedToQcow2(iotests.QMPTestCase): 478e1b5c51fSPavel Butsykin image_len = 64 * 1024 * 1024 # MB 4796a9d73bdSKevin Wolf target_fmt = {'type': 'qcow2', 'args': (), 'drive-opts': ''} 480e1b5c51fSPavel Butsykin 481e1b5c51fSPavel Butsykin def tearDown(self): 482e1b5c51fSPavel Butsykin self.vm.shutdown() 483e1b5c51fSPavel Butsykin os.remove(blockdev_target_img) 484e1b5c51fSPavel Butsykin try: 485e1b5c51fSPavel Butsykin os.remove(target_img) 486e1b5c51fSPavel Butsykin except OSError: 487e1b5c51fSPavel Butsykin pass 488e1b5c51fSPavel Butsykin 4898e837294SVladimir Sementsov-Ogievskiy def do_prepare_drives(self, attach_target): 4906a9d73bdSKevin Wolf self.vm = iotests.VM().add_drive('blkdebug::' + test_img, 4916a9d73bdSKevin Wolf opts=self.target_fmt['drive-opts']) 49200198eccSPavel Butsykin 4938e837294SVladimir Sementsov-Ogievskiy qemu_img('create', '-f', self.target_fmt['type'], blockdev_target_img, 4948e837294SVladimir Sementsov-Ogievskiy str(self.image_len), *self.target_fmt['args']) 4954797aeabSFam Zheng if attach_target: 4961d3d4b63SJohn Snow self.vm.add_drive(blockdev_target_img, 4978e837294SVladimir Sementsov-Ogievskiy img_format=self.target_fmt['type'], 4986a9d73bdSKevin Wolf interface="none", 4996a9d73bdSKevin Wolf opts=self.target_fmt['drive-opts']) 50000198eccSPavel Butsykin 50100198eccSPavel Butsykin self.vm.launch() 50200198eccSPavel Butsykin 5038e837294SVladimir Sementsov-Ogievskiy def do_test_compress_complete(self, cmd, attach_target, **args): 5048e837294SVladimir Sementsov-Ogievskiy self.do_prepare_drives(attach_target) 50500198eccSPavel Butsykin 506e1b5c51fSPavel Butsykin self.assert_no_active_block_jobs() 507e1b5c51fSPavel Butsykin 508*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd(cmd, device='drive0', sync='full', compress=True, **args) 509e1b5c51fSPavel Butsykin 510e1b5c51fSPavel Butsykin self.wait_until_completed() 511e1b5c51fSPavel Butsykin 512e1b5c51fSPavel Butsykin self.vm.shutdown() 513e1b5c51fSPavel Butsykin self.assertTrue(iotests.compare_images(test_img, blockdev_target_img, 5148e837294SVladimir Sementsov-Ogievskiy iotests.imgfmt, 5158e837294SVladimir Sementsov-Ogievskiy self.target_fmt['type']), 516e1b5c51fSPavel Butsykin 'target image does not match source after backup') 517e1b5c51fSPavel Butsykin 518e1b5c51fSPavel Butsykin def test_complete_compress_drive_backup(self): 5198e837294SVladimir Sementsov-Ogievskiy self.do_test_compress_complete('drive-backup', False, 5208e837294SVladimir Sementsov-Ogievskiy target=blockdev_target_img, 5218e837294SVladimir Sementsov-Ogievskiy mode='existing') 522e1b5c51fSPavel Butsykin 523e1b5c51fSPavel Butsykin def test_complete_compress_blockdev_backup(self): 5248e837294SVladimir Sementsov-Ogievskiy self.do_test_compress_complete('blockdev-backup', 5258e837294SVladimir Sementsov-Ogievskiy True, target='drive1') 526e1b5c51fSPavel Butsykin 5278e837294SVladimir Sementsov-Ogievskiy def do_test_compress_cancel(self, cmd, attach_target, **args): 5288e837294SVladimir Sementsov-Ogievskiy self.do_prepare_drives(attach_target) 52900198eccSPavel Butsykin 530e1b5c51fSPavel Butsykin self.assert_no_active_block_jobs() 531e1b5c51fSPavel Butsykin 532bc11aee2SMax Reitz self.vm.pause_drive('drive0') 533*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd(cmd, device='drive0', sync='full', compress=True, **args) 534e1b5c51fSPavel Butsykin 535bc11aee2SMax Reitz event = self.cancel_and_wait(resume=True) 536e1b5c51fSPavel Butsykin self.assert_qmp(event, 'data/type', 'backup') 537e1b5c51fSPavel Butsykin 53800198eccSPavel Butsykin self.vm.shutdown() 53900198eccSPavel Butsykin 540e1b5c51fSPavel Butsykin def test_compress_cancel_drive_backup(self): 5418e837294SVladimir Sementsov-Ogievskiy self.do_test_compress_cancel('drive-backup', False, 5428e837294SVladimir Sementsov-Ogievskiy target=blockdev_target_img, 5438e837294SVladimir Sementsov-Ogievskiy mode='existing') 544e1b5c51fSPavel Butsykin 545e1b5c51fSPavel Butsykin def test_compress_cancel_blockdev_backup(self): 5468e837294SVladimir Sementsov-Ogievskiy self.do_test_compress_cancel('blockdev-backup', True, 5474797aeabSFam Zheng target='drive1') 548e1b5c51fSPavel Butsykin 5498e837294SVladimir Sementsov-Ogievskiy def do_test_compress_pause(self, cmd, attach_target, **args): 5508e837294SVladimir Sementsov-Ogievskiy self.do_prepare_drives(attach_target) 55100198eccSPavel Butsykin 552e1b5c51fSPavel Butsykin self.assert_no_active_block_jobs() 553e1b5c51fSPavel Butsykin 554e1b5c51fSPavel Butsykin self.vm.pause_drive('drive0') 555*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd(cmd, device='drive0', sync='full', compress=True, **args) 556e1b5c51fSPavel Butsykin 557f03d9d24SJohn Snow self.pause_job('drive0', wait=False) 558e1b5c51fSPavel Butsykin self.vm.resume_drive('drive0') 559f03d9d24SJohn Snow self.pause_wait('drive0') 5602c93c5cbSKevin Wolf 561e1b5c51fSPavel Butsykin result = self.vm.qmp('query-block-jobs') 562e1b5c51fSPavel Butsykin offset = self.dictpath(result, 'return[0]/offset') 563e1b5c51fSPavel Butsykin 5642c93c5cbSKevin Wolf time.sleep(0.5) 565e1b5c51fSPavel Butsykin result = self.vm.qmp('query-block-jobs') 566e1b5c51fSPavel Butsykin self.assert_qmp(result, 'return[0]/offset', offset) 567e1b5c51fSPavel Butsykin 568*b6aed193SVladimir Sementsov-Ogievskiy self.vm.cmd('block-job-resume', device='drive0') 569e1b5c51fSPavel Butsykin 570e1b5c51fSPavel Butsykin self.wait_until_completed() 571e1b5c51fSPavel Butsykin 572e1b5c51fSPavel Butsykin self.vm.shutdown() 573e1b5c51fSPavel Butsykin self.assertTrue(iotests.compare_images(test_img, blockdev_target_img, 5748e837294SVladimir Sementsov-Ogievskiy iotests.imgfmt, 5758e837294SVladimir Sementsov-Ogievskiy self.target_fmt['type']), 576e1b5c51fSPavel Butsykin 'target image does not match source after backup') 577e1b5c51fSPavel Butsykin 578e1b5c51fSPavel Butsykin def test_compress_pause_drive_backup(self): 5798e837294SVladimir Sementsov-Ogievskiy self.do_test_compress_pause('drive-backup', False, 5808e837294SVladimir Sementsov-Ogievskiy target=blockdev_target_img, 5818e837294SVladimir Sementsov-Ogievskiy mode='existing') 582e1b5c51fSPavel Butsykin 583e1b5c51fSPavel Butsykin def test_compress_pause_blockdev_backup(self): 5848e837294SVladimir Sementsov-Ogievskiy self.do_test_compress_pause('blockdev-backup', True, 5854797aeabSFam Zheng target='drive1') 586e1b5c51fSPavel Butsykin 5878e837294SVladimir Sementsov-Ogievskiy 5888e837294SVladimir Sementsov-Ogievskiyclass TestCompressedToVmdk(TestCompressedToQcow2): 5896a9d73bdSKevin Wolf target_fmt = {'type': 'vmdk', 'args': ('-o', 'subformat=streamOptimized'), 5906a9d73bdSKevin Wolf 'drive-opts': 'cache.no-flush=on'} 5918e837294SVladimir Sementsov-Ogievskiy 592761cd2e7SVladimir Sementsov-Ogievskiy @iotests.skip_if_unsupported(['vmdk']) 593761cd2e7SVladimir Sementsov-Ogievskiy def setUp(self): 594761cd2e7SVladimir Sementsov-Ogievskiy pass 595761cd2e7SVladimir Sementsov-Ogievskiy 5968e837294SVladimir Sementsov-Ogievskiy 597e5ca8fddSStefan Hajnocziif __name__ == '__main__': 598103cbc77SMax Reitz iotests.main(supported_fmts=['raw', 'qcow2'], 599103cbc77SMax Reitz supported_protocols=['file']) 600