bdb7123fa1
Use getLogger method instead of a global log object Also use caplog fixture to capture log messages in unit tests. This Fixes #1244
101 lines
2.9 KiB
Python
101 lines
2.9 KiB
Python
import logging
|
|
from mock import (
|
|
patch, Mock, mock_open
|
|
)
|
|
from pytest import (
|
|
raises, fixture
|
|
)
|
|
|
|
from kiwi.system.result import Result
|
|
|
|
from kiwi.exceptions import KiwiResultError
|
|
|
|
|
|
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):
|
|
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
|
|
|
|
def test_print_results_no_data(self):
|
|
self.result.print_results()
|
|
assert not self._caplog.text
|
|
|
|
def test_print_results_data(self):
|
|
self.result.add('foo', 'bar')
|
|
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):
|
|
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')
|