test/py/tests/test_gpt.py

   1 # Copyright (c) 2017 Alison Chaiken
   2 #
   3 # SPDX-License-Identifier: GPL-2.0
   4
   5 # Test GPT manipulation commands.
   6
   7 import os
   8 import pytest
   9 import u_boot_utils
  10 import make_test_disk
  11
  12 """
  13 These tests rely on a 4 MB block device called testdisk.raw
  14 which is automatically removed at the end of the tests.
  15 """
  16
  17 @pytest.mark.buildconfigspec('cmd_gpt')
  18 def 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')
  29 def 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')
  40 def 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')
  54 def 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')