def __new__(self, table_type, storage_provider): # noqa: C901
host_architecture = platform.machine()
if host_architecture == 'x86_64':
if table_type == 'gpt':
return PartitionerGpt(storage_provider)
elif table_type == 'msdos':
return PartitionerMsDos(storage_provider)
elif host_architecture == 'i686' or host_architecture == 'i586':
if table_type == 'msdos':
return PartitionerMsDos(storage_provider)
elif 'ppc64' in host_architecture:
if table_type == 'gpt':
return PartitionerGpt(storage_provider)
elif table_type == 'msdos':
return PartitionerMsDos(storage_provider)
elif 's390' in host_architecture:
if table_type == 'dasd':
return PartitionerDasd(storage_provider)
elif table_type == 'msdos':
return PartitionerMsDos(storage_provider)
elif 'arm' in host_architecture or host_architecture == 'aarch64':
if table_type == 'gpt':
return PartitionerGpt(storage_provider)
elif table_type == 'msdos':
return PartitionerMsDos(storage_provider)
raise KiwiPartitionerSetupError(
'Support for partitioner on %s architecture not implemented' %
host_architecture)
def test_create_custom_start_sector(self, mock_temp, mock_flag,
mock_command):
disk_provider = Mock()
disk_provider.get_device = Mock(return_value='/dev/loop0')
partitioner = PartitionerMsDos(disk_provider, 4096)
mock_command.side_effect = Exception
temp_type = namedtuple('temp_type', ['name'])
mock_temp.return_value = temp_type(name='tempfile')
m_open = mock_open()
with patch('builtins.open', m_open, create=True):
partitioner.create('name', 100, 't.linux', ['f.active'])
partitioner.create('name', 100, 't.linux', ['f.active'])
mock_command.assert_has_calls([
call(['bash', '-c', 'cat tempfile | fdisk /dev/loop0']),
call(['bash', '-c', 'cat tempfile | fdisk /dev/loop0'])
])
assert mock_flag.call_args_list[0] == \
call(1, 't.linux')
assert mock_flag.call_args_list[1] == \
call(1, 'f.active')
m_open.return_value.write.assert_has_calls([
call('n\np\n1\n4096\n+100M\nw\nq\n'),
call('n\np\n2\n\n+100M\nw\nq\n')
])
def test_create_custom_start_sector(self, mock_open, mock_temp, mock_flag,
mock_command):
disk_provider = mock.Mock()
disk_provider.get_device = mock.Mock(return_value='/dev/loop0')
partitioner = PartitionerMsDos(disk_provider, 4096)
mock_command.side_effect = Exception
temp_type = namedtuple('temp_type', ['name'])
mock_temp.return_value = temp_type(name='tempfile')
context_manager_mock = mock.Mock()
mock_open.return_value = context_manager_mock
file_mock = mock.Mock()
enter_mock = mock.Mock()
exit_mock = mock.Mock()
enter_mock.return_value = file_mock
setattr(context_manager_mock, '__enter__', enter_mock)
setattr(context_manager_mock, '__exit__', exit_mock)
partitioner.create('name', 100, 't.linux', ['f.active'])
partitioner.create('name', 100, 't.linux', ['f.active'])
mock_command.assert_has_calls([
call(['bash', '-c', 'cat tempfile | fdisk /dev/loop0']),
call(['bash', '-c', 'cat tempfile | fdisk /dev/loop0'])
])
assert mock_flag.call_args_list[0] == \
call(1, 't.linux')
assert mock_flag.call_args_list[1] == \
call(1, 'f.active')
file_mock.write.assert_has_calls([
call('n\np\n1\n4096\n+100M\nw\nq\n'),
call('n\np\n2\n\n+100M\nw\nq\n')
])
def __new__(self,
table_type,
storage_provider,
start_sector=None): # noqa: C901
host_architecture = platform.machine()
if host_architecture == 'x86_64':
if table_type == 'gpt':
return PartitionerGpt(storage_provider, start_sector)
elif table_type == 'msdos':
return PartitionerMsDos(storage_provider, start_sector)
elif host_architecture == 'i686' or host_architecture == 'i586':
if table_type == 'msdos':
return PartitionerMsDos(storage_provider, start_sector)
elif 'ppc64' in host_architecture:
if table_type == 'gpt':
return PartitionerGpt(storage_provider, start_sector)
elif table_type == 'msdos':
return PartitionerMsDos(storage_provider, start_sector)
elif 's390' in host_architecture:
if table_type == 'dasd':
if start_sector:
log.warning('disk_start_sector value is ignored '
'for dasd partitions')
return PartitionerDasd(storage_provider)
elif table_type == 'msdos':
return PartitionerMsDos(storage_provider, start_sector)
elif 'arm' in host_architecture or host_architecture == 'aarch64':
if table_type == 'gpt':
return PartitionerGpt(storage_provider, start_sector)
elif table_type == 'msdos':
return PartitionerMsDos(storage_provider, start_sector)
raise KiwiPartitionerSetupError(
'Support for partitioner on %s architecture not implemented' %
host_architecture)
def setup(self):
disk_provider = Mock()
disk_provider.get_device = Mock(return_value='/dev/loop0')
self.partitioner = PartitionerMsDos(disk_provider)
self.partitioner_extended = PartitionerMsDos(disk_provider,
extended_layout=True)
def setup(self):
disk_provider = mock.Mock()
disk_provider.get_device = mock.Mock(return_value='/dev/loop0')
self.partitioner = PartitionerMsDos(disk_provider)
