xref: /openbmc/qemu/tests/qemu-iotests/041 (revision 3b9f27d2b34cb8c5cc6cec993712c7e1943e9de9)
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):
67*3b9f27d2SFam Zheng        iotests.create_image(backing_img, self.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'
166*3b9f27d2SFam Zheng                        % (self.image_len, self.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'
182*3b9f27d2SFam Zheng                        % (self.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
209*3b9f27d2SFam Zhengclass TestSingleDriveZeroLength(TestSingleDrive):
210*3b9f27d2SFam Zheng    image_len = 0
211*3b9f27d2SFam Zheng    test_small_buffer2 = None
212*3b9f27d2SFam Zheng    test_large_cluster = None
213*3b9f27d2SFam Zheng
21444c7ca5eSPaolo Bonziniclass TestMirrorNoBacking(ImageMirroringTestCase):
21544c7ca5eSPaolo Bonzini    image_len = 2 * 1024 * 1024 # MB
21644c7ca5eSPaolo Bonzini
21744c7ca5eSPaolo Bonzini    def complete_and_wait(self, drive='drive0', wait_ready=True):
2182499a096SStefan Hajnoczi        iotests.create_image(target_backing_img, TestMirrorNoBacking.image_len)
21944c7ca5eSPaolo Bonzini        return ImageMirroringTestCase.complete_and_wait(self, drive, wait_ready)
22044c7ca5eSPaolo Bonzini
22144c7ca5eSPaolo Bonzini    def compare_images(self, img1, img2):
2222499a096SStefan Hajnoczi        iotests.create_image(target_backing_img, TestMirrorNoBacking.image_len)
2233a3918c3SStefan Hajnoczi        return iotests.compare_images(img1, img2)
22444c7ca5eSPaolo Bonzini
22544c7ca5eSPaolo Bonzini    def setUp(self):
2262499a096SStefan Hajnoczi        iotests.create_image(backing_img, TestMirrorNoBacking.image_len)
22744c7ca5eSPaolo Bonzini        qemu_img('create', '-f', iotests.imgfmt, '-o', 'backing_file=%s' % backing_img, test_img)
22844c7ca5eSPaolo Bonzini        self.vm = iotests.VM().add_drive(test_img)
22944c7ca5eSPaolo Bonzini        self.vm.launch()
23044c7ca5eSPaolo Bonzini
23144c7ca5eSPaolo Bonzini    def tearDown(self):
23244c7ca5eSPaolo Bonzini        self.vm.shutdown()
23344c7ca5eSPaolo Bonzini        os.remove(test_img)
23444c7ca5eSPaolo Bonzini        os.remove(backing_img)
23544c7ca5eSPaolo Bonzini        os.remove(target_backing_img)
23644c7ca5eSPaolo Bonzini        os.remove(target_img)
23744c7ca5eSPaolo Bonzini
23844c7ca5eSPaolo Bonzini    def test_complete(self):
239ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
24044c7ca5eSPaolo Bonzini
24144c7ca5eSPaolo Bonzini        qemu_img('create', '-f', iotests.imgfmt, '-o', 'backing_file=%s' % backing_img, target_img)
24244c7ca5eSPaolo Bonzini        result = self.vm.qmp('drive-mirror', device='drive0', sync='full',
24344c7ca5eSPaolo Bonzini                             mode='existing', target=target_img)
24444c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'return', {})
24544c7ca5eSPaolo Bonzini
24644c7ca5eSPaolo Bonzini        self.complete_and_wait()
24744c7ca5eSPaolo Bonzini        result = self.vm.qmp('query-block')
24844c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'return[0]/inserted/file', target_img)
24944c7ca5eSPaolo Bonzini        self.vm.shutdown()
25044c7ca5eSPaolo Bonzini        self.assertTrue(self.compare_images(test_img, target_img),
25144c7ca5eSPaolo Bonzini                        'target image does not match source after mirroring')
25244c7ca5eSPaolo Bonzini
25344c7ca5eSPaolo Bonzini    def test_cancel(self):
254ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
25544c7ca5eSPaolo Bonzini
25644c7ca5eSPaolo Bonzini        qemu_img('create', '-f', iotests.imgfmt, '-o', 'backing_file=%s' % backing_img, target_img)
25744c7ca5eSPaolo Bonzini        result = self.vm.qmp('drive-mirror', device='drive0', sync='full',
25844c7ca5eSPaolo Bonzini                             mode='existing', target=target_img)
25944c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'return', {})
26044c7ca5eSPaolo Bonzini
2612575fe16SStefan Hajnoczi        self.wait_ready_and_cancel()
26244c7ca5eSPaolo Bonzini        result = self.vm.qmp('query-block')
26344c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'return[0]/inserted/file', test_img)
26444c7ca5eSPaolo Bonzini        self.vm.shutdown()
26544c7ca5eSPaolo Bonzini        self.assertTrue(self.compare_images(test_img, target_img),
26644c7ca5eSPaolo Bonzini                        'target image does not match source after mirroring')
26744c7ca5eSPaolo Bonzini
268b812f671SPaolo Bonzini    def test_large_cluster(self):
269ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
270b812f671SPaolo Bonzini
271b812f671SPaolo Bonzini        # qemu-img create fails if the image is not there
272b812f671SPaolo Bonzini        qemu_img('create', '-f', iotests.imgfmt, '-o', 'size=%d'
273b812f671SPaolo Bonzini                        %(TestMirrorNoBacking.image_len), target_backing_img)
274b812f671SPaolo Bonzini        qemu_img('create', '-f', iotests.imgfmt, '-o', 'cluster_size=%d,backing_file=%s'
275b812f671SPaolo Bonzini                        % (TestMirrorNoBacking.image_len, target_backing_img), target_img)
276b812f671SPaolo Bonzini        os.remove(target_backing_img)
277b812f671SPaolo Bonzini
278b812f671SPaolo Bonzini        result = self.vm.qmp('drive-mirror', device='drive0', sync='full',
279b812f671SPaolo Bonzini                             mode='existing', target=target_img)
280b812f671SPaolo Bonzini        self.assert_qmp(result, 'return', {})
281b812f671SPaolo Bonzini
282b812f671SPaolo Bonzini        self.complete_and_wait()
283b812f671SPaolo Bonzini        result = self.vm.qmp('query-block')
284b812f671SPaolo Bonzini        self.assert_qmp(result, 'return[0]/inserted/file', target_img)
285b812f671SPaolo Bonzini        self.vm.shutdown()
286b812f671SPaolo Bonzini        self.assertTrue(self.compare_images(test_img, target_img),
287b812f671SPaolo Bonzini                        'target image does not match source after mirroring')
288b812f671SPaolo Bonzini
289a04eca10SVishvananda Ishayaclass TestMirrorResized(ImageMirroringTestCase):
290a04eca10SVishvananda Ishaya    backing_len = 1 * 1024 * 1024 # MB
291a04eca10SVishvananda Ishaya    image_len = 2 * 1024 * 1024 # MB
292a04eca10SVishvananda Ishaya
293a04eca10SVishvananda Ishaya    def setUp(self):
2942499a096SStefan Hajnoczi        iotests.create_image(backing_img, TestMirrorResized.backing_len)
295a04eca10SVishvananda Ishaya        qemu_img('create', '-f', iotests.imgfmt, '-o', 'backing_file=%s' % backing_img, test_img)
296a04eca10SVishvananda Ishaya        qemu_img('resize', test_img, '2M')
297a04eca10SVishvananda Ishaya        self.vm = iotests.VM().add_drive(test_img)
298a04eca10SVishvananda Ishaya        self.vm.launch()
299a04eca10SVishvananda Ishaya
300a04eca10SVishvananda Ishaya    def tearDown(self):
301a04eca10SVishvananda Ishaya        self.vm.shutdown()
302a04eca10SVishvananda Ishaya        os.remove(test_img)
303a04eca10SVishvananda Ishaya        os.remove(backing_img)
304a04eca10SVishvananda Ishaya        try:
305a04eca10SVishvananda Ishaya            os.remove(target_img)
306a04eca10SVishvananda Ishaya        except OSError:
307a04eca10SVishvananda Ishaya            pass
308a04eca10SVishvananda Ishaya
309a04eca10SVishvananda Ishaya    def test_complete_top(self):
310ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
311a04eca10SVishvananda Ishaya
312a04eca10SVishvananda Ishaya        result = self.vm.qmp('drive-mirror', device='drive0', sync='top',
313a04eca10SVishvananda Ishaya                             target=target_img)
314a04eca10SVishvananda Ishaya        self.assert_qmp(result, 'return', {})
315a04eca10SVishvananda Ishaya
316a04eca10SVishvananda Ishaya        self.complete_and_wait()
317a04eca10SVishvananda Ishaya        result = self.vm.qmp('query-block')
318a04eca10SVishvananda Ishaya        self.assert_qmp(result, 'return[0]/inserted/file', target_img)
319a04eca10SVishvananda Ishaya        self.vm.shutdown()
3203a3918c3SStefan Hajnoczi        self.assertTrue(iotests.compare_images(test_img, target_img),
321a04eca10SVishvananda Ishaya                        'target image does not match source after mirroring')
322a04eca10SVishvananda Ishaya
323a04eca10SVishvananda Ishaya    def test_complete_full(self):
324ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
325a04eca10SVishvananda Ishaya
326a04eca10SVishvananda Ishaya        result = self.vm.qmp('drive-mirror', device='drive0', sync='full',
327a04eca10SVishvananda Ishaya                             target=target_img)
328a04eca10SVishvananda Ishaya        self.assert_qmp(result, 'return', {})
329a04eca10SVishvananda Ishaya
330a04eca10SVishvananda Ishaya        self.complete_and_wait()
331a04eca10SVishvananda Ishaya        result = self.vm.qmp('query-block')
332a04eca10SVishvananda Ishaya        self.assert_qmp(result, 'return[0]/inserted/file', target_img)
333a04eca10SVishvananda Ishaya        self.vm.shutdown()
3343a3918c3SStefan Hajnoczi        self.assertTrue(iotests.compare_images(test_img, target_img),
335a04eca10SVishvananda Ishaya                        'target image does not match source after mirroring')
336a04eca10SVishvananda Ishaya
3379dfa9f59SPaolo Bonziniclass TestReadErrors(ImageMirroringTestCase):
3389dfa9f59SPaolo Bonzini    image_len = 2 * 1024 * 1024 # MB
3399dfa9f59SPaolo Bonzini
3409dfa9f59SPaolo Bonzini    # this should be a multiple of twice the default granularity
3419dfa9f59SPaolo Bonzini    # so that we hit this offset first in state 1
3429dfa9f59SPaolo Bonzini    MIRROR_GRANULARITY = 1024 * 1024
3439dfa9f59SPaolo Bonzini
3449dfa9f59SPaolo Bonzini    def create_blkdebug_file(self, name, event, errno):
3459dfa9f59SPaolo Bonzini        file = open(name, 'w')
3469dfa9f59SPaolo Bonzini        file.write('''
3479dfa9f59SPaolo Bonzini[inject-error]
3489dfa9f59SPaolo Bonzinistate = "1"
3499dfa9f59SPaolo Bonzinievent = "%s"
3509dfa9f59SPaolo Bonzinierrno = "%d"
3519dfa9f59SPaolo Bonziniimmediately = "off"
3529dfa9f59SPaolo Bonzinionce = "on"
3539dfa9f59SPaolo Bonzinisector = "%d"
3549dfa9f59SPaolo Bonzini
3559dfa9f59SPaolo Bonzini[set-state]
3569dfa9f59SPaolo Bonzinistate = "1"
3579dfa9f59SPaolo Bonzinievent = "%s"
3589dfa9f59SPaolo Bonzininew_state = "2"
3599dfa9f59SPaolo Bonzini
3609dfa9f59SPaolo Bonzini[set-state]
3619dfa9f59SPaolo Bonzinistate = "2"
3629dfa9f59SPaolo Bonzinievent = "%s"
3639dfa9f59SPaolo Bonzininew_state = "1"
3649dfa9f59SPaolo Bonzini''' % (event, errno, self.MIRROR_GRANULARITY / 512, event, event))
3659dfa9f59SPaolo Bonzini        file.close()
3669dfa9f59SPaolo Bonzini
3679dfa9f59SPaolo Bonzini    def setUp(self):
3689dfa9f59SPaolo Bonzini        self.blkdebug_file = backing_img + ".blkdebug"
3692499a096SStefan Hajnoczi        iotests.create_image(backing_img, TestReadErrors.image_len)
3709dfa9f59SPaolo Bonzini        self.create_blkdebug_file(self.blkdebug_file, "read_aio", 5)
3719dfa9f59SPaolo Bonzini        qemu_img('create', '-f', iotests.imgfmt,
3729dfa9f59SPaolo Bonzini                 '-o', 'backing_file=blkdebug:%s:%s,backing_fmt=raw'
3739dfa9f59SPaolo Bonzini                       % (self.blkdebug_file, backing_img),
3749dfa9f59SPaolo Bonzini                 test_img)
375b812f671SPaolo Bonzini        # Write something for tests that use sync='top'
376b812f671SPaolo Bonzini        qemu_io('-c', 'write %d 512' % (self.MIRROR_GRANULARITY + 65536),
377b812f671SPaolo Bonzini                        test_img)
3789dfa9f59SPaolo Bonzini        self.vm = iotests.VM().add_drive(test_img)
3799dfa9f59SPaolo Bonzini        self.vm.launch()
3809dfa9f59SPaolo Bonzini
3819dfa9f59SPaolo Bonzini    def tearDown(self):
3829dfa9f59SPaolo Bonzini        self.vm.shutdown()
3839dfa9f59SPaolo Bonzini        os.remove(test_img)
3849dfa9f59SPaolo Bonzini        os.remove(backing_img)
3859dfa9f59SPaolo Bonzini        os.remove(self.blkdebug_file)
3869dfa9f59SPaolo Bonzini
3879dfa9f59SPaolo Bonzini    def test_report_read(self):
388ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
3899dfa9f59SPaolo Bonzini
3909dfa9f59SPaolo Bonzini        result = self.vm.qmp('drive-mirror', device='drive0', sync='full',
3919dfa9f59SPaolo Bonzini                             target=target_img)
3929dfa9f59SPaolo Bonzini        self.assert_qmp(result, 'return', {})
3939dfa9f59SPaolo Bonzini
3949dfa9f59SPaolo Bonzini        completed = False
3959dfa9f59SPaolo Bonzini        error = False
3969dfa9f59SPaolo Bonzini        while not completed:
3979dfa9f59SPaolo Bonzini            for event in self.vm.get_qmp_events(wait=True):
3989dfa9f59SPaolo Bonzini                if event['event'] == 'BLOCK_JOB_ERROR':
3999dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/device', 'drive0')
4009dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/operation', 'read')
4019dfa9f59SPaolo Bonzini                    error = True
4029dfa9f59SPaolo Bonzini                elif event['event'] == 'BLOCK_JOB_READY':
4039dfa9f59SPaolo Bonzini                    self.assertTrue(False, 'job completed unexpectedly')
4049dfa9f59SPaolo Bonzini                elif event['event'] == 'BLOCK_JOB_COMPLETED':
4059dfa9f59SPaolo Bonzini                    self.assertTrue(error, 'job completed unexpectedly')
4069dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/type', 'mirror')
4079dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/device', 'drive0')
4089dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/error', 'Input/output error')
4099dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/len', self.image_len)
4109dfa9f59SPaolo Bonzini                    completed = True
4119dfa9f59SPaolo Bonzini
412ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
4139dfa9f59SPaolo Bonzini        self.vm.shutdown()
4149dfa9f59SPaolo Bonzini
4159dfa9f59SPaolo Bonzini    def test_ignore_read(self):
416ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
4179dfa9f59SPaolo Bonzini
4189dfa9f59SPaolo Bonzini        result = self.vm.qmp('drive-mirror', device='drive0', sync='full',
4199dfa9f59SPaolo Bonzini                             target=target_img, on_source_error='ignore')
4209dfa9f59SPaolo Bonzini        self.assert_qmp(result, 'return', {})
4219dfa9f59SPaolo Bonzini
4229dfa9f59SPaolo Bonzini        event = self.vm.get_qmp_event(wait=True)
4239dfa9f59SPaolo Bonzini        self.assertEquals(event['event'], 'BLOCK_JOB_ERROR')
4249dfa9f59SPaolo Bonzini        self.assert_qmp(event, 'data/device', 'drive0')
4259dfa9f59SPaolo Bonzini        self.assert_qmp(event, 'data/operation', 'read')
4269dfa9f59SPaolo Bonzini        result = self.vm.qmp('query-block-jobs')
4279dfa9f59SPaolo Bonzini        self.assert_qmp(result, 'return[0]/paused', False)
4289dfa9f59SPaolo Bonzini        self.complete_and_wait()
4299dfa9f59SPaolo Bonzini        self.vm.shutdown()
4309dfa9f59SPaolo Bonzini
431b812f671SPaolo Bonzini    def test_large_cluster(self):
432ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
433b812f671SPaolo Bonzini
434b812f671SPaolo Bonzini        # Test COW into the target image.  The first half of the
435b812f671SPaolo Bonzini        # cluster at MIRROR_GRANULARITY has to be copied from
436b812f671SPaolo Bonzini        # backing_img, even though sync='top'.
437b812f671SPaolo Bonzini        qemu_img('create', '-f', iotests.imgfmt, '-ocluster_size=131072,backing_file=%s' %(backing_img), target_img)
438b812f671SPaolo Bonzini        result = self.vm.qmp('drive-mirror', device='drive0', sync='top',
439b812f671SPaolo Bonzini                             on_source_error='ignore',
440b812f671SPaolo Bonzini                             mode='existing', target=target_img)
441b812f671SPaolo Bonzini        self.assert_qmp(result, 'return', {})
442b812f671SPaolo Bonzini
443b812f671SPaolo Bonzini        event = self.vm.get_qmp_event(wait=True)
444b812f671SPaolo Bonzini        self.assertEquals(event['event'], 'BLOCK_JOB_ERROR')
445b812f671SPaolo Bonzini        self.assert_qmp(event, 'data/device', 'drive0')
446b812f671SPaolo Bonzini        self.assert_qmp(event, 'data/operation', 'read')
447b812f671SPaolo Bonzini        result = self.vm.qmp('query-block-jobs')
448b812f671SPaolo Bonzini        self.assert_qmp(result, 'return[0]/paused', False)
449b812f671SPaolo Bonzini        self.complete_and_wait()
450b812f671SPaolo Bonzini        self.vm.shutdown()
451b812f671SPaolo Bonzini
452b812f671SPaolo Bonzini        # Detach blkdebug to compare images successfully
453b812f671SPaolo Bonzini        qemu_img('rebase', '-f', iotests.imgfmt, '-u', '-b', backing_img, test_img)
4543a3918c3SStefan Hajnoczi        self.assertTrue(iotests.compare_images(test_img, target_img),
455b812f671SPaolo Bonzini                        'target image does not match source after mirroring')
456b812f671SPaolo Bonzini
4579dfa9f59SPaolo Bonzini    def test_stop_read(self):
458ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
4599dfa9f59SPaolo Bonzini
4609dfa9f59SPaolo Bonzini        result = self.vm.qmp('drive-mirror', device='drive0', sync='full',
4619dfa9f59SPaolo Bonzini                             target=target_img, on_source_error='stop')
4629dfa9f59SPaolo Bonzini        self.assert_qmp(result, 'return', {})
4639dfa9f59SPaolo Bonzini
4649dfa9f59SPaolo Bonzini        error = False
4659dfa9f59SPaolo Bonzini        ready = False
4669dfa9f59SPaolo Bonzini        while not ready:
4679dfa9f59SPaolo Bonzini            for event in self.vm.get_qmp_events(wait=True):
4689dfa9f59SPaolo Bonzini                if event['event'] == 'BLOCK_JOB_ERROR':
4699dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/device', 'drive0')
4709dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/operation', 'read')
4719dfa9f59SPaolo Bonzini
4729dfa9f59SPaolo Bonzini                    result = self.vm.qmp('query-block-jobs')
4739dfa9f59SPaolo Bonzini                    self.assert_qmp(result, 'return[0]/paused', True)
4749dfa9f59SPaolo Bonzini                    self.assert_qmp(result, 'return[0]/io-status', 'failed')
4759dfa9f59SPaolo Bonzini
4769dfa9f59SPaolo Bonzini                    result = self.vm.qmp('block-job-resume', device='drive0')
4779dfa9f59SPaolo Bonzini                    self.assert_qmp(result, 'return', {})
4789dfa9f59SPaolo Bonzini                    error = True
4799dfa9f59SPaolo Bonzini                elif event['event'] == 'BLOCK_JOB_READY':
4809dfa9f59SPaolo Bonzini                    self.assertTrue(error, 'job completed unexpectedly')
4819dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/device', 'drive0')
4829dfa9f59SPaolo Bonzini                    ready = True
4839dfa9f59SPaolo Bonzini
4849dfa9f59SPaolo Bonzini        result = self.vm.qmp('query-block-jobs')
4859dfa9f59SPaolo Bonzini        self.assert_qmp(result, 'return[0]/paused', False)
4869dfa9f59SPaolo Bonzini        self.assert_qmp(result, 'return[0]/io-status', 'ok')
4879dfa9f59SPaolo Bonzini
4889dfa9f59SPaolo Bonzini        self.complete_and_wait(wait_ready=False)
489ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
4909dfa9f59SPaolo Bonzini        self.vm.shutdown()
4919dfa9f59SPaolo Bonzini
4929dfa9f59SPaolo Bonziniclass TestWriteErrors(ImageMirroringTestCase):
4939dfa9f59SPaolo Bonzini    image_len = 2 * 1024 * 1024 # MB
4949dfa9f59SPaolo Bonzini
4959dfa9f59SPaolo Bonzini    # this should be a multiple of twice the default granularity
4969dfa9f59SPaolo Bonzini    # so that we hit this offset first in state 1
4979dfa9f59SPaolo Bonzini    MIRROR_GRANULARITY = 1024 * 1024
4989dfa9f59SPaolo Bonzini
4999dfa9f59SPaolo Bonzini    def create_blkdebug_file(self, name, event, errno):
5009dfa9f59SPaolo Bonzini        file = open(name, 'w')
5019dfa9f59SPaolo Bonzini        file.write('''
5029dfa9f59SPaolo Bonzini[inject-error]
5039dfa9f59SPaolo Bonzinistate = "1"
5049dfa9f59SPaolo Bonzinievent = "%s"
5059dfa9f59SPaolo Bonzinierrno = "%d"
5069dfa9f59SPaolo Bonziniimmediately = "off"
5079dfa9f59SPaolo Bonzinionce = "on"
5089dfa9f59SPaolo Bonzinisector = "%d"
5099dfa9f59SPaolo Bonzini
5109dfa9f59SPaolo Bonzini[set-state]
5119dfa9f59SPaolo Bonzinistate = "1"
5129dfa9f59SPaolo Bonzinievent = "%s"
5139dfa9f59SPaolo Bonzininew_state = "2"
5149dfa9f59SPaolo Bonzini
5159dfa9f59SPaolo Bonzini[set-state]
5169dfa9f59SPaolo Bonzinistate = "2"
5179dfa9f59SPaolo Bonzinievent = "%s"
5189dfa9f59SPaolo Bonzininew_state = "1"
5199dfa9f59SPaolo Bonzini''' % (event, errno, self.MIRROR_GRANULARITY / 512, event, event))
5209dfa9f59SPaolo Bonzini        file.close()
5219dfa9f59SPaolo Bonzini
5229dfa9f59SPaolo Bonzini    def setUp(self):
5239dfa9f59SPaolo Bonzini        self.blkdebug_file = target_img + ".blkdebug"
5242499a096SStefan Hajnoczi        iotests.create_image(backing_img, TestWriteErrors.image_len)
5259dfa9f59SPaolo Bonzini        self.create_blkdebug_file(self.blkdebug_file, "write_aio", 5)
5269dfa9f59SPaolo Bonzini        qemu_img('create', '-f', iotests.imgfmt, '-obacking_file=%s' %(backing_img), test_img)
5279dfa9f59SPaolo Bonzini        self.vm = iotests.VM().add_drive(test_img)
5289dfa9f59SPaolo Bonzini        self.target_img = 'blkdebug:%s:%s' % (self.blkdebug_file, target_img)
5299dfa9f59SPaolo Bonzini        qemu_img('create', '-f', iotests.imgfmt, '-osize=%d' %(TestWriteErrors.image_len), target_img)
5309dfa9f59SPaolo Bonzini        self.vm.launch()
5319dfa9f59SPaolo Bonzini
5329dfa9f59SPaolo Bonzini    def tearDown(self):
5339dfa9f59SPaolo Bonzini        self.vm.shutdown()
5349dfa9f59SPaolo Bonzini        os.remove(test_img)
5359dfa9f59SPaolo Bonzini        os.remove(backing_img)
5369dfa9f59SPaolo Bonzini        os.remove(self.blkdebug_file)
5379dfa9f59SPaolo Bonzini
5389dfa9f59SPaolo Bonzini    def test_report_write(self):
539ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
5409dfa9f59SPaolo Bonzini
5419dfa9f59SPaolo Bonzini        result = self.vm.qmp('drive-mirror', device='drive0', sync='full',
5429dfa9f59SPaolo Bonzini                             mode='existing', target=self.target_img)
5439dfa9f59SPaolo Bonzini        self.assert_qmp(result, 'return', {})
5449dfa9f59SPaolo Bonzini
5459dfa9f59SPaolo Bonzini        completed = False
5469dfa9f59SPaolo Bonzini        error = False
5479dfa9f59SPaolo Bonzini        while not completed:
5489dfa9f59SPaolo Bonzini            for event in self.vm.get_qmp_events(wait=True):
5499dfa9f59SPaolo Bonzini                if event['event'] == 'BLOCK_JOB_ERROR':
5509dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/device', 'drive0')
5519dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/operation', 'write')
5529dfa9f59SPaolo Bonzini                    error = True
5539dfa9f59SPaolo Bonzini                elif event['event'] == 'BLOCK_JOB_READY':
5549dfa9f59SPaolo Bonzini                    self.assertTrue(False, 'job completed unexpectedly')
5559dfa9f59SPaolo Bonzini                elif event['event'] == 'BLOCK_JOB_COMPLETED':
5569dfa9f59SPaolo Bonzini                    self.assertTrue(error, 'job completed unexpectedly')
5579dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/type', 'mirror')
5589dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/device', 'drive0')
5599dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/error', 'Input/output error')
5609dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/len', self.image_len)
5619dfa9f59SPaolo Bonzini                    completed = True
5629dfa9f59SPaolo Bonzini
563ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
5649dfa9f59SPaolo Bonzini        self.vm.shutdown()
5659dfa9f59SPaolo Bonzini
5669dfa9f59SPaolo Bonzini    def test_ignore_write(self):
567ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
5689dfa9f59SPaolo Bonzini
5699dfa9f59SPaolo Bonzini        result = self.vm.qmp('drive-mirror', device='drive0', sync='full',
5709dfa9f59SPaolo Bonzini                             mode='existing', target=self.target_img,
5719dfa9f59SPaolo Bonzini                             on_target_error='ignore')
5729dfa9f59SPaolo Bonzini        self.assert_qmp(result, 'return', {})
5739dfa9f59SPaolo Bonzini
5749dfa9f59SPaolo Bonzini        event = self.vm.get_qmp_event(wait=True)
5759dfa9f59SPaolo Bonzini        self.assertEquals(event['event'], 'BLOCK_JOB_ERROR')
5769dfa9f59SPaolo Bonzini        self.assert_qmp(event, 'data/device', 'drive0')
5779dfa9f59SPaolo Bonzini        self.assert_qmp(event, 'data/operation', 'write')
5789dfa9f59SPaolo Bonzini        result = self.vm.qmp('query-block-jobs')
5799dfa9f59SPaolo Bonzini        self.assert_qmp(result, 'return[0]/paused', False)
5809dfa9f59SPaolo Bonzini        self.complete_and_wait()
5819dfa9f59SPaolo Bonzini        self.vm.shutdown()
5829dfa9f59SPaolo Bonzini
5839dfa9f59SPaolo Bonzini    def test_stop_write(self):
584ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
5859dfa9f59SPaolo Bonzini
5869dfa9f59SPaolo Bonzini        result = self.vm.qmp('drive-mirror', device='drive0', sync='full',
5879dfa9f59SPaolo Bonzini                             mode='existing', target=self.target_img,
5889dfa9f59SPaolo Bonzini                             on_target_error='stop')
5899dfa9f59SPaolo Bonzini        self.assert_qmp(result, 'return', {})
5909dfa9f59SPaolo Bonzini
5919dfa9f59SPaolo Bonzini        error = False
5929dfa9f59SPaolo Bonzini        ready = False
5939dfa9f59SPaolo Bonzini        while not ready:
5949dfa9f59SPaolo Bonzini            for event in self.vm.get_qmp_events(wait=True):
5959dfa9f59SPaolo Bonzini                if event['event'] == 'BLOCK_JOB_ERROR':
5969dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/device', 'drive0')
5979dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/operation', 'write')
5989dfa9f59SPaolo Bonzini
5999dfa9f59SPaolo Bonzini                    result = self.vm.qmp('query-block-jobs')
6009dfa9f59SPaolo Bonzini                    self.assert_qmp(result, 'return[0]/paused', True)
6019dfa9f59SPaolo Bonzini                    self.assert_qmp(result, 'return[0]/io-status', 'failed')
6029dfa9f59SPaolo Bonzini
6039dfa9f59SPaolo Bonzini                    result = self.vm.qmp('block-job-resume', device='drive0')
6049dfa9f59SPaolo Bonzini                    self.assert_qmp(result, 'return', {})
6059dfa9f59SPaolo Bonzini
6069dfa9f59SPaolo Bonzini                    result = self.vm.qmp('query-block-jobs')
6079dfa9f59SPaolo Bonzini                    self.assert_qmp(result, 'return[0]/paused', False)
6089dfa9f59SPaolo Bonzini                    self.assert_qmp(result, 'return[0]/io-status', 'ok')
6099dfa9f59SPaolo Bonzini                    error = True
6109dfa9f59SPaolo Bonzini                elif event['event'] == 'BLOCK_JOB_READY':
6119dfa9f59SPaolo Bonzini                    self.assertTrue(error, 'job completed unexpectedly')
6129dfa9f59SPaolo Bonzini                    self.assert_qmp(event, 'data/device', 'drive0')
6139dfa9f59SPaolo Bonzini                    ready = True
6149dfa9f59SPaolo Bonzini
6159dfa9f59SPaolo Bonzini        self.complete_and_wait(wait_ready=False)
616ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
6179dfa9f59SPaolo Bonzini        self.vm.shutdown()
6189dfa9f59SPaolo Bonzini
61944c7ca5eSPaolo Bonziniclass TestSetSpeed(ImageMirroringTestCase):
62044c7ca5eSPaolo Bonzini    image_len = 80 * 1024 * 1024 # MB
62144c7ca5eSPaolo Bonzini
62244c7ca5eSPaolo Bonzini    def setUp(self):
62344c7ca5eSPaolo Bonzini        qemu_img('create', backing_img, str(TestSetSpeed.image_len))
62444c7ca5eSPaolo Bonzini        qemu_img('create', '-f', iotests.imgfmt, '-o', 'backing_file=%s' % backing_img, test_img)
62544c7ca5eSPaolo Bonzini        self.vm = iotests.VM().add_drive(test_img)
62644c7ca5eSPaolo Bonzini        self.vm.launch()
62744c7ca5eSPaolo Bonzini
62844c7ca5eSPaolo Bonzini    def tearDown(self):
62944c7ca5eSPaolo Bonzini        self.vm.shutdown()
63044c7ca5eSPaolo Bonzini        os.remove(test_img)
63144c7ca5eSPaolo Bonzini        os.remove(backing_img)
63244c7ca5eSPaolo Bonzini        os.remove(target_img)
63344c7ca5eSPaolo Bonzini
63444c7ca5eSPaolo Bonzini    def test_set_speed(self):
635ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
63644c7ca5eSPaolo Bonzini
63744c7ca5eSPaolo Bonzini        result = self.vm.qmp('drive-mirror', device='drive0', sync='full',
63844c7ca5eSPaolo Bonzini                             target=target_img)
63944c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'return', {})
64044c7ca5eSPaolo Bonzini
64144c7ca5eSPaolo Bonzini        # Default speed is 0
64244c7ca5eSPaolo Bonzini        result = self.vm.qmp('query-block-jobs')
64344c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'return[0]/device', 'drive0')
64444c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'return[0]/speed', 0)
64544c7ca5eSPaolo Bonzini
64644c7ca5eSPaolo Bonzini        result = self.vm.qmp('block-job-set-speed', device='drive0', speed=8 * 1024 * 1024)
64744c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'return', {})
64844c7ca5eSPaolo Bonzini
64944c7ca5eSPaolo Bonzini        # Ensure the speed we set was accepted
65044c7ca5eSPaolo Bonzini        result = self.vm.qmp('query-block-jobs')
65144c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'return[0]/device', 'drive0')
65244c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'return[0]/speed', 8 * 1024 * 1024)
65344c7ca5eSPaolo Bonzini
6542575fe16SStefan Hajnoczi        self.wait_ready_and_cancel()
65544c7ca5eSPaolo Bonzini
65644c7ca5eSPaolo Bonzini        # Check setting speed in drive-mirror works
65744c7ca5eSPaolo Bonzini        result = self.vm.qmp('drive-mirror', device='drive0', sync='full',
65844c7ca5eSPaolo Bonzini                             target=target_img, speed=4*1024*1024)
65944c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'return', {})
66044c7ca5eSPaolo Bonzini
66144c7ca5eSPaolo Bonzini        result = self.vm.qmp('query-block-jobs')
66244c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'return[0]/device', 'drive0')
66344c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'return[0]/speed', 4 * 1024 * 1024)
66444c7ca5eSPaolo Bonzini
6652575fe16SStefan Hajnoczi        self.wait_ready_and_cancel()
66644c7ca5eSPaolo Bonzini
66744c7ca5eSPaolo Bonzini    def test_set_speed_invalid(self):
668ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
66944c7ca5eSPaolo Bonzini
67044c7ca5eSPaolo Bonzini        result = self.vm.qmp('drive-mirror', device='drive0', sync='full',
67144c7ca5eSPaolo Bonzini                             target=target_img, speed=-1)
67244c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'error/class', 'GenericError')
67344c7ca5eSPaolo Bonzini
674ecc1c88eSStefan Hajnoczi        self.assert_no_active_block_jobs()
67544c7ca5eSPaolo Bonzini
67644c7ca5eSPaolo Bonzini        result = self.vm.qmp('drive-mirror', device='drive0', sync='full',
67744c7ca5eSPaolo Bonzini                             target=target_img)
67844c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'return', {})
67944c7ca5eSPaolo Bonzini
68044c7ca5eSPaolo Bonzini        result = self.vm.qmp('block-job-set-speed', device='drive0', speed=-1)
68144c7ca5eSPaolo Bonzini        self.assert_qmp(result, 'error/class', 'GenericError')
68244c7ca5eSPaolo Bonzini
6832575fe16SStefan Hajnoczi        self.wait_ready_and_cancel()
68444c7ca5eSPaolo Bonzini
685c15badeeSMax Reitzclass TestUnbackedSource(ImageMirroringTestCase):
686c15badeeSMax Reitz    image_len = 2 * 1024 * 1024 # MB
687c15badeeSMax Reitz
688c15badeeSMax Reitz    def setUp(self):
689c15badeeSMax Reitz        qemu_img('create', '-f', iotests.imgfmt, test_img,
690c15badeeSMax Reitz                 str(TestUnbackedSource.image_len))
691c15badeeSMax Reitz        self.vm = iotests.VM().add_drive(test_img)
692c15badeeSMax Reitz        self.vm.launch()
693c15badeeSMax Reitz
694c15badeeSMax Reitz    def tearDown(self):
695c15badeeSMax Reitz        self.vm.shutdown()
696c15badeeSMax Reitz        os.remove(test_img)
697c15badeeSMax Reitz        os.remove(target_img)
698c15badeeSMax Reitz
699171d6431SMax Reitz    def test_absolute_paths_full(self):
700171d6431SMax Reitz        self.assert_no_active_block_jobs()
701171d6431SMax Reitz        result = self.vm.qmp('drive-mirror', device='drive0',
702171d6431SMax Reitz                             sync='full', target=target_img,
703171d6431SMax Reitz                             mode='absolute-paths')
704171d6431SMax Reitz        self.assert_qmp(result, 'return', {})
705171d6431SMax Reitz        self.complete_and_wait()
706c15badeeSMax Reitz        self.assert_no_active_block_jobs()
707c15badeeSMax Reitz
708171d6431SMax Reitz    def test_absolute_paths_top(self):
709171d6431SMax Reitz        self.assert_no_active_block_jobs()
710c15badeeSMax Reitz        result = self.vm.qmp('drive-mirror', device='drive0',
711171d6431SMax Reitz                             sync='top', target=target_img,
712171d6431SMax Reitz                             mode='absolute-paths')
713171d6431SMax Reitz        self.assert_qmp(result, 'return', {})
714171d6431SMax Reitz        self.complete_and_wait()
715171d6431SMax Reitz        self.assert_no_active_block_jobs()
716171d6431SMax Reitz
717171d6431SMax Reitz    def test_absolute_paths_none(self):
718171d6431SMax Reitz        self.assert_no_active_block_jobs()
719171d6431SMax Reitz        result = self.vm.qmp('drive-mirror', device='drive0',
720171d6431SMax Reitz                             sync='none', target=target_img,
721c15badeeSMax Reitz                             mode='absolute-paths')
722c15badeeSMax Reitz        self.assert_qmp(result, 'return', {})
723c15badeeSMax Reitz        self.complete_and_wait()
724c15badeeSMax Reitz        self.assert_no_active_block_jobs()
725c15badeeSMax Reitz
72644c7ca5eSPaolo Bonziniif __name__ == '__main__':
72744c7ca5eSPaolo Bonzini    iotests.main(supported_fmts=['qcow2', 'qed'])
728