blob: 64fb4fc8d45a857ae3f0274dfcbce9b22c808c29 [file] [log] [blame]
# Copyright 2013 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Checks that the partition table extends nearly to the end of the storage
This test checks if the partition table allocates at least ``min_usage_pct``
percent of the storage. If not, this test expands stateful patition to the end
of the storage device by default.
This test doesn't check the actual size of the stateful partition, rather the
sector at which it ends.
This test doesn't support remote device.
Test Procedure
This is an automatic test that doesn't need any user interaction.
- `pygpt` utility.
To run this pytest with default arguments, add this in test list::
"pytest_name": "partition_table"
This is also predefined in ``generic_common.test_list.json`` as
If you can't expand stateful partition for some reason, override the argument
"inherit": "PartitionTable",
"args": {
"expand_stateful": false
import logging
import os
import factory_common # pylint: disable=unused-import
from cros.factory.device import device_utils
from cros.factory.test import test_case
from cros.factory.utils.arg_utils import Arg
from cros.factory.utils import pygpt
class PartitionTableTest(test_case.TestCase):
ARGS = [
Arg('min_usage_pct', (int, float),
'Percentage of the storage device that must be before the end of the '
'stateful partition. For example, if this is 95%, then the stateful '
'partition must end at a sector that is >=95% of the total number of '
'sectors on the device.',
Arg('expand_stateful', bool,
'Repair partition headers and tables and expand stateful partition '
'to all available free space',
def setUp(self):
self.dut = device_utils.CreateDUTInterface()
self.gpt = None
def runTest(self):
'This test only support local device')
dev =
self.gpt = pygpt.GPT.LoadFromFile(dev)
stateful_no = self.dut.partitions.STATEFUL.index
stateful_part = self.gpt.GetPartition(stateful_no)
start_sector = stateful_part.FirstLBA
sector_count = stateful_part.blocks
end_sector = start_sector + sector_count
sector_size = self.gpt.block_size
# Linux always considers sectors to be 512 bytes long independently of the
# devices real block size.
device_size = 512 * int(
self.dut.ReadFile('/sys/class/block/%s/size' % os.path.basename(dev)))
pct_used = end_sector * sector_size * 100.0 / device_size
'start_sector=%d, sector_count=%d, end_sector=%d, device_size=%d',
start_sector, sector_count, end_sector, device_size)'Stateful partition extends to %.3f%% of storage',
if pct_used < self.args.min_usage_pct:
if not self.args.expand_stateful:
self.FailTask('Stateful partition does not cover enough of storage '
# Repair partition headers and tables