mirror of
https://github.com/scrapy/scrapy.git
synced 2025-02-06 11:00:46 +00:00
91 lines
3.1 KiB
Python
91 lines
3.1 KiB
Python
from io import StringIO
|
|
from unittest import TestCase, mock
|
|
|
|
from scrapy.utils.display import pformat, pprint
|
|
|
|
|
|
class TestDisplay(TestCase):
|
|
object = {"a": 1}
|
|
colorized_strings = {
|
|
(
|
|
(
|
|
"{\x1b[33m'\x1b[39;49;00m\x1b[33ma\x1b[39;49;00m\x1b[33m'"
|
|
"\x1b[39;49;00m: \x1b[34m1\x1b[39;49;00m}"
|
|
)
|
|
+ suffix
|
|
)
|
|
for suffix in (
|
|
# https://github.com/pygments/pygments/issues/2313
|
|
"\n", # pygments ≤ 2.13
|
|
"\x1b[37m\x1b[39;49;00m\n", # pygments ≥ 2.14
|
|
)
|
|
}
|
|
plain_string = "{'a': 1}"
|
|
|
|
@mock.patch("sys.platform", "linux")
|
|
@mock.patch("sys.stdout.isatty")
|
|
def test_pformat(self, isatty):
|
|
isatty.return_value = True
|
|
self.assertIn(pformat(self.object), self.colorized_strings)
|
|
|
|
@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.assertIn(pformat(self.object), self.colorized_strings)
|
|
|
|
@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.assertIn(pformat(self.object), self.colorized_strings)
|
|
|
|
@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")
|