|
|
# Standard library imports
|
|
|
import unittest
|
|
|
|
|
|
# Local imports
|
|
|
from IPython.frontend.qt.console.ansi_code_processor import AnsiCodeProcessor
|
|
|
|
|
|
|
|
|
class TestAnsiCodeProcessor(unittest.TestCase):
|
|
|
|
|
|
def setUp(self):
|
|
|
self.processor = AnsiCodeProcessor()
|
|
|
|
|
|
def test_clear(self):
|
|
|
""" Do control sequences for clearing the console work?
|
|
|
"""
|
|
|
string = '\x1b[2J\x1b[K'
|
|
|
i = -1
|
|
|
for i, substring in enumerate(self.processor.split_string(string)):
|
|
|
if i == 0:
|
|
|
self.assertEqual(len(self.processor.actions), 1)
|
|
|
action = self.processor.actions[0]
|
|
|
self.assertEqual(action.action, 'erase')
|
|
|
self.assertEqual(action.area, 'screen')
|
|
|
self.assertEqual(action.erase_to, 'all')
|
|
|
elif i == 1:
|
|
|
self.assertEqual(len(self.processor.actions), 1)
|
|
|
action = self.processor.actions[0]
|
|
|
self.assertEqual(action.action, 'erase')
|
|
|
self.assertEqual(action.area, 'line')
|
|
|
self.assertEqual(action.erase_to, 'end')
|
|
|
else:
|
|
|
self.fail('Too many substrings.')
|
|
|
self.assertEqual(i, 1, 'Too few substrings.')
|
|
|
|
|
|
def test_colors(self):
|
|
|
""" Do basic controls sequences for colors work?
|
|
|
"""
|
|
|
string = 'first\x1b[34mblue\x1b[0mlast'
|
|
|
i = -1
|
|
|
for i, substring in enumerate(self.processor.split_string(string)):
|
|
|
if i == 0:
|
|
|
self.assertEqual(substring, 'first')
|
|
|
self.assertEqual(self.processor.foreground_color, None)
|
|
|
elif i == 1:
|
|
|
self.assertEqual(substring, 'blue')
|
|
|
self.assertEqual(self.processor.foreground_color, 4)
|
|
|
elif i == 2:
|
|
|
self.assertEqual(substring, 'last')
|
|
|
self.assertEqual(self.processor.foreground_color, None)
|
|
|
else:
|
|
|
self.fail('Too many substrings.')
|
|
|
self.assertEqual(i, 2, 'Too few substrings.')
|
|
|
|
|
|
def test_colors_xterm(self):
|
|
|
""" Do xterm-specific control sequences for colors work?
|
|
|
"""
|
|
|
string = '\x1b]4;20;rgb:ff/ff/ff\x1b' \
|
|
|
'\x1b]4;25;rgbi:1.0/1.0/1.0\x1b'
|
|
|
substrings = list(self.processor.split_string(string))
|
|
|
desired = { 20 : (255, 255, 255),
|
|
|
25 : (255, 255, 255) }
|
|
|
self.assertEqual(self.processor.color_map, desired)
|
|
|
|
|
|
string = '\x1b[38;5;20m\x1b[48;5;25m'
|
|
|
substrings = list(self.processor.split_string(string))
|
|
|
self.assertEqual(self.processor.foreground_color, 20)
|
|
|
self.assertEqual(self.processor.background_color, 25)
|
|
|
|
|
|
def test_scroll(self):
|
|
|
""" Do control sequences for scrolling the buffer work?
|
|
|
"""
|
|
|
string = '\x1b[5S\x1b[T'
|
|
|
i = -1
|
|
|
for i, substring in enumerate(self.processor.split_string(string)):
|
|
|
if i == 0:
|
|
|
self.assertEqual(len(self.processor.actions), 1)
|
|
|
action = self.processor.actions[0]
|
|
|
self.assertEqual(action.action, 'scroll')
|
|
|
self.assertEqual(action.dir, 'up')
|
|
|
self.assertEqual(action.unit, 'line')
|
|
|
self.assertEqual(action.count, 5)
|
|
|
elif i == 1:
|
|
|
self.assertEqual(len(self.processor.actions), 1)
|
|
|
action = self.processor.actions[0]
|
|
|
self.assertEqual(action.action, 'scroll')
|
|
|
self.assertEqual(action.dir, 'down')
|
|
|
self.assertEqual(action.unit, 'line')
|
|
|
self.assertEqual(action.count, 1)
|
|
|
else:
|
|
|
self.fail('Too many substrings.')
|
|
|
self.assertEqual(i, 1, 'Too few substrings.')
|
|
|
|
|
|
def test_formfeed(self):
|
|
|
""" Are formfeed characters processed correctly?
|
|
|
"""
|
|
|
string = '\f' # form feed
|
|
|
self.assertEqual(list(self.processor.split_string(string)), [''])
|
|
|
self.assertEqual(len(self.processor.actions), 1)
|
|
|
action = self.processor.actions[0]
|
|
|
self.assertEqual(action.action, 'scroll')
|
|
|
self.assertEqual(action.dir, 'down')
|
|
|
self.assertEqual(action.unit, 'page')
|
|
|
self.assertEqual(action.count, 1)
|
|
|
|
|
|
def test_carriage_return(self):
|
|
|
""" Are carriage return characters processed correctly?
|
|
|
"""
|
|
|
string = 'foo\rbar' # carriage return
|
|
|
splits = []
|
|
|
actions = []
|
|
|
for split in self.processor.split_string(string):
|
|
|
splits.append(split)
|
|
|
actions.append([action.action for action in self.processor.actions])
|
|
|
self.assertEqual(splits, ['foo', None, 'bar'])
|
|
|
self.assertEqual(actions, [[], ['carriage-return'], []])
|
|
|
|
|
|
def test_carriage_return_newline(self):
|
|
|
"""transform CRLF to LF"""
|
|
|
string = 'foo\rbar\r\ncat\r\n\n' # carriage return and newline
|
|
|
# only one CR action should occur, and '\r\n' should transform to '\n'
|
|
|
splits = []
|
|
|
actions = []
|
|
|
for split in self.processor.split_string(string):
|
|
|
splits.append(split)
|
|
|
actions.append([action.action for action in self.processor.actions])
|
|
|
self.assertEqual(splits, ['foo', None, 'bar', '\r\n', 'cat', '\r\n', '\n'])
|
|
|
self.assertEqual(actions, [[], ['carriage-return'], [], ['newline'], [], ['newline'], ['newline']])
|
|
|
|
|
|
def test_beep(self):
|
|
|
""" Are beep characters processed correctly?
|
|
|
"""
|
|
|
string = 'foo\abar' # bell
|
|
|
splits = []
|
|
|
actions = []
|
|
|
for split in self.processor.split_string(string):
|
|
|
splits.append(split)
|
|
|
actions.append([action.action for action in self.processor.actions])
|
|
|
self.assertEqual(splits, ['foo', None, 'bar'])
|
|
|
self.assertEqual(actions, [[], ['beep'], []])
|
|
|
|
|
|
def test_backspace(self):
|
|
|
""" Are backspace characters processed correctly?
|
|
|
"""
|
|
|
string = 'foo\bbar' # backspace
|
|
|
splits = []
|
|
|
actions = []
|
|
|
for split in self.processor.split_string(string):
|
|
|
splits.append(split)
|
|
|
actions.append([action.action for action in self.processor.actions])
|
|
|
self.assertEqual(splits, ['foo', None, 'bar'])
|
|
|
self.assertEqual(actions, [[], ['backspace'], []])
|
|
|
|
|
|
def test_combined(self):
|
|
|
""" Are CR and BS characters processed correctly in combination?
|
|
|
|
|
|
BS is treated as a change in print position, rather than a
|
|
|
backwards character deletion. Therefore a BS at EOL is
|
|
|
effectively ignored.
|
|
|
"""
|
|
|
string = 'abc\rdef\b' # CR and backspace
|
|
|
splits = []
|
|
|
actions = []
|
|
|
for split in self.processor.split_string(string):
|
|
|
splits.append(split)
|
|
|
actions.append([action.action for action in self.processor.actions])
|
|
|
self.assertEqual(splits, ['abc', None, 'def', None])
|
|
|
self.assertEqual(actions, [[], ['carriage-return'], [], ['backspace']])
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
unittest.main()
|
|
|
|
