[qemu.git] / tests / qemu-iotests / 155

#!/usr/bin/env python
#
# Test whether the backing BDSs are correct after completion of a
# mirror block job; in "existing" modes (drive-mirror with
# mode=existing and blockdev-mirror) the backing chain should not be
# overridden.
#
# Copyright (C) 2016 Red Hat, Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#

import os
import iotests
from iotests import qemu_img

back0_img = os.path.join(iotests.test_dir, 'back0.' + iotests.imgfmt)
back1_img = os.path.join(iotests.test_dir, 'back1.' + iotests.imgfmt)
back2_img = os.path.join(iotests.test_dir, 'back2.' + iotests.imgfmt)
source_img = os.path.join(iotests.test_dir, 'source.' + iotests.imgfmt)
target_img = os.path.join(iotests.test_dir, 'target.' + iotests.imgfmt)


# Class variables for controlling its behavior:
#
# existing: If True, explicitly create the target image and blockdev-add it
# target_backing: If existing is True: Use this filename as the backing file
#                 of the target image
#                 (None: no backing file)
# target_blockdev_backing: If existing is True: Pass this dict as "backing"
#                          for the blockdev-add command
#                          (None: do not pass "backing")
# target_real_backing: If existing is True: The real filename of the backing
#                      image during runtime, only makes sense if
#                      target_blockdev_backing is not None
#                      (None: same as target_backing)

class BaseClass(iotests.QMPTestCase):
    target_blockdev_backing = None
    target_real_backing = None

    def setUp(self):
        qemu_img('create', '-f', iotests.imgfmt, back0_img, '1M')
        qemu_img('create', '-f', iotests.imgfmt, '-b', back0_img, back1_img)
        qemu_img('create', '-f', iotests.imgfmt, '-b', back1_img, back2_img)
        qemu_img('create', '-f', iotests.imgfmt, '-b', back2_img, source_img)

        self.vm = iotests.VM()
        self.vm.add_drive(None, '', 'none')
        self.vm.launch()

        # Add the BDS via blockdev-add so it stays around after the mirror block
        # job has been completed
        result = self.vm.qmp('blockdev-add',
                             node_name='source',
                             driver=iotests.imgfmt,
                             file={'driver': 'file',
                                   'filename': source_img})
        self.assert_qmp(result, 'return', {})

        result = self.vm.qmp('x-blockdev-insert-medium',
                             device='drive0', node_name='source')
        self.assert_qmp(result, 'return', {})

        self.assertIntactSourceBackingChain()

        if self.existing:
            if self.target_backing:
                qemu_img('create', '-f', iotests.imgfmt,
                         '-b', self.target_backing, target_img, '1M')
            else:
                qemu_img('create', '-f', iotests.imgfmt, target_img, '1M')

            if self.cmd == 'blockdev-mirror':
                options = { 'node-name': 'target',
                            'driver': iotests.imgfmt,
                            'file': { 'driver': 'file',
                                      'filename': target_img } }
                if self.target_blockdev_backing:
                    options['backing'] = self.target_blockdev_backing

                result = self.vm.qmp('blockdev-add', **options)
                self.assert_qmp(result, 'return', {})

    def tearDown(self):
        self.vm.shutdown()
        os.remove(source_img)
        os.remove(back2_img)
        os.remove(back1_img)
        os.remove(back0_img)
        try:
            os.remove(target_img)
        except OSError:
            pass

    def findBlockNode(self, node_name, id=None):
        if id:
            result = self.vm.qmp('query-block')
            for device in result['return']:
                if device['device'] == id:
                    if node_name:
                        self.assert_qmp(device, 'inserted/node-name', node_name)
                    return device['inserted']
        else:
            result = self.vm.qmp('query-named-block-nodes')
            for node in result['return']:
                if node['node-name'] == node_name:
                    return node

        self.fail('Cannot find node %s/%s' % (id, node_name))

    def assertIntactSourceBackingChain(self):
        node = self.findBlockNode('source')

        self.assert_qmp(node, 'image' + '/backing-image' * 0 + '/filename',
                        source_img)
        self.assert_qmp(node, 'image' + '/backing-image' * 1 + '/filename',
                        back2_img)
        self.assert_qmp(node, 'image' + '/backing-image' * 2 + '/filename',
                        back1_img)
        self.assert_qmp(node, 'image' + '/backing-image' * 3 + '/filename',
                        back0_img)
        self.assert_qmp_absent(node, 'image' + '/backing-image' * 4)

    def assertCorrectBackingImage(self, node, default_image):
        if self.existing:
            if self.target_real_backing:
                image = self.target_real_backing
            else:
                image = self.target_backing
        else:
            image = default_image

        if image:
            self.assert_qmp(node, 'image/backing-image/filename', image)
        else:
            self.assert_qmp_absent(node, 'image/backing-image')


