| # Copyright 2022 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| from recipe_engine.post_process import DropExpectation, StatusSuccess |
| from recipe_engine.post_process import StatusFailure |
| from recipe_engine.recipe_api import Property |
| from textwrap import dedent |
| |
| DEPS = ['qemu', 'recipe_engine/path', 'recipe_engine/raw_io'] |
| |
| PYTHON_VERSION_COMPATIBILITY = 'PY3' |
| |
| |
| def RunSteps(api): |
| api.qemu.init('latest') |
| # test good cases for both create_empty_disk and create_disk |
| api.qemu.create_disk('fat_disk', 'fat', 20480) |
| # mock existence of cache directory |
| api.path.mock_add_directory(api.path.cache_dir) |
| # mock cleanup to be a file |
| api.path.mock_add_file(api.path.cleanup_dir) |
| api.qemu.create_disk('ext4_disk', 'ext4', 20480, |
| {api.path.cache_dir: 'got_cache/i_need_it'}) |
| |
| |
| def GenTests(api): |
| yield (api.test('Test create_disk pass') + api.post_process(StatusSuccess) + |
| api.step_data( |
| 'Copy files to ext4_disk.Mount loop', |
| api.raw_io.stream_output( |
| 'Mounted /dev/loop6 at /media/chrome-bot/test'), |
| retcode=0) + |
| # mock the free disk size to say there is enough |
| api.step_data( |
| 'Check free space on disk for fat_disk', |
| api.raw_io.stream_output( |
| dedent('''Avail |
| 41943050 |
| ''')), |
| retcode=0) + |
| # mock the free disk size to say there is enough |
| api.step_data( |
| 'Check free space on disk for ext4_disk', |
| api.raw_io.stream_output( |
| dedent('''Avail |
| 20971530 |
| ''')), |
| retcode=0) + api.step_data( |
| 'Estimate size required for ext4_disk', |
| api.raw_io.stream_output( |
| dedent('''123456 /cache/disk/files/keys/cert.pem |
| 234565 /cache/disk/files/ap/wpa.conf |
| 324551 /cache/disk/files/keymaster/km |
| |
| /cache/disk/files/ssh: |
| total 8 |
| 435623 /cache/disk/files/ssh/authorized_users |
| 1230 /cache/disk/files/ssh/id_pub |
| ''')), |
| retcode=0) + api.post_process(DropExpectation)) |
| |
| yield (api.test('Test create_disk fail (mount permission)', status='FAILURE') + |
| api.step_data( |
| 'Copy files to ext4_disk.Mount loop', |
| api.raw_io.stream_output('Permission denied: /dev/loop6'), |
| retcode=1) + |
| # mock the free disk size to say there is enough |
| api.step_data( |
| 'Check free space on disk for fat_disk', |
| api.raw_io.stream_output( |
| dedent('''Avail |
| 41943050 |
| ''')), |
| retcode=0) + |
| # mock the free disk size to say there is enough |
| api.step_data( |
| 'Check free space on disk for ext4_disk', |
| api.raw_io.stream_output( |
| dedent('''Avail |
| 20971530 |
| ''')), |
| retcode=0) + api.step_data( |
| 'Estimate size required for ext4_disk', |
| api.raw_io.stream_output( |
| dedent('''123456 chrome:primarygroup /cache/keys/cert.pem |
| 234565 chrome:primarygroup /cache/ap/wpa.conf |
| 324551 chrome:primarygroup /cache/keymaster/km |
| |
| /ssh: |
| total 8 |
| 435623 chrome:primarygroup /ssh/authorized_users |
| 1230 chrome:primarygroup /ssh/id_pub |
| ERROR: failed to read path /ssh |
| ''')), |
| retcode=0) + api.step_data( |
| 'Copy files to ext4_disk.Mount loop', |
| api.raw_io.stream_output('Permission denied: /dev/loop6'), |
| retcode=1) + api.post_process(DropExpectation)) |
| |
| yield (api.test('Test create_disk fail (out of disk)', status='FAILURE') + |
| # mock the free disk size to say there is enough |
| api.step_data( |
| 'Check free space on disk for fat_disk', |
| api.raw_io.stream_output( |
| dedent('''Avail |
| 27815012 |
| ''')), |
| retcode=0) + |
| # mock the free disk size to say not enough |
| api.step_data( |
| 'Check free space on disk for ext4_disk', |
| api.raw_io.stream_output( |
| dedent('''Avail |
| 6843492 |
| ''')), |
| retcode=0) + api.post_process(DropExpectation)) |
