3 # Tests for image streaming.
5 # Copyright (C) 2012 IBM Corp.
7 # This program is free software; you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation; either version 2 of the License, or
10 # (at your option) any later version.
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU General Public License for more details.
17 # You should have received a copy of the GNU General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
24 from iotests
import qemu_img
, qemu_io
26 backing_img
= os
.path
.join(iotests
.test_dir
, 'backing.img')
27 mid_img
= os
.path
.join(iotests
.test_dir
, 'mid.img')
28 test_img
= os
.path
.join(iotests
.test_dir
, 'test.img')
30 class TestSingleDrive(iotests
.QMPTestCase
):
31 image_len
= 1 * 1024 * 1024 # MB
34 iotests
.create_image(backing_img
, TestSingleDrive
.image_len
)
35 qemu_img('create', '-f', iotests
.imgfmt
, '-o', 'backing_file=%s' % backing_img
, mid_img
)
36 qemu_img('create', '-f', iotests
.imgfmt
, '-o', 'backing_file=%s' % mid_img
, test_img
)
37 qemu_io('-f', 'raw', '-c', 'write -P 0x1 0 512', backing_img
)
38 qemu_io('-f', iotests
.imgfmt
, '-c', 'write -P 0x1 524288 512', mid_img
)
39 self
.vm
= iotests
.VM().add_drive("blkdebug::" + test_img
, "backing.node-name=mid")
46 os
.remove(backing_img
)
48 def test_stream(self
):
49 self
.assert_no_active_block_jobs()
51 result
= self
.vm
.qmp('block-stream', device
='drive0')
52 self
.assert_qmp(result
, 'return', {})
54 self
.wait_until_completed()
56 self
.assert_no_active_block_jobs()
59 self
.assertEqual(qemu_io('-f', 'raw', '-c', 'map', backing_img
),
60 qemu_io('-f', iotests
.imgfmt
, '-c', 'map', test_img
),
61 'image file map does not match backing file after streaming')
63 def test_stream_intermediate(self
):
64 self
.assert_no_active_block_jobs()
66 self
.assertNotEqual(qemu_io('-f', 'raw', '-c', 'map', backing_img
),
67 qemu_io('-f', iotests
.imgfmt
, '-c', 'map', mid_img
),
68 'image file map matches backing file before streaming')
70 result
= self
.vm
.qmp('block-stream', device
='mid', job_id
='stream-mid')
71 self
.assert_qmp(result
, 'return', {})
73 self
.wait_until_completed(drive
='stream-mid')
75 self
.assert_no_active_block_jobs()
78 self
.assertEqual(qemu_io('-f', 'raw', '-c', 'map', backing_img
),
79 qemu_io('-f', iotests
.imgfmt
, '-c', 'map', mid_img
),
80 'image file map does not match backing file after streaming')
82 def test_stream_pause(self
):
83 self
.assert_no_active_block_jobs()
85 self
.vm
.pause_drive('drive0')
86 result
= self
.vm
.qmp('block-stream', device
='drive0')
87 self
.assert_qmp(result
, 'return', {})
89 result
= self
.vm
.qmp('block-job-pause', device
='drive0')
90 self
.assert_qmp(result
, 'return', {})
93 result
= self
.vm
.qmp('query-block-jobs')
94 offset
= self
.dictpath(result
, 'return[0]/offset')
97 result
= self
.vm
.qmp('query-block-jobs')
98 self
.assert_qmp(result
, 'return[0]/offset', offset
)
100 result
= self
.vm
.qmp('block-job-resume', device
='drive0')
101 self
.assert_qmp(result
, 'return', {})
103 self
.vm
.resume_drive('drive0')
104 self
.wait_until_completed()
106 self
.assert_no_active_block_jobs()
109 self
.assertEqual(qemu_io('-f', 'raw', '-c', 'map', backing_img
),
110 qemu_io('-f', iotests
.imgfmt
, '-c', 'map', test_img
),
111 'image file map does not match backing file after streaming')
113 def test_stream_no_op(self
):
114 self
.assert_no_active_block_jobs()
116 # The image map is empty before the operation
117 empty_map
= qemu_io('-f', iotests
.imgfmt
, '-c', 'map', test_img
)
119 # This is a no-op: no data should ever be copied from the base image
120 result
= self
.vm
.qmp('block-stream', device
='drive0', base
=mid_img
)
121 self
.assert_qmp(result
, 'return', {})
123 self
.wait_until_completed()
125 self
.assert_no_active_block_jobs()
128 self
.assertEqual(qemu_io('-f', iotests
.imgfmt
, '-c', 'map', test_img
),
129 empty_map
, 'image file map changed after a no-op')
131 def test_stream_partial(self
):
132 self
.assert_no_active_block_jobs()
134 result
= self
.vm
.qmp('block-stream', device
='drive0', base
=backing_img
)
135 self
.assert_qmp(result
, 'return', {})
137 self
.wait_until_completed()
139 self
.assert_no_active_block_jobs()
142 self
.assertEqual(qemu_io('-f', iotests
.imgfmt
, '-c', 'map', mid_img
),
143 qemu_io('-f', iotests
.imgfmt
, '-c', 'map', test_img
),
144 'image file map does not match backing file after streaming')
146 def test_device_not_found(self
):
147 result
= self
.vm
.qmp('block-stream', device
='nonexistent')
148 self
.assert_qmp(result
, 'error/class', 'GenericError')
151 class TestParallelOps(iotests
.QMPTestCase
):
152 num_ops
= 4 # Number of parallel block-stream operations
153 num_imgs
= num_ops
* 2 + 1
154 image_len
= num_ops
* 1024 * 1024
161 # Initialize file names and command-line options
162 for i
in range(self
.num_imgs
):
163 img_depth
= self
.num_imgs
- i
- 1
164 opts
.append("backing." * img_depth
+ "node-name=node%d" % i
)
165 self
.imgs
.append(os
.path
.join(iotests
.test_dir
, 'img-%d.img' % i
))
168 iotests
.create_image(self
.imgs
[0], self
.image_len
)
169 for i
in range(1, self
.num_imgs
):
170 qemu_img('create', '-f', iotests
.imgfmt
,
171 '-o', 'backing_file=%s' % self
.imgs
[i
-1], self
.imgs
[i
])
173 # Put data into the images we are copying data from
174 for i
in range(self
.num_imgs
/ 2):
175 img_index
= i
* 2 + 1
176 # Alternate between 512k and 1M.
177 # This way jobs will not finish in the same order they were created
178 num_kb
= 512 + 512 * (i
% 2)
179 qemu_io('-f', iotests
.imgfmt
,
180 '-c', 'write -P %d %d %d' % (i
, i
*1024*1024, num_kb
* 1024),
181 self
.imgs
[img_index
])
183 # Attach the drive to the VM
184 self
.vm
= iotests
.VM()
185 self
.vm
.add_drive(self
.imgs
[-1], ','.join(opts
))
190 for img
in self
.imgs
:
193 # Test that it's possible to run several block-stream operations
194 # in parallel in the same snapshot chain
195 def test_stream_parallel(self
):
196 self
.assert_no_active_block_jobs()
198 # Check that the maps don't match before the streaming operations
199 for i
in range(2, self
.num_imgs
, 2):
200 self
.assertNotEqual(qemu_io('-f', iotests
.imgfmt
, '-c', 'map', self
.imgs
[i
]),
201 qemu_io('-f', iotests
.imgfmt
, '-c', 'map', self
.imgs
[i
-1]),
202 'image file map matches backing file before streaming')
204 # Create all streaming jobs
206 for i
in range(2, self
.num_imgs
, 2):
207 node_name
= 'node%d' % i
208 job_id
= 'stream-%s' % node_name
209 pending_jobs
.append(job_id
)
210 result
= self
.vm
.qmp('block-stream', device
=node_name
, job_id
=job_id
, base
=self
.imgs
[i
-2], speed
=512*1024)
211 self
.assert_qmp(result
, 'return', {})
213 # Wait for all jobs to be finished.
214 while len(pending_jobs
) > 0:
215 for event
in self
.vm
.get_qmp_events(wait
=True):
216 if event
['event'] == 'BLOCK_JOB_COMPLETED':
217 job_id
= self
.dictpath(event
, 'data/device')
218 self
.assertTrue(job_id
in pending_jobs
)
219 self
.assert_qmp_absent(event
, 'data/error')
220 pending_jobs
.remove(job_id
)
222 self
.assert_no_active_block_jobs()
225 # Check that all maps match now
226 for i
in range(2, self
.num_imgs
, 2):
227 self
.assertEqual(qemu_io('-f', iotests
.imgfmt
, '-c', 'map', self
.imgs
[i
]),
228 qemu_io('-f', iotests
.imgfmt
, '-c', 'map', self
.imgs
[i
-1]),
229 'image file map does not match backing file after streaming')
231 # Test that it's not possible to perform two block-stream
232 # operations if there are nodes involved in both.
233 def test_overlapping_1(self
):
234 self
.assert_no_active_block_jobs()
236 # Set a speed limit to make sure that this job blocks the rest
237 result
= self
.vm
.qmp('block-stream', device
='node4', job_id
='stream-node4', base
=self
.imgs
[1], speed
=1024*1024)
238 self
.assert_qmp(result
, 'return', {})
240 result
= self
.vm
.qmp('block-stream', device
='node5', job_id
='stream-node5', base
=self
.imgs
[2])
241 self
.assert_qmp(result
, 'error/class', 'GenericError')
243 result
= self
.vm
.qmp('block-stream', device
='node3', job_id
='stream-node3', base
=self
.imgs
[2])
244 self
.assert_qmp(result
, 'error/class', 'GenericError')
246 result
= self
.vm
.qmp('block-stream', device
='node4', job_id
='stream-node4-v2')
247 self
.assert_qmp(result
, 'error/class', 'GenericError')
249 # block-commit should also fail if it touches nodes used by the stream job
250 result
= self
.vm
.qmp('block-commit', device
='drive0', base
=self
.imgs
[4], job_id
='commit-node4')
251 self
.assert_qmp(result
, 'error/class', 'GenericError')
253 result
= self
.vm
.qmp('block-commit', device
='drive0', base
=self
.imgs
[1], top
=self
.imgs
[3], job_id
='commit-node1')
254 self
.assert_qmp(result
, 'error/class', 'GenericError')
256 # This fails because it needs to modify the backing string in node2, which is blocked
257 result
= self
.vm
.qmp('block-commit', device
='drive0', base
=self
.imgs
[0], top
=self
.imgs
[1], job_id
='commit-node0')
258 self
.assert_qmp(result
, 'error/class', 'GenericError')
260 self
.wait_until_completed(drive
='stream-node4')
261 self
.assert_no_active_block_jobs()
263 # Similar to test_overlapping_1, but with block-commit
264 # blocking the other jobs
265 def test_overlapping_2(self
):
266 self
.assertLessEqual(9, self
.num_imgs
)
267 self
.assert_no_active_block_jobs()
269 # Set a speed limit to make sure that this job blocks the rest
270 result
= self
.vm
.qmp('block-commit', device
='drive0', top
=self
.imgs
[5], base
=self
.imgs
[3], job_id
='commit-node3', speed
=1024*1024)
271 self
.assert_qmp(result
, 'return', {})
273 result
= self
.vm
.qmp('block-stream', device
='node3', job_id
='stream-node3')
274 self
.assert_qmp(result
, 'error/class', 'GenericError')
276 result
= self
.vm
.qmp('block-stream', device
='node6', base
=self
.imgs
[2], job_id
='stream-node6')
277 self
.assert_qmp(result
, 'error/class', 'GenericError')
279 result
= self
.vm
.qmp('block-stream', device
='node4', base
=self
.imgs
[2], job_id
='stream-node4')
280 self
.assert_qmp(result
, 'error/class', 'GenericError')
282 result
= self
.vm
.qmp('block-stream', device
='node6', base
=self
.imgs
[4], job_id
='stream-node6-v2')
283 self
.assert_qmp(result
, 'error/class', 'GenericError')
285 # This fails because block-commit needs to block node6, the overlay of the 'top' image
286 result
= self
.vm
.qmp('block-stream', device
='node7', base
=self
.imgs
[5], job_id
='stream-node6-v3')
287 self
.assert_qmp(result
, 'error/class', 'GenericError')
289 # This fails because block-commit currently blocks the active layer even if it's not used
290 result
= self
.vm
.qmp('block-stream', device
='drive0', base
=self
.imgs
[5], job_id
='stream-drive0')
291 self
.assert_qmp(result
, 'error/class', 'GenericError')
293 self
.wait_until_completed(drive
='commit-node3')
295 # Similar to test_overlapping_2, but here block-commit doesn't use the 'top' parameter.
296 # Internally this uses a mirror block job, hence the separate test case.
297 def test_overlapping_3(self
):
298 self
.assertLessEqual(8, self
.num_imgs
)
299 self
.assert_no_active_block_jobs()
301 # Set a speed limit to make sure that this job blocks the rest
302 result
= self
.vm
.qmp('block-commit', device
='drive0', base
=self
.imgs
[3], job_id
='commit-drive0', speed
=1024*1024)
303 self
.assert_qmp(result
, 'return', {})
305 result
= self
.vm
.qmp('block-stream', device
='node5', base
=self
.imgs
[3], job_id
='stream-node6')
306 self
.assert_qmp(result
, 'error/class', 'GenericError')
308 event
= self
.vm
.get_qmp_event(wait
=True)
309 self
.assertEqual(event
['event'], 'BLOCK_JOB_READY')
310 self
.assert_qmp(event
, 'data/device', 'commit-drive0')
311 self
.assert_qmp(event
, 'data/type', 'commit')
312 self
.assert_qmp_absent(event
, 'data/error')
314 result
= self
.vm
.qmp('block-job-complete', device
='commit-drive0')
315 self
.assert_qmp(result
, 'return', {})
317 self
.wait_until_completed(drive
='commit-drive0')
319 # Test a block-stream and a block-commit job in parallel
320 def test_stream_commit(self
):
321 self
.assertLessEqual(8, self
.num_imgs
)
322 self
.assert_no_active_block_jobs()
324 # Stream from node0 into node2
325 result
= self
.vm
.qmp('block-stream', device
='node2', job_id
='node2')
326 self
.assert_qmp(result
, 'return', {})
328 # Commit from the active layer into node3
329 result
= self
.vm
.qmp('block-commit', device
='drive0', base
=self
.imgs
[3])
330 self
.assert_qmp(result
, 'return', {})
332 # Wait for all jobs to be finished.
333 pending_jobs
= ['node2', 'drive0']
334 while len(pending_jobs
) > 0:
335 for event
in self
.vm
.get_qmp_events(wait
=True):
336 if event
['event'] == 'BLOCK_JOB_COMPLETED':
337 node_name
= self
.dictpath(event
, 'data/device')
338 self
.assertTrue(node_name
in pending_jobs
)
339 self
.assert_qmp_absent(event
, 'data/error')
340 pending_jobs
.remove(node_name
)
341 if event
['event'] == 'BLOCK_JOB_READY':
342 self
.assert_qmp(event
, 'data/device', 'drive0')
343 self
.assert_qmp(event
, 'data/type', 'commit')
344 self
.assert_qmp_absent(event
, 'data/error')
345 self
.assertTrue('drive0' in pending_jobs
)
346 self
.vm
.qmp('block-job-complete', device
='drive0')
348 self
.assert_no_active_block_jobs()
350 class TestQuorum(iotests
.QMPTestCase
):
356 opts
= ['driver=quorum', 'vote-threshold=2']
358 # Initialize file names and command-line options
359 for i
in range(self
.num_children
):
360 child_img
= os
.path
.join(iotests
.test_dir
, 'img-%d.img' % i
)
361 backing_img
= os
.path
.join(iotests
.test_dir
, 'backing-%d.img' % i
)
362 self
.children
.append(child_img
)
363 self
.backing
.append(backing_img
)
364 qemu_img('create', '-f', iotests
.imgfmt
, backing_img
, '1M')
365 qemu_io('-f', iotests
.imgfmt
,
366 '-c', 'write -P 0x55 0 1024', backing_img
)
367 qemu_img('create', '-f', iotests
.imgfmt
,
368 '-o', 'backing_file=%s' % backing_img
, child_img
)
369 opts
.append("children.%d.file.filename=%s" % (i
, child_img
))
370 opts
.append("children.%d.node-name=node%d" % (i
, i
))
372 # Attach the drive to the VM
373 self
.vm
= iotests
.VM()
374 self
.vm
.add_drive(path
= None, opts
= ','.join(opts
))
379 for img
in self
.children
:
381 for img
in self
.backing
:
384 def test_stream_quorum(self
):
385 if not iotests
.supports_quorum():
388 self
.assertNotEqual(qemu_io('-f', iotests
.imgfmt
, '-c', 'map', self
.children
[0]),
389 qemu_io('-f', iotests
.imgfmt
, '-c', 'map', self
.backing
[0]),
390 'image file map matches backing file before streaming')
392 self
.assert_no_active_block_jobs()
394 result
= self
.vm
.qmp('block-stream', device
='node0', job_id
='stream-node0')
395 self
.assert_qmp(result
, 'return', {})
397 self
.wait_until_completed(drive
='stream-node0')
399 self
.assert_no_active_block_jobs()
402 self
.assertEqual(qemu_io('-f', iotests
.imgfmt
, '-c', 'map', self
.children
[0]),
403 qemu_io('-f', iotests
.imgfmt
, '-c', 'map', self
.backing
[0]),
404 'image file map does not match backing file after streaming')
406 class TestSmallerBackingFile(iotests
.QMPTestCase
):
407 backing_len
= 1 * 1024 * 1024 # MB
408 image_len
= 2 * backing_len
411 iotests
.create_image(backing_img
, self
.backing_len
)
412 qemu_img('create', '-f', iotests
.imgfmt
, '-o', 'backing_file=%s' % backing_img
, test_img
, str(self
.image_len
))
413 self
.vm
= iotests
.VM().add_drive(test_img
)
416 # If this hangs, then you are missing a fix to complete streaming when the
417 # end of the backing file is reached.
418 def test_stream(self
):
419 self
.assert_no_active_block_jobs()
421 result
= self
.vm
.qmp('block-stream', device
='drive0')
422 self
.assert_qmp(result
, 'return', {})
424 self
.wait_until_completed()
426 self
.assert_no_active_block_jobs()
429 class TestErrors(iotests
.QMPTestCase
):
430 image_len
= 2 * 1024 * 1024 # MB
432 # this should match STREAM_BUFFER_SIZE/512 in block/stream.c
433 STREAM_BUFFER_SIZE
= 512 * 1024
435 def create_blkdebug_file(self
, name
, event
, errno
):
436 file = open(name
, 'w')
455 ''' % (event
, errno
, self
.STREAM_BUFFER_SIZE
/ 512, event
, event
))
458 class TestEIO(TestErrors
):
460 self
.blkdebug_file
= backing_img
+ ".blkdebug"
461 iotests
.create_image(backing_img
, TestErrors
.image_len
)
462 self
.create_blkdebug_file(self
.blkdebug_file
, "read_aio", 5)
463 qemu_img('create', '-f', iotests
.imgfmt
,
464 '-o', 'backing_file=blkdebug:%s:%s,backing_fmt=raw'
465 % (self
.blkdebug_file
, backing_img
),
467 self
.vm
= iotests
.VM().add_drive(test_img
)
473 os
.remove(backing_img
)
474 os
.remove(self
.blkdebug_file
)
476 def test_report(self
):
477 self
.assert_no_active_block_jobs()
479 result
= self
.vm
.qmp('block-stream', device
='drive0')
480 self
.assert_qmp(result
, 'return', {})
485 for event
in self
.vm
.get_qmp_events(wait
=True):
486 if event
['event'] == 'BLOCK_JOB_ERROR':
487 self
.assert_qmp(event
, 'data/device', 'drive0')
488 self
.assert_qmp(event
, 'data/operation', 'read')
490 elif event
['event'] == 'BLOCK_JOB_COMPLETED':
491 self
.assertTrue(error
, 'job completed unexpectedly')
492 self
.assert_qmp(event
, 'data/type', 'stream')
493 self
.assert_qmp(event
, 'data/device', 'drive0')
494 self
.assert_qmp(event
, 'data/error', 'Input/output error')
495 self
.assert_qmp(event
, 'data/offset', self
.STREAM_BUFFER_SIZE
)
496 self
.assert_qmp(event
, 'data/len', self
.image_len
)
499 self
.assert_no_active_block_jobs()
502 def test_ignore(self
):
503 self
.assert_no_active_block_jobs()
505 result
= self
.vm
.qmp('block-stream', device
='drive0', on_error
='ignore')
506 self
.assert_qmp(result
, 'return', {})
511 for event
in self
.vm
.get_qmp_events(wait
=True):
512 if event
['event'] == 'BLOCK_JOB_ERROR':
513 self
.assert_qmp(event
, 'data/device', 'drive0')
514 self
.assert_qmp(event
, 'data/operation', 'read')
515 result
= self
.vm
.qmp('query-block-jobs')
516 self
.assert_qmp(result
, 'return[0]/paused', False)
518 elif event
['event'] == 'BLOCK_JOB_COMPLETED':
519 self
.assertTrue(error
, 'job completed unexpectedly')
520 self
.assert_qmp(event
, 'data/type', 'stream')
521 self
.assert_qmp(event
, 'data/device', 'drive0')
522 self
.assert_qmp(event
, 'data/error', 'Input/output error')
523 self
.assert_qmp(event
, 'data/offset', self
.image_len
)
524 self
.assert_qmp(event
, 'data/len', self
.image_len
)
527 self
.assert_no_active_block_jobs()
531 self
.assert_no_active_block_jobs()
533 result
= self
.vm
.qmp('block-stream', device
='drive0', on_error
='stop')
534 self
.assert_qmp(result
, 'return', {})
539 for event
in self
.vm
.get_qmp_events(wait
=True):
540 if event
['event'] == 'BLOCK_JOB_ERROR':
542 self
.assert_qmp(event
, 'data/device', 'drive0')
543 self
.assert_qmp(event
, 'data/operation', 'read')
545 result
= self
.vm
.qmp('query-block-jobs')
546 self
.assert_qmp(result
, 'return[0]/paused', True)
547 self
.assert_qmp(result
, 'return[0]/offset', self
.STREAM_BUFFER_SIZE
)
548 self
.assert_qmp(result
, 'return[0]/io-status', 'failed')
550 result
= self
.vm
.qmp('block-job-resume', device
='drive0')
551 self
.assert_qmp(result
, 'return', {})
553 result
= self
.vm
.qmp('query-block-jobs')
554 if result
== {'return': []}:
555 # Race; likely already finished. Check.
557 self
.assert_qmp(result
, 'return[0]/paused', False)
558 self
.assert_qmp(result
, 'return[0]/io-status', 'ok')
559 elif event
['event'] == 'BLOCK_JOB_COMPLETED':
560 self
.assertTrue(error
, 'job completed unexpectedly')
561 self
.assert_qmp(event
, 'data/type', 'stream')
562 self
.assert_qmp(event
, 'data/device', 'drive0')
563 self
.assert_qmp_absent(event
, 'data/error')
564 self
.assert_qmp(event
, 'data/offset', self
.image_len
)
565 self
.assert_qmp(event
, 'data/len', self
.image_len
)
568 self
.assert_no_active_block_jobs()
571 def test_enospc(self
):
572 self
.assert_no_active_block_jobs()
574 result
= self
.vm
.qmp('block-stream', device
='drive0', on_error
='enospc')
575 self
.assert_qmp(result
, 'return', {})
580 for event
in self
.vm
.get_qmp_events(wait
=True):
581 if event
['event'] == 'BLOCK_JOB_ERROR':
582 self
.assert_qmp(event
, 'data/device', 'drive0')
583 self
.assert_qmp(event
, 'data/operation', 'read')
585 elif event
['event'] == 'BLOCK_JOB_COMPLETED':
586 self
.assertTrue(error
, 'job completed unexpectedly')
587 self
.assert_qmp(event
, 'data/type', 'stream')
588 self
.assert_qmp(event
, 'data/device', 'drive0')
589 self
.assert_qmp(event
, 'data/error', 'Input/output error')
590 self
.assert_qmp(event
, 'data/offset', self
.STREAM_BUFFER_SIZE
)
591 self
.assert_qmp(event
, 'data/len', self
.image_len
)
594 self
.assert_no_active_block_jobs()
597 class TestENOSPC(TestErrors
):
599 self
.blkdebug_file
= backing_img
+ ".blkdebug"
600 iotests
.create_image(backing_img
, TestErrors
.image_len
)
601 self
.create_blkdebug_file(self
.blkdebug_file
, "read_aio", 28)
602 qemu_img('create', '-f', iotests
.imgfmt
,
603 '-o', 'backing_file=blkdebug:%s:%s,backing_fmt=raw'
604 % (self
.blkdebug_file
, backing_img
),
606 self
.vm
= iotests
.VM().add_drive(test_img
)
612 os
.remove(backing_img
)
613 os
.remove(self
.blkdebug_file
)
615 def test_enospc(self
):
616 self
.assert_no_active_block_jobs()
618 result
= self
.vm
.qmp('block-stream', device
='drive0', on_error
='enospc')
619 self
.assert_qmp(result
, 'return', {})
624 for event
in self
.vm
.get_qmp_events(wait
=True):
625 if event
['event'] == 'BLOCK_JOB_ERROR':
626 self
.assert_qmp(event
, 'data/device', 'drive0')
627 self
.assert_qmp(event
, 'data/operation', 'read')
629 result
= self
.vm
.qmp('query-block-jobs')
630 self
.assert_qmp(result
, 'return[0]/paused', True)
631 self
.assert_qmp(result
, 'return[0]/offset', self
.STREAM_BUFFER_SIZE
)
632 self
.assert_qmp(result
, 'return[0]/io-status', 'nospace')
634 result
= self
.vm
.qmp('block-job-resume', device
='drive0')
635 self
.assert_qmp(result
, 'return', {})
637 result
= self
.vm
.qmp('query-block-jobs')
638 self
.assert_qmp(result
, 'return[0]/paused', False)
639 self
.assert_qmp(result
, 'return[0]/io-status', 'ok')
641 elif event
['event'] == 'BLOCK_JOB_COMPLETED':
642 self
.assertTrue(error
, 'job completed unexpectedly')
643 self
.assert_qmp(event
, 'data/type', 'stream')
644 self
.assert_qmp(event
, 'data/device', 'drive0')
645 self
.assert_qmp_absent(event
, 'data/error')
646 self
.assert_qmp(event
, 'data/offset', self
.image_len
)
647 self
.assert_qmp(event
, 'data/len', self
.image_len
)
650 self
.assert_no_active_block_jobs()
653 class TestStreamStop(iotests
.QMPTestCase
):
654 image_len
= 8 * 1024 * 1024 * 1024 # GB
657 qemu_img('create', backing_img
, str(TestStreamStop
.image_len
))
658 qemu_io('-f', 'raw', '-c', 'write -P 0x1 0 32M', backing_img
)
659 qemu_img('create', '-f', iotests
.imgfmt
, '-o', 'backing_file=%s' % backing_img
, test_img
)
660 qemu_io('-f', iotests
.imgfmt
, '-c', 'write -P 0x1 32M 32M', test_img
)
661 self
.vm
= iotests
.VM().add_drive("blkdebug::" + test_img
)
667 os
.remove(backing_img
)
669 def test_stream_stop(self
):
670 self
.assert_no_active_block_jobs()
672 self
.vm
.pause_drive('drive0')
673 result
= self
.vm
.qmp('block-stream', device
='drive0')
674 self
.assert_qmp(result
, 'return', {})
677 events
= self
.vm
.get_qmp_events(wait
=False)
678 self
.assertEqual(events
, [], 'unexpected QMP event: %s' % events
)
680 self
.cancel_and_wait(resume
=True)
682 class TestSetSpeed(iotests
.QMPTestCase
):
683 image_len
= 80 * 1024 * 1024 # MB
686 qemu_img('create', backing_img
, str(TestSetSpeed
.image_len
))
687 qemu_io('-f', 'raw', '-c', 'write -P 0x1 0 32M', backing_img
)
688 qemu_img('create', '-f', iotests
.imgfmt
, '-o', 'backing_file=%s' % backing_img
, test_img
)
689 qemu_io('-f', iotests
.imgfmt
, '-c', 'write -P 0x1 32M 32M', test_img
)
690 self
.vm
= iotests
.VM().add_drive('blkdebug::' + test_img
)
696 os
.remove(backing_img
)
698 # This is a short performance test which is not run by default.
699 # Invoke "IMGFMT=qed ./030 TestSetSpeed.perf_test_throughput"
700 def perf_test_throughput(self
):
701 self
.assert_no_active_block_jobs()
703 result
= self
.vm
.qmp('block-stream', device
='drive0')
704 self
.assert_qmp(result
, 'return', {})
706 result
= self
.vm
.qmp('block-job-set-speed', device
='drive0', speed
=8 * 1024 * 1024)
707 self
.assert_qmp(result
, 'return', {})
709 self
.wait_until_completed()
711 self
.assert_no_active_block_jobs()
713 def test_set_speed(self
):
714 self
.assert_no_active_block_jobs()
716 self
.vm
.pause_drive('drive0')
717 result
= self
.vm
.qmp('block-stream', device
='drive0')
718 self
.assert_qmp(result
, 'return', {})
721 result
= self
.vm
.qmp('query-block-jobs')
722 self
.assert_qmp(result
, 'return[0]/device', 'drive0')
723 self
.assert_qmp(result
, 'return[0]/speed', 0)
725 result
= self
.vm
.qmp('block-job-set-speed', device
='drive0', speed
=8 * 1024 * 1024)
726 self
.assert_qmp(result
, 'return', {})
728 # Ensure the speed we set was accepted
729 result
= self
.vm
.qmp('query-block-jobs')
730 self
.assert_qmp(result
, 'return[0]/device', 'drive0')
731 self
.assert_qmp(result
, 'return[0]/speed', 8 * 1024 * 1024)
733 self
.cancel_and_wait(resume
=True)
734 self
.vm
.pause_drive('drive0')
736 # Check setting speed in block-stream works
737 result
= self
.vm
.qmp('block-stream', device
='drive0', speed
=4 * 1024 * 1024)
738 self
.assert_qmp(result
, 'return', {})
740 result
= self
.vm
.qmp('query-block-jobs')
741 self
.assert_qmp(result
, 'return[0]/device', 'drive0')
742 self
.assert_qmp(result
, 'return[0]/speed', 4 * 1024 * 1024)
744 self
.cancel_and_wait(resume
=True)
746 def test_set_speed_invalid(self
):
747 self
.assert_no_active_block_jobs()
749 result
= self
.vm
.qmp('block-stream', device
='drive0', speed
=-1)
750 self
.assert_qmp(result
, 'error/class', 'GenericError')
752 self
.assert_no_active_block_jobs()
754 result
= self
.vm
.qmp('block-stream', device
='drive0')
755 self
.assert_qmp(result
, 'return', {})
757 result
= self
.vm
.qmp('block-job-set-speed', device
='drive0', speed
=-1)
758 self
.assert_qmp(result
, 'error/class', 'GenericError')
760 self
.cancel_and_wait()
762 if __name__
== '__main__':
763 iotests
.main(supported_fmts
=['qcow2', 'qed'])