# Class variables for controlling its behavior:
#
# cmd: Mirroring command to execute, either drive-mirror or blockdev-mirror

class MirrorBaseClass(BaseClass):
    def runMirror(self, sync):
        if self.cmd == 'blockdev-mirror':
            result = self.vm.qmp(self.cmd, device='drive0', sync=sync,
                                 target='target')
        else:
            if self.existing:
                mode = 'existing'
            else:
                mode = 'absolute-paths'
            result = self.vm.qmp(self.cmd, device='drive0', sync=sync,
                                 target=target_img, format=iotests.imgfmt,
                                 mode=mode, node_name='target')

        self.assert_qmp(result, 'return', {})

        self.vm.event_wait('BLOCK_JOB_READY')

        result = self.vm.qmp('block-job-complete', device='drive0')
        self.assert_qmp(result, 'return', {})

        self.vm.event_wait('BLOCK_JOB_COMPLETED')

    def testFull(self):
        self.runMirror('full')

        node = self.findBlockNode('target', 'drive0')
        self.assertCorrectBackingImage(node, None)
        self.assertIntactSourceBackingChain()

    def testTop(self):
        self.runMirror('top')

        node = self.findBlockNode('target', 'drive0')
        self.assertCorrectBackingImage(node, back2_img)
        self.assertIntactSourceBackingChain()

    def testNone(self):
        self.runMirror('none')

        node = self.findBlockNode('target', 'drive0')
        self.assertCorrectBackingImage(node, source_img)
        self.assertIntactSourceBackingChain()


class TestDriveMirrorAbsolutePaths(MirrorBaseClass):
    cmd = 'drive-mirror'
    existing = False

class TestDriveMirrorExistingNoBacking(MirrorBaseClass):
    cmd = 'drive-mirror'
    existing = True
    target_backing = None

class TestDriveMirrorExistingBacking(MirrorBaseClass):
    cmd = 'drive-mirror'
    existing = True
    target_backing = 'null-co://'

class TestBlockdevMirrorNoBacking(MirrorBaseClass):
    cmd = 'blockdev-mirror'
    existing = True
    target_backing = None

class TestBlockdevMirrorBacking(MirrorBaseClass):
    cmd = 'blockdev-mirror'
    existing = True
    target_backing = 'null-co://'

class TestBlockdevMirrorForcedBacking(MirrorBaseClass):
    cmd = 'blockdev-mirror'
    existing = True
    target_backing = None
    target_blockdev_backing = { 'driver': 'null-co' }
    target_real_backing = 'null-co://'


class TestCommit(BaseClass):
    existing = False

    def testCommit(self):
        result = self.vm.qmp('block-commit', device='drive0', base=back1_img)
        self.assert_qmp(result, 'return', {})

        self.vm.event_wait('BLOCK_JOB_READY')

        result = self.vm.qmp('block-job-complete', device='drive0')
        self.assert_qmp(result, 'return', {})

        self.vm.event_wait('BLOCK_JOB_COMPLETED')

        node = self.findBlockNode(None, 'drive0')
        self.assert_qmp(node, 'image' + '/backing-image' * 0 + '/filename',
                        back1_img)
        self.assert_qmp(node, 'image' + '/backing-image' * 1 + '/filename',
                        back0_img)
        self.assert_qmp_absent(node, 'image' + '/backing-image' * 2 +
                               '/filename')

        self.assertIntactSourceBackingChain()


BaseClass = None
MirrorBaseClass = None

if __name__ == '__main__':
    iotests.main(supported_fmts=['qcow2'])
