100 lines
3.1 KiB
Python
100 lines
3.1 KiB
Python
from mock import patch
|
|
|
|
import mock
|
|
|
|
from .test_helper import patch_open, raises
|
|
|
|
from kiwi.system.result import Result
|
|
from kiwi.exceptions import KiwiResultError
|
|
|
|
|
|
class TestResult(object):
|
|
def setup(self):
|
|
self.context_manager_mock = mock.MagicMock()
|
|
self.file_mock = mock.MagicMock()
|
|
self.enter_mock = mock.MagicMock()
|
|
self.exit_mock = mock.MagicMock()
|
|
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.xml_state = mock.Mock()
|
|
|
|
self.result = Result(self.xml_state)
|
|
|
|
def test_add(self):
|
|
self.result.add('foo', 'bar')
|
|
result = self.result.get_results()
|
|
assert result['foo'].filename == 'bar'
|
|
assert result['foo'].use_for_bundle is True
|
|
assert result['foo'].compress is False
|
|
|
|
@patch('kiwi.logger.log.info')
|
|
def test_print_results_no_data(self, mock_info):
|
|
self.result.print_results()
|
|
assert mock_info.called == 0
|
|
|
|
@patch('kiwi.logger.log.info')
|
|
def test_print_results_data(self, mock_info):
|
|
self.result.add('foo', 'bar')
|
|
self.result.print_results()
|
|
assert mock_info.called
|
|
|
|
@patch('pickle.dump')
|
|
@patch_open
|
|
def test_dump(self, mock_open, mock_pickle_dump):
|
|
mock_open.return_value = self.context_manager_mock
|
|
self.result.dump('kiwi.result')
|
|
mock_open.assert_called_once_with(
|
|
'kiwi.result', 'wb'
|
|
)
|
|
mock_pickle_dump.assert_called_once_with(
|
|
self.result, self.file_mock
|
|
)
|
|
|
|
@patch('pickle.dump')
|
|
@patch_open
|
|
@raises(KiwiResultError)
|
|
def test_dump_failed(self, mock_open, mock_pickle_dump):
|
|
mock_pickle_dump.side_effect = Exception
|
|
self.result.dump('kiwi.result')
|
|
|
|
@patch('pickle.load')
|
|
@patch('os.path.exists')
|
|
@patch_open
|
|
def test_load(self, mock_open, mock_exists, mock_pickle_load):
|
|
mock_open.return_value = self.context_manager_mock
|
|
mock_exists.return_value = True
|
|
Result.load('kiwi.result')
|
|
mock_open.assert_called_once_with(
|
|
'kiwi.result', 'rb'
|
|
)
|
|
mock_pickle_load.assert_called_once_with(
|
|
self.file_mock
|
|
)
|
|
|
|
@patch('os.path.exists')
|
|
@raises(KiwiResultError)
|
|
def test_load_result_not_present(self, mock_exists):
|
|
mock_exists.return_value = False
|
|
Result.load('kiwi.result')
|
|
|
|
@patch('pickle.load')
|
|
@patch('os.path.exists')
|
|
@raises(KiwiResultError)
|
|
def test_load_failed(self, mock_exists, mock_pickle_load):
|
|
mock_exists.return_value = True
|
|
mock_pickle_load.side_effect = Exception
|
|
Result.load('kiwi.result')
|
|
|
|
@patch('os.path.getsize')
|
|
def test_build_constraint(self, mock_getsize):
|
|
mock_getsize.return_value = 524288000
|
|
self.result.verify_image_size(524288000, 'foo')
|
|
|
|
@patch('os.path.getsize')
|
|
@raises(KiwiResultError)
|
|
def test_build_constraint_failure(self, mock_getsize):
|
|
mock_getsize.return_value = 524288000
|
|
self.result.verify_image_size(524287999, 'foo')
|