def setup(self, mock_open, mock_temp, mock_command):
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)
mock_temp.return_value = self.tempfile
mock_open.return_value = self.context_manager_mock
disk_provider = mock.Mock()
disk_provider.get_device = mock.Mock(return_value='/dev/loop0')
self.partitioner = PartitionerDasd(disk_provider)
class TestPartitionerDasd(object):
@patch('kiwi.partitioner_dasd.Command.run')
@patch('kiwi.partitioner_dasd.NamedTemporaryFile')
@patch('__builtin__.open')
def setup(self, mock_open, mock_temp, mock_command):
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)
mock_temp.return_value = self.tempfile
mock_open.return_value = self.context_manager_mock
disk_provider = mock.Mock()
disk_provider.get_device = mock.Mock(return_value='/dev/loop0')
self.partitioner = PartitionerDasd(disk_provider)
def test_get_id(self):
assert self.partitioner.get_id() == 0
@patch('kiwi.partitioner_dasd.Command.run')
@patch('kiwi.partitioner_dasd.NamedTemporaryFile')
@patch('__builtin__.open')
def test_create(self, mock_open, mock_temp, mock_command):
mock_temp.return_value = self.tempfile
mock_open.return_value = self.context_manager_mock
self.partitioner.create('name', 100, 't.linux', ['f.active'])
self.file_mock.write.assert_called_once_with('n\np\n\n+100M\nw\nq\n')
mock_command.assert_called_once_with(
['bash', '-c', 'cat tempfile | fdasd -f /dev/loop0'])
@patch('kiwi.partitioner_dasd.Command.run')
@patch('kiwi.partitioner_dasd.NamedTemporaryFile')
@patch('__builtin__.open')
def test_create_all_free(self, mock_open, mock_temp, mock_command):
mock_temp.return_value = self.tempfile
mock_open.return_value = self.context_manager_mock
self.partitioner.create('name', 'all_free', 't.linux')
self.file_mock.write.assert_called_once_with('n\np\n\n\nw\nq\n')
def test_set_flag(self):
# This is a noop for fdasd, just does nothing but has to
# exist to complete the partitioner interface
self.partitioner.set_flag('id', 'flag_name')
def setup(self, mock_open, mock_temp, mock_command):
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)
mock_temp.return_value = self.tempfile
mock_open.return_value = self.context_manager_mock
disk_provider = mock.Mock()
disk_provider.get_device = mock.Mock(
return_value='/dev/loop0'
)
self.partitioner = PartitionerDasd(disk_provider)
class TestPartitionerDasd(object):
@patch('kiwi.partitioner_dasd.Command.run')
@patch('kiwi.partitioner_dasd.NamedTemporaryFile')
@patch('__builtin__.open')
def setup(self, mock_open, mock_temp, mock_command):
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)
mock_temp.return_value = self.tempfile
mock_open.return_value = self.context_manager_mock
disk_provider = mock.Mock()
disk_provider.get_device = mock.Mock(
return_value='/dev/loop0'
)
self.partitioner = PartitionerDasd(disk_provider)
def test_get_id(self):
assert self.partitioner.get_id() == 0
@patch('kiwi.partitioner_dasd.Command.run')
@patch('kiwi.partitioner_dasd.NamedTemporaryFile')
@patch('__builtin__.open')
def test_create(self, mock_open, mock_temp, mock_command):
mock_temp.return_value = self.tempfile
mock_open.return_value = self.context_manager_mock
self.partitioner.create('name', 100, 't.linux', ['f.active'])
self.file_mock.write.assert_called_once_with(
'n\np\n\n+100M\nw\nq\n'
)
mock_command.assert_called_once_with(
['bash', '-c', 'cat tempfile | fdasd -f /dev/loop0']
)
@patch('kiwi.partitioner_dasd.Command.run')
@patch('kiwi.partitioner_dasd.NamedTemporaryFile')
@patch('__builtin__.open')
def test_create_all_free(
self, mock_open, mock_temp, mock_command
):
mock_temp.return_value = self.tempfile
mock_open.return_value = self.context_manager_mock
self.partitioner.create('name', 'all_free', 't.linux')
self.file_mock.write.assert_called_once_with(
'n\np\n\n\nw\nq\n'
)
def test_set_flag(self):
# This is a noop for fdasd, just does nothing but has to
# exist to complete the partitioner interface
self.partitioner.set_flag('id', 'flag_name')