1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
from unittest.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.BlockID.get_blkid')
def test_get_ptuuid(self, mock_get_blkid):
self.blkid.get_ptuuid()
mock_get_blkid.assert_called_once_with('PTUUID')
@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
|
