aboutsummaryrefslogtreecommitdiff
path: root/test/py/tests/test_gpt.py
blob: 06f24b66cef8732c86879a8f72bf7ba787046d59 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
# Copyright (c) 2017 Alison Chaiken
#
# SPDX-License-Identifier: GPL-2.0

# Test GPT manipulation commands.

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.
"""

@pytest.mark.buildconfigspec('cmd_gpt')
def test_gpt_guid(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')
    output = u_boot_console.run_command('gpt guid host 0')
    assert '375a56f7-d6c9-4e81-b5f0-09d41ca89efe' in output

@pytest.mark.buildconfigspec('cmd_gpt')
def test_gpt_save_guid(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')
    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.buildconfigspec('cmd_gpt')
def test_gpt_rename_partition(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('gpt rename host 0 1 first')
    output = u_boot_console.run_command('gpt read host 0')
    assert 'name first' in output
    u_boot_console.run_command('gpt rename host 0 2 second')
    output = u_boot_console.run_command('gpt read host 0')
    assert 'name second' in output

@pytest.mark.buildconfigspec('cmd_gpt')
def test_gpt_swap_partitions(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')
    output = u_boot_console.run_command('part list host 0')
    assert '0x000007ff	"first"' in output
    assert '0x000017ff	"second"' in output
    u_boot_console.run_command('gpt swap host 0 first second')
    output = u_boot_console.run_command('part list host 0')
    assert '0x000007ff	"second"' in output
    assert '0x000017ff	"first"' in output
    os.remove('testdisk.raw')