xref: /openbmc/u-boot/test/py/tests/test_gpt.py (revision bb737ced) 
1# Copyright (c) 2017 Alison Chaiken
2#
3# SPDX-License-Identifier: GPL-2.0
4
5# Test GPT manipulation commands.
6
7import os
8import pytest
9import u_boot_utils
10import make_test_disk
11
12"""
13These tests rely on a 4 MB block device called testdisk.raw
14which is automatically removed at the end of the tests.
15"""
16
17@pytest.mark.buildconfigspec('cmd_gpt')
18def test_gpt_guid(u_boot_console):
19    """Test the gpt guid command."""
20
21    if u_boot_console.config.buildconfig.get('config_cmd_gpt', 'n') != 'y':
22        pytest.skip('gpt command not supported')
23    make_test_disk.makeDisk()
24    u_boot_console.run_command('host bind 0 testdisk.raw')
25    output = u_boot_console.run_command('gpt guid host 0')
26    assert '375a56f7-d6c9-4e81-b5f0-09d41ca89efe' in output
27
28@pytest.mark.buildconfigspec('cmd_gpt')
29def test_gpt_save_guid(u_boot_console):
30    """Test the gpt guid command to save GUID into a string."""
31
32    if u_boot_console.config.buildconfig.get('config_cmd_gpt', 'n') != 'y':
33        pytest.skip('gpt command not supported')
34    u_boot_console.run_command('host bind 0 testdisk.raw')
35    output = u_boot_console.run_command('gpt guid host 0 newguid')
36    output = u_boot_console.run_command('printenv newguid')
37    assert '375a56f7-d6c9-4e81-b5f0-09d41ca89efe' in output
38
39@pytest.mark.buildconfigspec('cmd_gpt')
40def test_gpt_rename_partition(u_boot_console):
41    """Test the gpt rename command to write partition names."""
42
43    if u_boot_console.config.buildconfig.get('config_cmd_gpt_rename', 'n') != 'y':
44        pytest.skip('gpt rename command not supported')
45    u_boot_console.run_command('host bind 0 testdisk.raw')
46    u_boot_console.run_command('gpt rename host 0 1 first')
47    output = u_boot_console.run_command('gpt read host 0')
48    assert 'name first' in output
49    u_boot_console.run_command('gpt rename host 0 2 second')
50    output = u_boot_console.run_command('gpt read host 0')
51    assert 'name second' in output
52
53@pytest.mark.buildconfigspec('cmd_gpt')
54def test_gpt_swap_partitions(u_boot_console):
55    """Test the gpt swap command to exchange two partition names."""
56
57    if u_boot_console.config.buildconfig.get('config_cmd_gpt_rename', 'n') != 'y':
58        pytest.skip('gpt rename command not supported')
59    if u_boot_console.config.buildconfig.get('config_cmd_part', 'n') != 'y':
60        pytest.skip('gpt swap test needs CMD_PART')
61    u_boot_console.run_command('host bind 0 testdisk.raw')
62    output = u_boot_console.run_command('part list host 0')
63    assert '0x000007ff	"first"' in output
64    assert '0x000017ff	"second"' in output
65    u_boot_console.run_command('gpt swap host 0 first second')
66    output = u_boot_console.run_command('part list host 0')
67    assert '0x000007ff	"second"' in output
68    assert '0x000017ff	"first"' in output
69    os.remove('testdisk.raw')
70