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')
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')
class TestResult:
@fixture(autouse=True)
def inject_fixtures(self, caplog):
self._caplog = caplog
def setup(self):
self.xml_state = Mock()
self.result = Result(self.xml_state)
def test_add(self):
assert self.result.add('foo', 'bar') is None
result = self.result.get_results()
assert isinstance(result, dict)
assert result['foo'].filename == 'bar'
assert result['foo'].use_for_bundle is True
assert result['foo'].compress is False
def test_print_results_no_data(self):
assert self.result.print_results() is None
assert not self._caplog.text
def test_print_results_data(self):
assert self.result.add('foo', 'bar') is None
with self._caplog.at_level(logging.INFO):
self.result.print_results()
@patch('pickle.dump')
def test_dump(self, mock_pickle_dump):
m_open = mock_open()
with patch('builtins.open', m_open, create=True):
assert self.result.dump('kiwi.result') is None
m_open.assert_called_once_with('kiwi.result', 'wb')
mock_pickle_dump.assert_called_once_with(self.result,
m_open.return_value)
@patch('pickle.dump')
def test_dump_failed(self, mock_pickle_dump):
mock_pickle_dump.side_effect = Exception
with patch('builtins.open'):
with raises(KiwiResultError):
self.result.dump('kiwi.result')
@patch('pickle.load')
@patch('os.path.exists')
def test_load(self, mock_exists, mock_pickle_load):
mock_exists.return_value = True
m_open = mock_open()
mock_pickle_load.return_value = Result
with patch('builtins.open', m_open, create=True):
assert Result.load('kiwi.result') is Result
m_open.assert_called_once_with('kiwi.result', 'rb')
mock_pickle_load.assert_called_once_with(m_open.return_value)
@patch('os.path.exists')
def test_load_result_not_present(self, mock_exists):
mock_exists.return_value = False
with raises(KiwiResultError):
Result.load('kiwi.result')
@patch('pickle.load')
@patch('os.path.exists')
def test_load_failed(self, mock_exists, mock_pickle_load):
mock_exists.return_value = True
mock_pickle_load.side_effect = Exception
with raises(KiwiResultError):
Result.load('kiwi.result')
@patch('os.path.getsize')
def test_build_constraint(self, mock_getsize):
mock_getsize.return_value = 524288000
assert self.result.verify_image_size(524288000, 'foo') is None
@patch('os.path.getsize')
def test_build_constraint_failure(self, mock_getsize):
mock_getsize.return_value = 524288000
with raises(KiwiResultError):
self.result.verify_image_size(524287999, 'foo')
class TestResult:
def setup(self):
self.xml_state = 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')
def test_dump(self, mock_pickle_dump):
m_open = mock_open()
with patch('builtins.open', m_open, create=True):
self.result.dump('kiwi.result')
m_open.assert_called_once_with(
'kiwi.result', 'wb'
)
mock_pickle_dump.assert_called_once_with(
self.result, m_open.return_value
)
@patch('pickle.dump')
def test_dump_failed(self, mock_pickle_dump):
mock_pickle_dump.side_effect = Exception
with patch('builtins.open'):
with raises(KiwiResultError):
self.result.dump('kiwi.result')
@patch('pickle.load')
@patch('os.path.exists')
def test_load(self, mock_exists, mock_pickle_load):
mock_exists.return_value = True
m_open = mock_open()
with patch('builtins.open', m_open, create=True):
Result.load('kiwi.result')
m_open.assert_called_once_with(
'kiwi.result', 'rb'
)
mock_pickle_load.assert_called_once_with(
m_open.return_value
)
@patch('os.path.exists')
def test_load_result_not_present(self, mock_exists):
mock_exists.return_value = False
with raises(KiwiResultError):
Result.load('kiwi.result')
@patch('pickle.load')
@patch('os.path.exists')
def test_load_failed(self, mock_exists, mock_pickle_load):
mock_exists.return_value = True
mock_pickle_load.side_effect = Exception
with raises(KiwiResultError):
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')
def test_build_constraint_failure(self, mock_getsize):
mock_getsize.return_value = 524288000
with raises(KiwiResultError):
self.result.verify_image_size(524287999, 'foo')
