mirror of
https://github.com/kovidgoyal/kitty.git
synced 2024-11-10 13:04:03 +03:00
80 lines
2.7 KiB
Python
80 lines
2.7 KiB
Python
#!/usr/bin/env python
|
|
# vim:fileencoding=utf-8
|
|
# License: GPL v3 Copyright: 2016, Kovid Goyal <kovid at kovidgoyal.net>
|
|
|
|
from functools import partial
|
|
|
|
from . import BaseTest
|
|
|
|
from kitty.fast_data_types import parse_bytes, parse_bytes_dump
|
|
|
|
|
|
class CmdDump(list):
|
|
|
|
def __call__(self, *a):
|
|
if len(a) == 1:
|
|
a = a[0]
|
|
self.append(a)
|
|
|
|
|
|
class TestScreen(BaseTest):
|
|
|
|
def parse_buytes_dump(self, s, x, *cmds):
|
|
cd = CmdDump()
|
|
if isinstance(x, str):
|
|
x = x.encode('utf-8')
|
|
parse_bytes_dump(s, x, cd)
|
|
self.ae(tuple(cd), cmds)
|
|
|
|
def test_simple_parsing(self):
|
|
s = self.create_screen()
|
|
pb = partial(self.parse_buytes_dump, s)
|
|
|
|
pb('12')
|
|
self.ae(str(s.line(0)), '12 ')
|
|
pb('3456')
|
|
self.ae(str(s.line(0)), '12345')
|
|
self.ae(str(s.line(1)), '6 ')
|
|
pb(b'\n123\n\r45', ('screen_linefeed', ord('\n')), ('screen_linefeed', ord('\n')), ('screen_carriage_return', ord('\r')))
|
|
self.ae(str(s.line(1)), '6 ')
|
|
self.ae(str(s.line(2)), ' 123 ')
|
|
self.ae(str(s.line(3)), '45 ')
|
|
parse_bytes(s, b'\rabcde')
|
|
self.ae(str(s.line(3)), 'abcde')
|
|
parse_bytes(s, '\rßxyz1'.encode('utf-8'))
|
|
self.ae(str(s.line(3)), 'ßxyz1')
|
|
pb('ニチ '.encode('utf-8'))
|
|
self.ae(str(s.line(4)), 'ニチ ')
|
|
|
|
def test_esc_codes(self):
|
|
s = self.create_screen()
|
|
pb = partial(self.parse_buytes_dump, s)
|
|
pb('12\033Da', 'screen_index')
|
|
self.ae(str(s.line(0)), '12 ')
|
|
self.ae(str(s.line(1)), ' a ')
|
|
pb('\033x', 'Unknown char in escape_dispatch: %d' % ord('x'))
|
|
pb('\033c123', 'screen_reset')
|
|
self.ae(str(s.line(0)), '123 ')
|
|
|
|
def test_csi_codes(self):
|
|
s = self.create_screen()
|
|
pb = partial(self.parse_buytes_dump, s)
|
|
pb('abcde')
|
|
s.cursor_back(5)
|
|
pb('x\033[2@y', ('screen_insert_characters', 2))
|
|
self.ae(str(s.line(0)), 'xy bc')
|
|
pb('x\033[2;7@y', ('screen_insert_characters', 2))
|
|
pb('x\033[@y', ('screen_insert_characters', 1))
|
|
pb('x\033[345@y', ('screen_insert_characters', 345))
|
|
pb('x\033[345;@y', ('screen_insert_characters', 345))
|
|
pb('\033[H', ('screen_cursor_position', 1, 1))
|
|
self.ae(s.cursor.x, 0), self.ae(s.cursor.y, 0)
|
|
pb('\033[4H', ('screen_cursor_position', 4, 1))
|
|
pb('\033[3;2H', ('screen_cursor_position', 3, 2))
|
|
pb('\033[3;2;H', ('screen_cursor_position', 3, 2))
|
|
self.ae(s.cursor.x, 1), self.ae(s.cursor.y, 2)
|
|
pb('\033[J', ('screen_erase_in_display', 0, 0))
|
|
pb('\033[?J', ('screen_erase_in_display', 0, 1))
|
|
pb('\033[?2J', ('screen_erase_in_display', 2, 1))
|
|
s.reset()
|