]>
git.ipfire.org Git - people/ms/u-boot.git/blob - test/py/tests/test_gpt.py
e2bbd08e6d42a6cbc4bb9da73684c37f46c2b95a
1 # Copyright (c) 2017 Alison Chaiken
2 # Copyright (c) 2017, NVIDIA CORPORATION. All rights reserved.
4 # SPDX-License-Identifier: GPL-2.0
6 # Test GPT manipulation commands.
13 These tests rely on a 4 MB disk image, which is automatically created by
17 class GptTestDiskImage(object):
18 """Disk Image used by the GPT tests."""
20 def __init__(self
, u_boot_console
):
21 """Initialize a new GptTestDiskImage object.
24 u_boot_console: A U-Boot console.
30 filename
= 'test_gpt_disk_image.bin'
31 self
.path
= u_boot_console
.config
.persistent_data_dir
+ '/' + filename
33 if os
.path
.exists(self
.path
):
34 u_boot_console
.log
.action('Disk image file ' + self
.path
+
37 u_boot_console
.log
.action('Generating ' + self
.path
)
38 fd
= os
.open(self
.path
, os
.O_RDWR | os
.O_CREAT
)
39 os
.ftruncate(fd
, 4194304)
41 sgdisk
= '/sbin/sgdisk'
42 cmd
= (sgdisk
, '-U', '375a56f7-d6c9-4e81-b5f0-09d41ca89efe',
44 u_boot_utils
.run_and_log(u_boot_console
, cmd
)
45 cmd
= (sgdisk
, '--new=1:2048:2560', self
.path
)
46 u_boot_utils
.run_and_log(u_boot_console
, cmd
)
47 cmd
= (sgdisk
, '--new=2:4096:4608', self
.path
)
48 u_boot_utils
.run_and_log(u_boot_console
, cmd
)
49 cmd
= (sgdisk
, '-l', self
.path
)
50 u_boot_utils
.run_and_log(u_boot_console
, cmd
)
53 @pytest.fixture(scope
='function')
54 def state_disk_image(u_boot_console
):
55 """pytest fixture to provide a GptTestDiskImage object to tests.
56 This is function-scoped because it uses u_boot_console, which is also
57 function-scoped. However, we don't need to actually do any function-scope
58 work, so this simply returns the same object over and over each time."""
62 gtdi
= GptTestDiskImage(u_boot_console
)
65 @pytest.mark
.boardspec('sandbox')
66 @pytest.mark
.buildconfigspec('cmd_gpt')
67 def test_gpt_guid(state_disk_image
, u_boot_console
):
68 """Test the gpt guid command."""
70 u_boot_console
.run_command('host bind 0 ' + state_disk_image
.path
)
71 output
= u_boot_console
.run_command('gpt guid host 0')
72 assert '375a56f7-d6c9-4e81-b5f0-09d41ca89efe' in output
74 @pytest.mark
.boardspec('sandbox')
75 @pytest.mark
.buildconfigspec('cmd_gpt')
76 def test_gpt_save_guid(state_disk_image
, u_boot_console
):
77 """Test the gpt guid command to save GUID into a string."""
79 if u_boot_console
.config
.buildconfig
.get('config_cmd_gpt', 'n') != 'y':
80 pytest
.skip('gpt command not supported')
81 u_boot_console
.run_command('host bind 0 ' + state_disk_image
.path
)
82 output
= u_boot_console
.run_command('gpt guid host 0 newguid')
83 output
= u_boot_console
.run_command('printenv newguid')
84 assert '375a56f7-d6c9-4e81-b5f0-09d41ca89efe' in output
86 @pytest.mark
.boardspec('sandbox')
87 @pytest.mark
.buildconfigspec('cmd_gpt')
88 @pytest.mark
.buildconfigspec('cmd_gpt_rename')
89 def test_gpt_rename_partition(state_disk_image
, u_boot_console
):
90 """Test the gpt rename command to write partition names."""
92 u_boot_console
.run_command('host bind 0 ' + state_disk_image
.path
)
93 u_boot_console
.run_command('gpt rename host 0 1 first')
94 output
= u_boot_console
.run_command('gpt read host 0')
95 assert 'name first' in output
96 u_boot_console
.run_command('gpt rename host 0 2 second')
97 output
= u_boot_console
.run_command('gpt read host 0')
98 assert 'name second' in output
100 @pytest.mark
.boardspec('sandbox')
101 @pytest.mark
.buildconfigspec('cmd_gpt')
102 @pytest.mark
.buildconfigspec('cmd_gpt_rename')
103 @pytest.mark
.buildconfigspec('cmd_part')
104 def test_gpt_swap_partitions(state_disk_image
, u_boot_console
):
105 """Test the gpt swap command to exchange two partition names."""
107 u_boot_console
.run_command('host bind 0 ' + state_disk_image
.path
)
108 output
= u_boot_console
.run_command('part list host 0')
109 assert '0x000007ff "first"' in output
110 assert '0x000017ff "second"' in output
111 u_boot_console
.run_command('gpt swap host 0 first second')
112 output
= u_boot_console
.run_command('part list host 0')
113 assert '0x000007ff "second"' in output
114 assert '0x000017ff "first"' in output