116 lines
4.4 KiB
Python
116 lines
4.4 KiB
Python
from mock import patch
|
|
|
|
import mock
|
|
|
|
from .test_helper import raises
|
|
|
|
from kiwi.partitioner import Partitioner
|
|
|
|
from kiwi.exceptions import KiwiPartitionerSetupError
|
|
|
|
|
|
class TestPartitioner(object):
|
|
@patch('platform.machine')
|
|
@raises(KiwiPartitionerSetupError)
|
|
def test_partitioner_not_implemented(self, mock_machine):
|
|
mock_machine.return_value = 'x86_64'
|
|
Partitioner('foo', mock.Mock())
|
|
|
|
@patch('platform.machine')
|
|
@raises(KiwiPartitionerSetupError)
|
|
def test_partitioner_for_arch_not_implemented(self, mock_machine):
|
|
mock_machine.return_value = 'some-arch'
|
|
Partitioner('foo', mock.Mock())
|
|
|
|
@patch('kiwi.partitioner.PartitionerGpt')
|
|
@patch('platform.machine')
|
|
def test_partitioner_x86_64_gpt(self, mock_machine, mock_gpt):
|
|
mock_machine.return_value = 'x86_64'
|
|
storage_provider = mock.Mock()
|
|
Partitioner('gpt', storage_provider)
|
|
mock_gpt.assert_called_once_with(storage_provider, None)
|
|
|
|
@patch('kiwi.partitioner.PartitionerMsDos')
|
|
@patch('platform.machine')
|
|
def test_partitioner_x86_64_msdos(self, mock_machine, mock_dos):
|
|
mock_machine.return_value = 'x86_64'
|
|
storage_provider = mock.Mock()
|
|
Partitioner('msdos', storage_provider)
|
|
mock_dos.assert_called_once_with(storage_provider, None)
|
|
|
|
@patch('kiwi.partitioner.PartitionerMsDos')
|
|
@patch('platform.machine')
|
|
def test_partitioner_i686_msdos(self, mock_machine, mock_dos):
|
|
mock_machine.return_value = 'i686'
|
|
storage_provider = mock.Mock()
|
|
Partitioner('msdos', storage_provider)
|
|
mock_dos.assert_called_once_with(storage_provider, None)
|
|
|
|
@patch('kiwi.partitioner.PartitionerMsDos')
|
|
@patch('platform.machine')
|
|
def test_partitioner_i586_msdos(self, mock_machine, mock_dos):
|
|
mock_machine.return_value = 'i586'
|
|
storage_provider = mock.Mock()
|
|
Partitioner('msdos', storage_provider)
|
|
mock_dos.assert_called_once_with(storage_provider, None)
|
|
|
|
@patch('kiwi.partitioner.PartitionerDasd')
|
|
@patch('platform.machine')
|
|
def test_partitioner_s390_dasd(self, mock_machine, mock_dasd):
|
|
mock_machine.return_value = 's390'
|
|
storage_provider = mock.Mock()
|
|
Partitioner('dasd', storage_provider)
|
|
mock_dasd.assert_called_once_with(storage_provider)
|
|
|
|
@patch('kiwi.logger.log.warning')
|
|
@patch('kiwi.partitioner.PartitionerDasd')
|
|
@patch('platform.machine')
|
|
def test_partitioner_s390_dasd_with_custom_start_sector(
|
|
self, mock_machine, mock_dasd, mock_warning
|
|
):
|
|
mock_machine.return_value = 's390'
|
|
storage_provider = mock.Mock()
|
|
Partitioner('dasd', storage_provider, 4096)
|
|
mock_dasd.assert_called_once_with(storage_provider)
|
|
assert mock_warning.called
|
|
|
|
@patch('kiwi.partitioner.PartitionerMsDos')
|
|
@patch('platform.machine')
|
|
def test_partitioner_s390_msdos(self, mock_machine, mock_dos):
|
|
mock_machine.return_value = 's390'
|
|
storage_provider = mock.Mock()
|
|
Partitioner('msdos', storage_provider)
|
|
mock_dos.assert_called_once_with(storage_provider, None)
|
|
|
|
@patch('kiwi.partitioner.PartitionerMsDos')
|
|
@patch('platform.machine')
|
|
def test_partitioner_ppc_msdos(self, mock_machine, mock_dos):
|
|
mock_machine.return_value = 'ppc64'
|
|
storage_provider = mock.Mock()
|
|
Partitioner('msdos', storage_provider)
|
|
mock_dos.assert_called_once_with(storage_provider, None)
|
|
|
|
@patch('kiwi.partitioner.PartitionerGpt')
|
|
@patch('platform.machine')
|
|
def test_partitioner_ppc_gpt(self, mock_machine, mock_gpt):
|
|
mock_machine.return_value = 'ppc64'
|
|
storage_provider = mock.Mock()
|
|
Partitioner('gpt', storage_provider)
|
|
mock_gpt.assert_called_once_with(storage_provider, None)
|
|
|
|
@patch('kiwi.partitioner.PartitionerGpt')
|
|
@patch('platform.machine')
|
|
def test_partitioner_arm_gpt(self, mock_machine, mock_gpt):
|
|
mock_machine.return_value = 'aarch64'
|
|
storage_provider = mock.Mock()
|
|
Partitioner('gpt', storage_provider)
|
|
mock_gpt.assert_called_once_with(storage_provider, None)
|
|
|
|
@patch('kiwi.partitioner.PartitionerMsDos')
|
|
@patch('platform.machine')
|
|
def test_partitioner_arm_msdos(self, mock_machine, mock_dos):
|
|
mock_machine.return_value = 'armv7l'
|
|
storage_provider = mock.Mock()
|
|
Partitioner('msdos', storage_provider)
|
|
mock_dos.assert_called_once_with(storage_provider, None)
|