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'
32 persistent = u_boot_console.config.persistent_data_dir + '/' + filename
33 self.path = u_boot_console.config.result_dir + '/' + filename
35 if os.path.exists(persistent):
36 u_boot_console.log.action('Disk image file ' + persistent +
39 u_boot_console.log.action('Generating ' + persistent)
40 fd = os.open(persistent, os.O_RDWR | os.O_CREAT)
41 os.ftruncate(fd, 4194304)
43 cmd = ('sgdisk', '-U', '375a56f7-d6c9-4e81-b5f0-09d41ca89efe',
45 u_boot_utils.run_and_log(u_boot_console, cmd)
46 cmd = ('sgdisk', '--new=1:2048:2560', persistent)
47 u_boot_utils.run_and_log(u_boot_console, cmd)
48 cmd = ('sgdisk', '--new=2:4096:4608', persistent)
49 u_boot_utils.run_and_log(u_boot_console, cmd)
50 cmd = ('sgdisk', '-l', persistent)
51 u_boot_utils.run_and_log(u_boot_console, cmd)
53 cmd = ('cp', persistent, self.path)
54 u_boot_utils.run_and_log(u_boot_console, cmd)
57 @pytest.fixture(scope='function')
58 def state_disk_image(u_boot_console):
59 """pytest fixture to provide a GptTestDiskImage object to tests.
60 This is function-scoped because it uses u_boot_console, which is also
61 function-scoped. However, we don't need to actually do any function-scope
62 work, so this simply returns the same object over and over each time."""
66 gtdi = GptTestDiskImage(u_boot_console)
69 @pytest.mark.boardspec('sandbox')
70 @pytest.mark.buildconfigspec('cmd_gpt')
71 @pytest.mark.requiredtool('sgdisk')
72 def test_gpt_guid(state_disk_image, u_boot_console):
73 """Test the gpt guid command."""
75 u_boot_console.run_command('host bind 0 ' + state_disk_image.path)
76 output = u_boot_console.run_command('gpt guid host 0')
77 assert '375a56f7-d6c9-4e81-b5f0-09d41ca89efe' in output
79 @pytest.mark.boardspec('sandbox')
80 @pytest.mark.buildconfigspec('cmd_gpt')
81 @pytest.mark.requiredtool('sgdisk')
82 def test_gpt_save_guid(state_disk_image, u_boot_console):
83 """Test the gpt guid command to save GUID into a string."""
85 if u_boot_console.config.buildconfig.get('config_cmd_gpt', 'n') != 'y':
86 pytest.skip('gpt command not supported')
87 u_boot_console.run_command('host bind 0 ' + state_disk_image.path)
88 output = u_boot_console.run_command('gpt guid host 0 newguid')
89 output = u_boot_console.run_command('printenv newguid')
90 assert '375a56f7-d6c9-4e81-b5f0-09d41ca89efe' in output
92 @pytest.mark.boardspec('sandbox')
93 @pytest.mark.buildconfigspec('cmd_gpt')
94 @pytest.mark.buildconfigspec('cmd_gpt_rename')
95 @pytest.mark.requiredtool('sgdisk')
96 def test_gpt_rename_partition(state_disk_image, u_boot_console):
97 """Test the gpt rename command to write partition names."""
99 u_boot_console.run_command('host bind 0 ' + state_disk_image.path)
100 u_boot_console.run_command('gpt rename host 0 1 first')
101 output = u_boot_console.run_command('gpt read host 0')
102 assert 'name first' in output
103 u_boot_console.run_command('gpt rename host 0 2 second')
104 output = u_boot_console.run_command('gpt read host 0')
105 assert 'name second' in output
107 @pytest.mark.boardspec('sandbox')
108 @pytest.mark.buildconfigspec('cmd_gpt')
109 @pytest.mark.buildconfigspec('cmd_gpt_rename')
110 @pytest.mark.buildconfigspec('cmd_part')
111 @pytest.mark.requiredtool('sgdisk')
112 def test_gpt_swap_partitions(state_disk_image, u_boot_console):
113 """Test the gpt swap command to exchange two partition names."""
115 u_boot_console.run_command('host bind 0 ' + state_disk_image.path)
116 output = u_boot_console.run_command('part list host 0')
117 assert '0x000007ff "first"' in output
118 assert '0x000017ff "second"' in output
119 u_boot_console.run_command('gpt swap host 0 first second')
120 output = u_boot_console.run_command('part list host 0')
121 assert '0x000007ff "second"' in output
122 assert '0x000017ff "first"' in output