68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
from mock import patch
|
|
|
|
import mock
|
|
|
|
from .test_helper import patch_open
|
|
|
|
from kiwi.partitioner.dasd import PartitionerDasd
|
|
|
|
|
|
class TestPartitionerDasd(object):
|
|
@patch('kiwi.partitioner.dasd.Command.run')
|
|
@patch('kiwi.partitioner.dasd.NamedTemporaryFile')
|
|
@patch_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)
|
|
|
|
@patch('kiwi.partitioner.dasd.Command.run')
|
|
@patch('kiwi.partitioner.dasd.NamedTemporaryFile')
|
|
@patch_open
|
|
@patch('kiwi.logger.log.debug')
|
|
def test_create(self, mock_debug, mock_open, mock_temp, mock_command):
|
|
mock_command.side_effect = Exception
|
|
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']
|
|
)
|
|
assert mock_debug.called
|
|
|
|
@patch('kiwi.partitioner.dasd.Command.run')
|
|
@patch('kiwi.partitioner.dasd.NamedTemporaryFile')
|
|
@patch_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'
|
|
)
|