xref: /openbmc/qemu/tests/qemu-iotests/041 (revision c15badee200f9e743e3639cc5e3e7dd62c69332f)
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