Commit	Line	Data
298c6009 HR	1	#!/usr/bin/env python
	2	#
	3	# Test whether the backing BDSs are correct after completion of a
	4	# mirror block job; in "existing" modes (drive-mirror with
	5	# mode=existing and blockdev-mirror) the backing chain should not be
	6	# overridden.
	7	#
	8	# Copyright (C) 2016 Red Hat, Inc.
	9	#
	10	# This program is free software; you can redistribute it and/or modify
	11	# it under the terms of the GNU General Public License as published by
	12	# the Free Software Foundation; either version 2 of the License, or
	13	# (at your option) any later version.
	14	#
	15	# This program is distributed in the hope that it will be useful,
	16	# but WITHOUT ANY WARRANTY; without even the implied warranty of
	17	# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	18	# GNU General Public License for more details.
	19	#
	20	# You should have received a copy of the GNU General Public License
	21	# along with this program. If not, see <http://www.gnu.org/licenses/>.
	22	#
	23
	24	import os
	25	import iotests
	26	from iotests import qemu_img
	27
	28	back0_img = os.path.join(iotests.test_dir, 'back0.' + iotests.imgfmt)
	29	back1_img = os.path.join(iotests.test_dir, 'back1.' + iotests.imgfmt)
	30	back2_img = os.path.join(iotests.test_dir, 'back2.' + iotests.imgfmt)
	31	source_img = os.path.join(iotests.test_dir, 'source.' + iotests.imgfmt)
	32	target_img = os.path.join(iotests.test_dir, 'target.' + iotests.imgfmt)
	33
	34
	35	# Class variables for controlling its behavior:
	36	#
	37	# existing: If True, explicitly create the target image and blockdev-add it
	38	# target_backing: If existing is True: Use this filename as the backing file
	39	# of the target image
	40	# (None: no backing file)
	41	# target_blockdev_backing: If existing is True: Pass this dict as "backing"
	42	# for the blockdev-add command
	43	# (None: do not pass "backing")
	44	# target_real_backing: If existing is True: The real filename of the backing
	45	# image during runtime, only makes sense if
	46	# target_blockdev_backing is not None
	47	# (None: same as target_backing)
	48
	49	class BaseClass(iotests.QMPTestCase):
	50	target_blockdev_backing = None
	51	target_real_backing = None
	52
	53	def setUp(self):
	54	qemu_img('create', '-f', iotests.imgfmt, back0_img, '1M')
	55	qemu_img('create', '-f', iotests.imgfmt, '-b', back0_img, back1_img)
	56	qemu_img('create', '-f', iotests.imgfmt, '-b', back1_img, back2_img)
	57	qemu_img('create', '-f', iotests.imgfmt, '-b', back2_img, source_img)
	58
	59	self.vm = iotests.VM()
	60	self.vm.add_drive(None, '', 'none')
	61	self.vm.launch()
	62
	63	# Add the BDS via blockdev-add so it stays around after the mirror block
	64	# job has been completed
65	result = self.vm.qmp('blockdev-add',
0153d2f5 KW	66	node_name='source',
	67	driver=iotests.imgfmt,
	68	file={'driver': 'file',
	69	'filename': source_img})
298c6009 HR	70	self.assert_qmp(result, 'return', {})
	71
	72	result = self.vm.qmp('x-blockdev-insert-medium',
	73	device='drive0', node_name='source')
	74	self.assert_qmp(result, 'return', {})
	75
	76	self.assertIntactSourceBackingChain()
	77
	78	if self.existing:
	79	if self.target_backing:
	80	qemu_img('create', '-f', iotests.imgfmt,
	81	'-b', self.target_backing, target_img, '1M')
	82	else:
	83	qemu_img('create', '-f', iotests.imgfmt, target_img, '1M')
	84
	85	if self.cmd == 'blockdev-mirror':
	86	options = { 'node-name': 'target',
	87	'driver': iotests.imgfmt,
	88	'file': { 'driver': 'file',
	89	'filename': target_img } }
	90	if self.target_blockdev_backing:
	91	options['backing'] = self.target_blockdev_backing
	92
0153d2f5	93	result = self.vm.qmp('blockdev-add', **options)
298c6009 HR	94	self.assert_qmp(result, 'return', {})
	95
	96	def tearDown(self):
	97	self.vm.shutdown()
	98	os.remove(source_img)
	99	os.remove(back2_img)
	100	os.remove(back1_img)
	101	os.remove(back0_img)
	102	try:
	103	os.remove(target_img)
	104	except OSError:
	105	pass
	106
	107	def findBlockNode(self, node_name, id=None):
	108	if id:
	109	result = self.vm.qmp('query-block')
	110	for device in result['return']:
	111	if device['device'] == id:
	112	if node_name:
	113	self.assert_qmp(device, 'inserted/node-name', node_name)
	114	return device['inserted']
	115	else:
	116	result = self.vm.qmp('query-named-block-nodes')
	117	for node in result['return']:
	118	if node['node-name'] == node_name:
	119	return node
	120
	121	self.fail('Cannot find node %s/%s' % (id, node_name))
	122
	123	def assertIntactSourceBackingChain(self):
	124	node = self.findBlockNode('source')
	125
	126	self.assert_qmp(node, 'image' + '/backing-image' * 0 + '/filename',
	127	source_img)
	128	self.assert_qmp(node, 'image' + '/backing-image' * 1 + '/filename',
	129	back2_img)
	130	self.assert_qmp(node, 'image' + '/backing-image' * 2 + '/filename',
	131	back1_img)
	132	self.assert_qmp(node, 'image' + '/backing-image' * 3 + '/filename',
	133	back0_img)
	134	self.assert_qmp_absent(node, 'image' + '/backing-image' * 4)
	135
	136	def assertCorrectBackingImage(self, node, default_image):
	137	if self.existing:
	138	if self.target_real_backing:
	139	image = self.target_real_backing
	140	else:
	141	image = self.target_backing
	142	else:
	143	image = default_image
	144
	145	if image:
	146	self.assert_qmp(node, 'image/backing-image/filename', image)
	147	else:
	148	self.assert_qmp_absent(node, 'image/backing-image')
	149
	150
	151	# Class variables for controlling its behavior:
	152	#
	153	# cmd: Mirroring command to execute, either drive-mirror or blockdev-mirror
	154
	155	class MirrorBaseClass(BaseClass):
	156	def runMirror(self, sync):
	157	if self.cmd == 'blockdev-mirror':
