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         self.path = u_boot_console.config.persistent_data_dir + '/' + filename
  32
  33         if os.path.exists(self.path):
  34             u_boot_console.log.action('Disk image file ' + self.path +
  35                 ' already exists')
  36         else:
  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)
  40             os.close(fd)
  41             sgdisk = '/sbin/sgdisk'
  42             cmd = (sgdisk, '-U', '375a56f7-d6c9-4e81-b5f0-09d41ca89efe',
  43                 self.path)
  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)
  51
  52 gtdi = None
  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."""
  59
  60     global gtdi
  61     if not gtdi:
  62         gtdi = GptTestDiskImage(u_boot_console)
  63     return gtdi
  64
  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."""
  69
  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
  73
  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."""
  78
  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
  85
  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."""
  91
  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
  99
 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."""
 106
 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