def setup(self, mock_open, mock_partitioner):
self.tempfile = mock.Mock()
self.tempfile.name = 'tempfile'
self.context_manager_mock = mock.Mock()
self.file_mock = mock.Mock()
self.enter_mock = mock.Mock()
self.exit_mock = mock.Mock()
self.enter_mock.return_value = self.file_mock
setattr(self.context_manager_mock, '__enter__', self.enter_mock)
setattr(self.context_manager_mock, '__exit__', self.exit_mock)
self.partitioner = mock.Mock()
self.partitioner.create = mock.Mock()
self.partitioner.get_id = mock.Mock(
return_value=1
)
mock_partitioner.return_value = self.partitioner
self.storage_provider = mock.Mock()
self.storage_provider.is_loop = mock.Mock(
return_value=True
)
self.storage_provider.get_device = mock.Mock(
return_value='/dev/loop0'
)
self.disk = Disk('gpt', self.storage_provider)
def setup(self, mock_open, mock_partitioner):
self.tempfile = mock.Mock()
self.tempfile.name = 'tempfile'
self.context_manager_mock = mock.Mock()
self.file_mock = mock.Mock()
self.enter_mock = mock.Mock()
self.exit_mock = mock.Mock()
self.enter_mock.return_value = self.file_mock
setattr(self.context_manager_mock, '__enter__', self.enter_mock)
setattr(self.context_manager_mock, '__exit__', self.exit_mock)
self.partitioner = mock.Mock()
self.partitioner.create = mock.Mock()
self.partitioner.get_id = mock.Mock(return_value=1)
mock_partitioner.return_value = self.partitioner
self.storage_provider = mock.Mock()
self.storage_provider.is_loop = mock.Mock(return_value=True)
self.storage_provider.get_device = mock.Mock(return_value='/dev/loop0')
self.disk = Disk('gpt', self.storage_provider)
class TestDisk(object):
@patch('kiwi.disk.Partitioner')
@patch('__builtin__.open')
def setup(self, mock_open, mock_partitioner):
self.tempfile = mock.Mock()
self.tempfile.name = 'tempfile'
self.context_manager_mock = mock.Mock()
self.file_mock = mock.Mock()
self.enter_mock = mock.Mock()
self.exit_mock = mock.Mock()
self.enter_mock.return_value = self.file_mock
setattr(self.context_manager_mock, '__enter__', self.enter_mock)
setattr(self.context_manager_mock, '__exit__', self.exit_mock)
self.partitioner = mock.Mock()
self.partitioner.create = mock.Mock()
self.partitioner.get_id = mock.Mock(return_value=1)
mock_partitioner.return_value = self.partitioner
self.storage_provider = mock.Mock()
self.storage_provider.is_loop = mock.Mock(return_value=True)
self.storage_provider.get_device = mock.Mock(return_value='/dev/loop0')
self.disk = Disk('gpt', self.storage_provider)
@patch('os.path.exists')
def test_get_device(self, mock_exists):
mock_exists.return_value = True
self.disk.partition_map['root'] = '/dev/root-device'
assert self.disk.get_device()['root'].get_device(
) == '/dev/root-device'
def test_is_loop(self):
self.disk.is_loop()
self.storage_provider.is_loop.called_once_with()
def test_create_root_partition(self):
self.disk.create_root_partition(100)
self.partitioner.create.assert_called_once_with(
'p.lxroot', 100, 't.linux')
def test_create_root_and_boot_partition(self):
self.disk.create_root_partition(200)
self.partitioner.create.assert_called_once_with(
'p.lxroot', 200, 't.linux')
assert self.disk.partition_id_map['kiwi_RootPart'] == 1
assert self.disk.partition_id_map['kiwi_BootPart'] == 1
def test_create_root_lvm_partition(self):
self.disk.create_root_lvm_partition(100)
self.partitioner.create.assert_called_once_with(
'p.lxlvm', 100, 't.lvm')
assert self.disk.partition_id_map['kiwi_RootPart'] == 1
assert self.disk.partition_id_map['kiwi_RootPartVol'] == 'LVRoot'
def test_create_root_raid_partition(self):
self.disk.create_root_raid_partition(100)
self.partitioner.create.assert_called_once_with(
'p.lxraid', 100, 't.raid')
assert self.disk.partition_id_map['kiwi_RootPart'] == 1
assert self.disk.partition_id_map['kiwi_RaidPart'] == 1
assert self.disk.partition_id_map['kiwi_RaidDev'] == '/dev/md0'
def test_create_boot_partition(self):
self.disk.create_boot_partition(100)
self.partitioner.create.assert_called_once_with(
'p.lxboot', 100, 't.linux')
assert self.disk.partition_id_map['kiwi_BootPart'] == 1
def test_create_efi_csm_partition(self):
self.disk.create_efi_csm_partition(100)
self.partitioner.create.assert_called_once_with(
'p.legacy', 100, 't.csm')
assert self.disk.partition_id_map['kiwi_BiosGrub'] == 1
def test_create_efi_partition(self):
self.disk.create_efi_partition(100)
self.partitioner.create.assert_called_once_with('p.UEFI', 100, 't.efi')
assert self.disk.partition_id_map['kiwi_JumpPart'] == 1
@patch('kiwi.disk.Command.run')
def test_device_map_loop(self, mock_command):
self.disk.create_efi_partition(100)
self.disk.map_partitions()
assert self.disk.partition_map == {'efi': '/dev/mapper/loop0p1'}
self.disk.is_mapped = False
@patch('kiwi.disk.Command.run')
def test_device_map_linux_dev_sda(self, mock_command):
self.storage_provider.is_loop.return_value = False
self.storage_provider.get_device = mock.Mock(return_value='/dev/sda')
self.disk.create_efi_partition(100)
self.disk.map_partitions()
assert self.disk.partition_map == {'efi': '/dev/sda1'}
self.disk.is_mapped = False
@patch('kiwi.disk.Command.run')
def test_device_map_linux_dev_c0d0(self, mock_command):
self.storage_provider.is_loop.return_value = False
self.storage_provider.get_device = mock.Mock(return_value='/dev/c0d0')
self.disk.create_efi_partition(100)
self.disk.map_partitions()
assert self.disk.partition_map == {'efi': '/dev/c0d0p1'}
self.disk.is_mapped = False
@patch('kiwi.disk.Command.run')
def test_activate_boot_partition_is_boot_partition(self, mock_command):
self.disk.create_boot_partition(100)
self.disk.create_root_partition(100)
self.disk.activate_boot_partition()
self.partitioner.set_flag(1, 'f.active')
@patch('kiwi.disk.Command.run')
def test_activate_boot_partition_is_root_partition(self, mock_command):
self.disk.create_root_partition(100)
self.disk.activate_boot_partition()
self.partitioner.set_flag(1, 'f.active')
@patch('kiwi.disk.Command.run')
def test_wipe_gpt(self, mock_command):
self.disk.wipe()
mock_command.assert_called_once_with(
['sgdisk', '--zap-all', '/dev/loop0'])
@patch('kiwi.disk.Command.run')
@patch('__builtin__.open')
@patch('kiwi.disk.NamedTemporaryFile')
def test_wipe_dasd(self, mock_temp, mock_open, mock_command):
self.disk.table_type = 'dasd'
mock_temp.return_value = self.tempfile
mock_open.return_value = self.context_manager_mock
self.disk.wipe()
self.file_mock.write.assert_called_once_with('y\n\nw\nq\n')
mock_command.assert_called_once_with(
['bash', '-c', 'cat tempfile | fdasd -f /dev/loop0'])
@patch('kiwi.disk.Command.run')
def test_map_partitions_loop(self, mock_command):
self.disk.map_partitions()
mock_command.assert_called_once_with(
['kpartx', '-s', '-a', '/dev/loop0'])
self.disk.is_mapped = False
@patch('kiwi.disk.Command.run')
def test_map_partitions_other(self, mock_command):
self.storage_provider.is_loop.return_value = False
self.disk.map_partitions()
mock_command.assert_called_once_with(['partprobe', '/dev/loop0'])
@patch('kiwi.disk.Command.run')
@patch('kiwi.logger.log.warning')
def test_destructor(self, mock_log_warn, mock_command):
self.disk.is_mapped = True
mock_command.side_effect = Exception
self.disk.__del__()
mock_command.assert_called_once_with(
['kpartx', '-s', '-d', '/dev/loop0'])
assert mock_log_warn.called
self.disk.is_mapped = False
def test_get_partition_id_map(self):
assert self.disk.get_partition_id_map() == {}
class TestDisk(object):
@patch('kiwi.disk.Partitioner')
@patch('__builtin__.open')
def setup(self, mock_open, mock_partitioner):
self.tempfile = mock.Mock()
self.tempfile.name = 'tempfile'
self.context_manager_mock = mock.Mock()
self.file_mock = mock.Mock()
self.enter_mock = mock.Mock()
self.exit_mock = mock.Mock()
self.enter_mock.return_value = self.file_mock
setattr(self.context_manager_mock, '__enter__', self.enter_mock)
setattr(self.context_manager_mock, '__exit__', self.exit_mock)
self.partitioner = mock.Mock()
self.partitioner.create = mock.Mock()
self.partitioner.get_id = mock.Mock(
return_value=1
)
mock_partitioner.return_value = self.partitioner
self.storage_provider = mock.Mock()
self.storage_provider.is_loop = mock.Mock(
return_value=True
)
self.storage_provider.get_device = mock.Mock(
return_value='/dev/loop0'
)
self.disk = Disk('gpt', self.storage_provider)
@patch('os.path.exists')
def test_get_device(self, mock_exists):
mock_exists.return_value = True
self.disk.partition_map['root'] = '/dev/root-device'
assert self.disk.get_device()['root'].get_device() == '/dev/root-device'
def test_is_loop(self):
self.disk.is_loop()
self.storage_provider.is_loop.called_once_with()
def test_create_root_partition(self):
self.disk.create_root_partition(100)
self.partitioner.create.assert_called_once_with(
'p.lxroot', 100, 't.linux'
)
def test_create_root_and_boot_partition(self):
self.disk.create_root_partition(200)
self.partitioner.create.assert_called_once_with(
'p.lxroot', 200, 't.linux'
)
assert self.disk.partition_id_map['kiwi_RootPart'] == 1
assert self.disk.partition_id_map['kiwi_BootPart'] == 1
def test_create_root_lvm_partition(self):
self.disk.create_root_lvm_partition(100)
self.partitioner.create.assert_called_once_with(
'p.lxlvm', 100, 't.lvm'
)
assert self.disk.partition_id_map['kiwi_RootPart'] == 1
assert self.disk.partition_id_map['kiwi_RootPartVol'] == 'LVRoot'
def test_create_root_raid_partition(self):
self.disk.create_root_raid_partition(100)
self.partitioner.create.assert_called_once_with(
'p.lxraid', 100, 't.raid'
)
assert self.disk.partition_id_map['kiwi_RootPart'] == 1
assert self.disk.partition_id_map['kiwi_RaidPart'] == 1
assert self.disk.partition_id_map['kiwi_RaidDev'] == '/dev/md0'
def test_create_boot_partition(self):
self.disk.create_boot_partition(100)
self.partitioner.create.assert_called_once_with(
'p.lxboot', 100, 't.linux'
)
assert self.disk.partition_id_map['kiwi_BootPart'] == 1
def test_create_efi_csm_partition(self):
self.disk.create_efi_csm_partition(100)
self.partitioner.create.assert_called_once_with(
'p.legacy', 100, 't.csm'
)
assert self.disk.partition_id_map['kiwi_BiosGrub'] == 1
def test_create_efi_partition(self):
self.disk.create_efi_partition(100)
self.partitioner.create.assert_called_once_with(
'p.UEFI', 100, 't.efi'
)
assert self.disk.partition_id_map['kiwi_JumpPart'] == 1
@patch('kiwi.disk.Command.run')
def test_device_map_loop(self, mock_command):
self.disk.create_efi_partition(100)
self.disk.map_partitions()
assert self.disk.partition_map == {'efi': '/dev/mapper/loop0p1'}
self.disk.is_mapped = False
@patch('kiwi.disk.Command.run')
def test_device_map_linux_dev_sda(self, mock_command):
self.storage_provider.is_loop.return_value = False
self.storage_provider.get_device = mock.Mock(
return_value='/dev/sda'
)
self.disk.create_efi_partition(100)
self.disk.map_partitions()
assert self.disk.partition_map == {'efi': '/dev/sda1'}
self.disk.is_mapped = False
@patch('kiwi.disk.Command.run')
def test_device_map_linux_dev_c0d0(self, mock_command):
self.storage_provider.is_loop.return_value = False
self.storage_provider.get_device = mock.Mock(
return_value='/dev/c0d0'
)
self.disk.create_efi_partition(100)
self.disk.map_partitions()
assert self.disk.partition_map == {'efi': '/dev/c0d0p1'}
self.disk.is_mapped = False
@patch('kiwi.disk.Command.run')
def test_activate_boot_partition_is_boot_partition(self, mock_command):
self.disk.create_boot_partition(100)
self.disk.create_root_partition(100)
self.disk.activate_boot_partition()
self.partitioner.set_flag(1, 'f.active')
@patch('kiwi.disk.Command.run')
def test_activate_boot_partition_is_root_partition(self, mock_command):
self.disk.create_root_partition(100)
self.disk.activate_boot_partition()
self.partitioner.set_flag(1, 'f.active')
@patch('kiwi.disk.Command.run')
def test_wipe_gpt(self, mock_command):
self.disk.wipe()
mock_command.assert_called_once_with(
['sgdisk', '--zap-all', '/dev/loop0']
)
@patch('kiwi.disk.Command.run')
@patch('__builtin__.open')
@patch('kiwi.disk.NamedTemporaryFile')
def test_wipe_dasd(self, mock_temp, mock_open, mock_command):
self.disk.table_type = 'dasd'
mock_temp.return_value = self.tempfile
mock_open.return_value = self.context_manager_mock
self.disk.wipe()
self.file_mock.write.assert_called_once_with(
'y\n\nw\nq\n'
)
mock_command.assert_called_once_with(
['bash', '-c', 'cat tempfile | fdasd -f /dev/loop0']
)
@patch('kiwi.disk.Command.run')
def test_map_partitions_loop(self, mock_command):
self.disk.map_partitions()
mock_command.assert_called_once_with(
['kpartx', '-s', '-a', '/dev/loop0']
)
self.disk.is_mapped = False
@patch('kiwi.disk.Command.run')
def test_map_partitions_other(self, mock_command):
self.storage_provider.is_loop.return_value = False
self.disk.map_partitions()
mock_command.assert_called_once_with(
['partprobe', '/dev/loop0']
)
@patch('kiwi.disk.Command.run')
@patch('kiwi.logger.log.warning')
def test_destructor(self, mock_log_warn, mock_command):
self.disk.is_mapped = True
mock_command.side_effect = Exception
self.disk.__del__()
mock_command.assert_called_once_with(
['kpartx', '-s', '-d', '/dev/loop0']
)
assert mock_log_warn.called
self.disk.is_mapped = False
def test_get_partition_id_map(self):
assert self.disk.get_partition_id_map() == {}