158	result = self.vm.qmp(self.cmd, device='drive0', sync=sync,
159	target='target')
160	else:
161	if self.existing:
162	mode = 'existing'
163	else:
164	mode = 'absolute-paths'
165	result = self.vm.qmp(self.cmd, device='drive0', sync=sync,
166	target=target_img, format=iotests.imgfmt,
167	mode=mode, node_name='target')
168
169	self.assert_qmp(result, 'return', {})
170
171	self.vm.event_wait('BLOCK_JOB_READY')
172
173	result = self.vm.qmp('block-job-complete', device='drive0')
174	self.assert_qmp(result, 'return', {})
175
176	self.vm.event_wait('BLOCK_JOB_COMPLETED')
177
178	def testFull(self):
179	self.runMirror('full')
180
181	node = self.findBlockNode('target', 'drive0')
182	self.assertCorrectBackingImage(node, None)
183	self.assertIntactSourceBackingChain()
184
185	def testTop(self):
186	self.runMirror('top')
187
188	node = self.findBlockNode('target', 'drive0')
189	self.assertCorrectBackingImage(node, back2_img)
190	self.assertIntactSourceBackingChain()
191
192	def testNone(self):
193	self.runMirror('none')
194
195	node = self.findBlockNode('target', 'drive0')
196	self.assertCorrectBackingImage(node, source_img)
197	self.assertIntactSourceBackingChain()
198
199
200	class TestDriveMirrorAbsolutePaths(MirrorBaseClass):
201	cmd = 'drive-mirror'
202	existing = False
203
204	class TestDriveMirrorExistingNoBacking(MirrorBaseClass):
205	cmd = 'drive-mirror'
206	existing = True
207	target_backing = None
208
209	class TestDriveMirrorExistingBacking(MirrorBaseClass):
210	cmd = 'drive-mirror'
211	existing = True
212	target_backing = 'null-co://'
213
214	class TestBlockdevMirrorNoBacking(MirrorBaseClass):
215	cmd = 'blockdev-mirror'
216	existing = True
217	target_backing = None
218
219	class TestBlockdevMirrorBacking(MirrorBaseClass):
220	cmd = 'blockdev-mirror'
221	existing = True
222	target_backing = 'null-co://'
223
224	class TestBlockdevMirrorForcedBacking(MirrorBaseClass):
225	cmd = 'blockdev-mirror'
226	existing = True
227	target_backing = None
228	target_blockdev_backing = { 'driver': 'null-co' }
229	target_real_backing = 'null-co://'
230
231
232	class TestCommit(BaseClass):
233	existing = False
234
235	def testCommit(self):
236	result = self.vm.qmp('block-commit', device='drive0', base=back1_img)
237	self.assert_qmp(result, 'return', {})
238
239	self.vm.event_wait('BLOCK_JOB_READY')
240
241	result = self.vm.qmp('block-job-complete', device='drive0')
242	self.assert_qmp(result, 'return', {})
243
244	self.vm.event_wait('BLOCK_JOB_COMPLETED')
245
246	node = self.findBlockNode(None, 'drive0')
247	self.assert_qmp(node, 'image' + '/backing-image' * 0 + '/filename',
248	back1_img)
249	self.assert_qmp(node, 'image' + '/backing-image' * 1 + '/filename',
250	back0_img)
251	self.assert_qmp_absent(node, 'image' + '/backing-image' * 2 +
252	'/filename')
253
254	self.assertIntactSourceBackingChain()
255
256
257	BaseClass = None
258	MirrorBaseClass = None
259
260	if __name__ == '__main__':
261	iotests.main(supported_fmts=['qcow2'])