The attribute disk_start_sector allows to specify a custom start sector for the first partition of the disk. On GPT tables everything works nicely, on DOS tables the used tools fdisk/sfdisk are not able to manage the start/end values of subsequent partitions if the first partition doesn't start with the tooling default. This patch allows to set the start sector after the partition table has been created
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
from mock import (
|
|
Mock, patch
|
|
)
|
|
|
|
from kiwi.utils.block import BlockID
|
|
|
|
|
|
class TestBlockID:
|
|
def setup(self):
|
|
self.blkid = BlockID('device')
|
|
|
|
def setup_method(self, cls):
|
|
self.setup()
|
|
|
|
@patch('kiwi.utils.block.Command.run')
|
|
def test_setup_with_uuid_format(self, mock_command):
|
|
BlockID('UUID=uuid')
|
|
mock_command.assert_called_once_with(
|
|
['blkid', '--uuid', 'uuid']
|
|
)
|
|
|
|
@patch('kiwi.utils.block.Command.run')
|
|
def test_get_blkid(self, mock_command):
|
|
self.blkid.get_blkid('LABEL')
|
|
mock_command.assert_called_once_with(
|
|
['blkid', 'device', '-s', 'LABEL', '-o', 'value'],
|
|
raise_on_error=False
|
|
)
|
|
|
|
@patch('kiwi.utils.block.BlockID.get_blkid')
|
|
def test_get_filesystem(self, mock_get_blkid):
|
|
self.blkid.get_filesystem()
|
|
mock_get_blkid.assert_called_once_with('TYPE')
|
|
|
|
@patch('kiwi.utils.block.BlockID.get_blkid')
|
|
def test_get_label(self, mock_get_blkid):
|
|
self.blkid.get_label()
|
|
mock_get_blkid.assert_called_once_with('LABEL')
|
|
|
|
@patch('kiwi.utils.block.BlockID.get_blkid')
|
|
def test_get_uuid(self, mock_get_blkid):
|
|
self.blkid.get_uuid()
|
|
mock_get_blkid.assert_called_once_with('UUID')
|
|
|
|
@patch('kiwi.utils.block.Command.run')
|
|
def test_get_partition_count(self, mock_Command_run):
|
|
lsblk_call = Mock()
|
|
lsblk_call.output = "NAME TYPE\nsda disk\nsda4 part \nsda3 part"
|
|
mock_Command_run.return_value = lsblk_call
|
|
assert self.blkid.get_partition_count() == 2
|