diff options
Diffstat (limited to 'test/py/tests/test_gpt.py')
| -rw-r--r-- | test/py/tests/test_gpt.py | 87 |
1 files changed, 66 insertions, 21 deletions
diff --git a/test/py/tests/test_gpt.py b/test/py/tests/test_gpt.py index 06f24b6..e2bbd08 100644 --- a/test/py/tests/test_gpt.py +++ b/test/py/tests/test_gpt.py @@ -1,4 +1,5 @@ # Copyright (c) 2017 Alison Chaiken +# Copyright (c) 2017, NVIDIA CORPORATION. All rights reserved. # # SPDX-License-Identifier: GPL-2.0 @@ -7,42 +8,88 @@ import os import pytest import u_boot_utils -import make_test_disk """ -These tests rely on a 4 MB block device called testdisk.raw -which is automatically removed at the end of the tests. +These tests rely on a 4 MB disk image, which is automatically created by +the test. """ +class GptTestDiskImage(object): + """Disk Image used by the GPT tests.""" + + def __init__(self, u_boot_console): + """Initialize a new GptTestDiskImage object. + + Args: + u_boot_console: A U-Boot console. + + Returns: + Nothing. + """ + + filename = 'test_gpt_disk_image.bin' + self.path = u_boot_console.config.persistent_data_dir + '/' + filename + + if os.path.exists(self.path): + u_boot_console.log.action('Disk image file ' + self.path + + ' already exists') + else: + u_boot_console.log.action('Generating ' + self.path) + fd = os.open(self.path, os.O_RDWR | os.O_CREAT) + os.ftruncate(fd, 4194304) + os.close(fd) + sgdisk = '/sbin/sgdisk' + cmd = (sgdisk, '-U', '375a56f7-d6c9-4e81-b5f0-09d41ca89efe', + self.path) + u_boot_utils.run_and_log(u_boot_console, cmd) + cmd = (sgdisk, '--new=1:2048:2560', self.path) + u_boot_utils.run_and_log(u_boot_console, cmd) + cmd = (sgdisk, '--new=2:4096:4608', self.path) + u_boot_utils.run_and_log(u_boot_console, cmd) + cmd = (sgdisk, '-l', self.path) + u_boot_utils.run_and_log(u_boot_console, cmd) + +gtdi = None +@pytest.fixture(scope='function') +def state_disk_image(u_boot_console): + """pytest fixture to provide a GptTestDiskImage object to tests. + This is function-scoped because it uses u_boot_console, which is also + function-scoped. However, we don't need to actually do any function-scope + work, so this simply returns the same object over and over each time.""" + + global gtdi + if not gtdi: + gtdi = GptTestDiskImage(u_boot_console) + return gtdi + +@pytest.mark.boardspec('sandbox') @pytest.mark.buildconfigspec('cmd_gpt') -def test_gpt_guid(u_boot_console): +def test_gpt_guid(state_disk_image, u_boot_console): """Test the gpt guid command.""" - if u_boot_console.config.buildconfig.get('config_cmd_gpt', 'n') != 'y': - pytest.skip('gpt command not supported') - make_test_disk.makeDisk() - u_boot_console.run_command('host bind 0 testdisk.raw') + u_boot_console.run_command('host bind 0 ' + state_disk_image.path) output = u_boot_console.run_command('gpt guid host 0') assert '375a56f7-d6c9-4e81-b5f0-09d41ca89efe' in output +@pytest.mark.boardspec('sandbox') @pytest.mark.buildconfigspec('cmd_gpt') -def test_gpt_save_guid(u_boot_console): +def test_gpt_save_guid(state_disk_image, u_boot_console): """Test the gpt guid command to save GUID into a string.""" if u_boot_console.config.buildconfig.get('config_cmd_gpt', 'n') != 'y': pytest.skip('gpt command not supported') - u_boot_console.run_command('host bind 0 testdisk.raw') + u_boot_console.run_command('host bind 0 ' + state_disk_image.path) output = u_boot_console.run_command('gpt guid host 0 newguid') output = u_boot_console.run_command('printenv newguid') assert '375a56f7-d6c9-4e81-b5f0-09d41ca89efe' in output +@pytest.mark.boardspec('sandbox') @pytest.mark.buildconfigspec('cmd_gpt') -def test_gpt_rename_partition(u_boot_console): +@pytest.mark.buildconfigspec('cmd_gpt_rename') +def test_gpt_rename_partition(state_disk_image, u_boot_console): """Test the gpt rename command to write partition names.""" - if u_boot_console.config.buildconfig.get('config_cmd_gpt_rename', 'n') != 'y': - pytest.skip('gpt rename command not supported') - u_boot_console.run_command('host bind 0 testdisk.raw') + u_boot_console.run_command('host bind 0 ' + state_disk_image.path) u_boot_console.run_command('gpt rename host 0 1 first') output = u_boot_console.run_command('gpt read host 0') assert 'name first' in output @@ -50,15 +97,14 @@ def test_gpt_rename_partition(u_boot_console): output = u_boot_console.run_command('gpt read host 0') assert 'name second' in output +@pytest.mark.boardspec('sandbox') @pytest.mark.buildconfigspec('cmd_gpt') -def test_gpt_swap_partitions(u_boot_console): +@pytest.mark.buildconfigspec('cmd_gpt_rename') +@pytest.mark.buildconfigspec('cmd_part') +def test_gpt_swap_partitions(state_disk_image, u_boot_console): """Test the gpt swap command to exchange two partition names.""" - if u_boot_console.config.buildconfig.get('config_cmd_gpt_rename', 'n') != 'y': - pytest.skip('gpt rename command not supported') - if u_boot_console.config.buildconfig.get('config_cmd_part', 'n') != 'y': - pytest.skip('gpt swap test needs CMD_PART') - u_boot_console.run_command('host bind 0 testdisk.raw') + u_boot_console.run_command('host bind 0 ' + state_disk_image.path) output = u_boot_console.run_command('part list host 0') assert '0x000007ff "first"' in output assert '0x000017ff "second"' in output @@ -66,4 +112,3 @@ def test_gpt_swap_partitions(u_boot_console): output = u_boot_console.run_command('part list host 0') assert '0x000007ff "second"' in output assert '0x000017ff "first"' in output - os.remove('testdisk.raw') |