bdb7123fa1
Use getLogger method instead of a global log object Also use caplog fixture to capture log messages in unit tests. This Fixes #1244
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
import logging
|
|
from pytest import fixture
|
|
from mock import patch
|
|
from kiwi.utils.output import DataOutput
|
|
import json
|
|
import mock
|
|
|
|
|
|
class TestDataOutput:
|
|
@fixture(autouse=True)
|
|
def inject_fixtures(self, caplog):
|
|
self._caplog = caplog
|
|
|
|
def setup(self):
|
|
test_data = {
|
|
'some-name': 'some-data'
|
|
}
|
|
self.expected_out = json.dumps(
|
|
test_data, sort_keys=True, indent=4, separators=(',', ': ')
|
|
)
|
|
self.out = DataOutput(test_data)
|
|
|
|
@patch('sys.stdout')
|
|
def test_display(self, mock_stdout):
|
|
self.out.display()
|
|
mock_stdout.write.assert_any_call(self.expected_out)
|
|
|
|
@patch('sys.stdout')
|
|
@patch('os.system')
|
|
@patch('kiwi.utils.output.NamedTemporaryFile')
|
|
def test_display_color(self, mock_temp, mock_system, mock_stdout):
|
|
out_file = mock.Mock()
|
|
out_file.name = 'tmpfile'
|
|
mock_temp.return_value = out_file
|
|
self.out.style = 'color'
|
|
self.out.color_json = True
|
|
self.out.display()
|
|
mock_system.assert_called_once_with(
|
|
'cat tmpfile | pjson'
|
|
)
|
|
|
|
@patch('sys.stdout')
|
|
def test_display_color_no_pjson(self, mock_stdout):
|
|
self.out.style = 'color'
|
|
self.out.color_json = False
|
|
with self._caplog.at_level(logging.WARNING):
|
|
self.out.display()
|
|
assert 'pjson for color output not installed' in self._caplog.text
|
|
assert 'run: pip install pjson' in self._caplog.text
|