git.sur5r.net Git - u-boot/blob - test/py/tests/test_gpt.py

   1 # Copyright (c) 2017 Alison Chaiken
   2 # Copyright (c) 2017, NVIDIA CORPORATION. All rights reserved.
   3 #
   4 # SPDX-License-Identifier: GPL-2.0
   5
   6 # Test GPT manipulation commands.
   7
   8 import os
   9 import pytest
  10 import u_boot_utils
  11
  12 """
  13 These tests rely on a 4 MB disk image, which is automatically created by
  14 the test.
  15 """
  16
  17 class GptTestDiskImage(object):
  18     """Disk Image used by the GPT tests."""
  19
  20     def __init__(self, u_boot_console):
  21         """Initialize a new GptTestDiskImage object.
  22
  23         Args:
  24             u_boot_console: A U-Boot console.
  25
  26         Returns:
  27             Nothing.
  28         """
  29
  30         filename = 'test_gpt_disk_image.bin'
  31
  32         persistent = u_boot_console.config.persistent_data_dir + '/' + filename
  33         self.path = u_boot_console.config.result_dir  + '/' + filename
  34
  35         if os.path.exists(persistent):
  36             u_boot_console.log.action('Disk image file ' + persistent +
  37                 ' already exists')
  38         else:
  39             u_boot_console.log.action('Generating ' + persistent)
  40             fd = os.open(persistent, os.O_RDWR | os.O_CREAT)
  41             os.ftruncate(fd, 4194304)
  42             os.close(fd)
  43             cmd = ('sgdisk', '-U', '375a56f7-d6c9-4e81-b5f0-09d41ca89efe',
  44                 persistent)
  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)
  52
  53         cmd = ('cp', persistent, self.path)
  54         u_boot_utils.run_and_log(u_boot_console, cmd)
  55
  56 gtdi = None
  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."""
  63
  64     global gtdi
  65     if not gtdi:
  66         gtdi = GptTestDiskImage(u_boot_console)
  67     return gtdi
  68
  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."""
  74
  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
  78
  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."""
  84
  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
  91
  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."""
  98
  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
 106
 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."""
 114
 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