mirror of https://github.com/scrapy/scrapy.git
79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
from io import StringIO
|
|
|
|
from unittest import mock, TestCase
|
|
|
|
from scrapy.utils.display import pformat, pprint
|
|
|
|
|
|
class TestDisplay(TestCase):
|
|
object = {'a': 1}
|
|
colorized_string = (
|
|
"{\x1b[33m'\x1b[39;49;00m\x1b[33ma\x1b[39;49;00m\x1b[33m'"
|
|
"\x1b[39;49;00m: \x1b[34m1\x1b[39;49;00m}\n"
|
|
)
|
|
plain_string = "{'a': 1}"
|
|
|
|
@mock.patch('sys.platform', 'linux')
|
|
@mock.patch("sys.stdout.isatty")
|
|
def test_pformat(self, isatty):
|
|
isatty.return_value = True
|
|
self.assertEqual(pformat(self.object), self.colorized_string)
|
|
|
|
@mock.patch("sys.stdout.isatty")
|
|
def test_pformat_dont_colorize(self, isatty):
|
|
isatty.return_value = True
|
|
self.assertEqual(pformat(self.object, colorize=False), self.plain_string)
|
|
|
|
def test_pformat_not_tty(self):
|
|
self.assertEqual(pformat(self.object), self.plain_string)
|
|
|
|
@mock.patch('sys.platform', 'win32')
|
|
@mock.patch('platform.version')
|
|
@mock.patch("sys.stdout.isatty")
|
|
def test_pformat_old_windows(self, isatty, version):
|
|
isatty.return_value = True
|
|
version.return_value = '10.0.14392'
|
|
self.assertEqual(pformat(self.object), self.colorized_string)
|
|
|
|
@mock.patch('sys.platform', 'win32')
|
|
@mock.patch('scrapy.utils.display._enable_windows_terminal_processing')
|
|
@mock.patch('platform.version')
|
|
@mock.patch("sys.stdout.isatty")
|
|
def test_pformat_windows_no_terminal_processing(self, isatty, version, terminal_processing):
|
|
isatty.return_value = True
|
|
version.return_value = '10.0.14393'
|
|
terminal_processing.return_value = False
|
|
self.assertEqual(pformat(self.object), self.plain_string)
|
|
|
|
@mock.patch('sys.platform', 'win32')
|
|
@mock.patch('scrapy.utils.display._enable_windows_terminal_processing')
|
|
@mock.patch('platform.version')
|
|
@mock.patch("sys.stdout.isatty")
|
|
def test_pformat_windows(self, isatty, version, terminal_processing):
|
|
isatty.return_value = True
|
|
version.return_value = '10.0.14393'
|
|
terminal_processing.return_value = True
|
|
self.assertEqual(pformat(self.object), self.colorized_string)
|
|
|
|
@mock.patch('sys.platform', 'linux')
|
|
@mock.patch("sys.stdout.isatty")
|
|
def test_pformat_no_pygments(self, isatty):
|
|
isatty.return_value = True
|
|
|
|
import builtins
|
|
real_import = builtins.__import__
|
|
|
|
def mock_import(name, globals, locals, fromlist, level):
|
|
if 'pygments' in name:
|
|
raise ImportError
|
|
return real_import(name, globals, locals, fromlist, level)
|
|
|
|
builtins.__import__ = mock_import
|
|
self.assertEqual(pformat(self.object), self.plain_string)
|
|
builtins.__import__ = real_import
|
|
|
|
def test_pprint(self):
|
|
with mock.patch('sys.stdout', new=StringIO()) as mock_out:
|
|
pprint(self.object)
|
|
self.assertEqual(mock_out.getvalue(), "{'a': 1}\n")
|