kiwi-el8/test/unit/utils/block_test.py
Marcus Schäfer 0540d6ddb9
Fixed custom disk start sector setup
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
2022-07-07 12:21:08 +02:00

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