#!/usr/bin/env python # vim:fileencoding=utf-8 # License: GPL v3 Copyright: 2016, Kovid Goyal from . import BaseTest from kitty.fast_data_types import DECAWM, IRM, Cursor, DECCOLM, DECOM class TestScreen(BaseTest): def test_draw_fast(self): s = self.create_screen() # Test in line-wrap, non-insert mode s.draw('a' * 5) self.ae(str(s.line(0)), 'a' * 5) self.ae(s.cursor.x, 5), self.ae(s.cursor.y, 0) self.assertChanges(s, ignore='cursor', cells={0: ((0, 4),)}) s.draw('b' * 7) self.assertTrue(s.linebuf.is_continued(1)) self.assertTrue(s.linebuf.is_continued(2)) self.ae(str(s.line(0)), 'a' * 5) self.ae(str(s.line(1)), 'b' * 5) self.ae(str(s.line(2)), 'b' * 2 + ' ' * 3) self.ae(s.cursor.x, 2), self.ae(s.cursor.y, 2) self.assertChanges(s, ignore='cursor', cells={1: ((0, 4),), 2: ((0, 1),)}) s.draw('c' * 15) self.ae(str(s.line(0)), 'b' * 5) self.ae(str(s.line(1)), 'bbccc') # Now test without line-wrap s.reset(), s.reset_dirty() s.reset_mode(DECAWM) s.draw('0123456789') self.ae(str(s.line(0)), '01239') self.ae(s.cursor.x, 5), self.ae(s.cursor.y, 0) self.assertChanges(s, ignore='cursor', cells={0: ((0, 4),)}) s.draw('ab') self.ae(str(s.line(0)), '0123b') self.ae(s.cursor.x, 5), self.ae(s.cursor.y, 0) self.assertChanges(s, ignore='cursor', cells={0: ((4, 4),)}) # Now test in insert mode s.reset(), s.reset_dirty() s.set_mode(IRM) s.draw('12345' * 5) s.cursor_back(5) self.ae(s.cursor.x, 0), self.ae(s.cursor.y, 4) s.reset_dirty() s.draw('ab') self.ae(str(s.line(4)), 'ab123') self.ae((s.cursor.x, s.cursor.y), (2, 4)) self.assertChanges(s, ignore='cursor', cells={4: ((0, 4),)}) def test_draw_char(self): # Test in line-wrap, non-insert mode s = self.create_screen() s.draw('ココx') self.ae(str(s.line(0)), 'ココx') self.ae(tuple(map(s.line(0).width, range(5))), (2, 0, 2, 0, 1)) self.ae(s.cursor.x, 5), self.ae(s.cursor.y, 0) self.assertChanges(s, ignore='cursor', cells={0: ((0, 4),)}) s.draw('ニチハ') self.ae(str(s.line(0)), 'ココx') self.ae(str(s.line(1)), 'ニチ ') self.ae(str(s.line(2)), 'ハ ') self.assertChanges(s, ignore='cursor', cells={1: ((0, 3),), 2: ((0, 1),)}) self.ae(s.cursor.x, 2), self.ae(s.cursor.y, 2) s.draw('Ƶ̧\u0308') self.ae(str(s.line(2)), 'ハƵ̧\u0308 ') self.ae(s.cursor.x, 3), self.ae(s.cursor.y, 2) self.assertChanges(s, ignore='cursor', cells={2: ((2, 2),)}) s.draw('xy'), s.draw('\u0306') self.ae(str(s.line(2)), 'ハƵ̧\u0308xy\u0306') self.ae(s.cursor.x, 5), self.ae(s.cursor.y, 2) self.assertChanges(s, ignore='cursor', cells={2: ((3, 4),)}) s.draw('c' * 15) self.ae(str(s.line(0)), 'ニチ ') # Now test without line-wrap s.reset(), s.reset_dirty() s.reset_mode(DECAWM) s.draw('0\u030612345\u03066789\u0306') self.ae(str(s.line(0)), '0\u03061239\u0306') self.ae(s.cursor.x, 5), self.ae(s.cursor.y, 0) self.assertChanges(s, ignore='cursor', cells={0: ((0, 4),)}) s.draw('ab\u0306') self.ae(str(s.line(0)), '0\u0306123b\u0306') self.ae(s.cursor.x, 5), self.ae(s.cursor.y, 0) self.assertChanges(s, ignore='cursor', cells={0: ((4, 4),)}) # Now test in insert mode s.reset(), s.reset_dirty() s.set_mode(IRM) s.draw('1\u03062345' * 5) s.cursor_back(5) self.ae(s.cursor.x, 0), self.ae(s.cursor.y, 4) s.reset_dirty() s.draw('a\u0306b') self.ae(str(s.line(4)), 'a\u0306b1\u030623') self.ae((s.cursor.x, s.cursor.y), (2, 4)) self.assertChanges(s, ignore='cursor', cells={4: ((0, 4),)}) def test_char_manipulation(self): s = self.create_screen() def init(): s.reset(), s.reset_dirty() s.draw('abcde') s.cursor.bold = True s.cursor_back(4) s.reset_dirty() self.ae(s.cursor.x, 1) init() s.insert_characters(2) self.ae(str(s.line(0)), 'a bc') self.assertTrue(s.line(0).cursor_from(1).bold) self.assertChanges(s, ignore='cursor', cells={0: ((1, 4),)}) s.cursor_back(1) s.insert_characters(20) self.ae(str(s.line(0)), ' ') self.assertChanges(s, ignore='cursor', cells={0: ((0, 4),)}) s.draw('xココ') s.cursor_back(5) s.reset_dirty() s.insert_characters(1) self.ae(str(s.line(0)), ' xコ ') self.assertChanges(s, ignore='cursor', cells={0: ((0, 4),)}) c = Cursor() c.italic = True s.line(0).apply_cursor(c, 0, 5) self.ae(s.line(0).width(2), 2) self.assertTrue(s.line(0).cursor_from(2).italic) self.assertFalse(s.line(0).cursor_from(2).bold) init() s.delete_characters(2) self.ae(str(s.line(0)), 'ade ') self.assertTrue(s.line(0).cursor_from(4).bold) self.assertFalse(s.line(0).cursor_from(2).bold) self.assertChanges(s, ignore='cursor', cells={0: ((1, 4),)}) init() s.erase_characters(2) self.ae(str(s.line(0)), 'a de') self.assertTrue(s.line(0).cursor_from(1).bold) self.assertFalse(s.line(0).cursor_from(4).bold) self.assertChanges(s, cells={0: ((1, 2),)}) s.erase_characters(20) self.ae(str(s.line(0)), 'a ') init() s.erase_in_line() self.ae(str(s.line(0)), 'a ') self.assertTrue(s.line(0).cursor_from(1).bold) self.assertFalse(s.line(0).cursor_from(0).bold) self.assertChanges(s, cells={0: ((1, 4),)}) init() s.erase_in_line(1) self.ae(str(s.line(0)), ' cde') self.assertChanges(s, cells={0: ((0, 1),)}) init() s.erase_in_line(2) self.ae(str(s.line(0)), ' ') self.assertChanges(s, cells={0: ((0, 4),)}) init() s.erase_in_line(2, True) self.ae((False, False, False, False, False), tuple(map(lambda i: s.line(0).cursor_from(i).bold, range(5)))) def test_erase_in_screen(self): s = self.create_screen() def init(): s.reset() s.draw('12345' * 5) s.reset_dirty() s.cursor.x, s.cursor.y = 2, 1 s.cursor.bold = True def all_lines(s): return tuple(str(s.line(i)) for i in range(s.lines)) init() s.erase_in_display() self.ae(all_lines(s), ('12345', '12 ', ' ', ' ', ' ')) self.assertChanges(s, lines={2, 3, 4}, cells={1: ((2, 4),)}) init() s.erase_in_display(1) self.ae(all_lines(s), (' ', ' 45', '12345', '12345', '12345')) self.assertChanges(s, lines={0}, cells={1: ((0, 2),)}) init() s.erase_in_display(2) self.ae(all_lines(s), (' ', ' ', ' ', ' ', ' ')) self.assertChanges(s, lines=set(range(5))) self.assertTrue(s.line(0).cursor_from(1).bold) init() s.erase_in_display(2, True) self.ae(all_lines(s), (' ', ' ', ' ', ' ', ' ')) self.assertChanges(s, lines=set(range(5))) self.assertFalse(s.line(0).cursor_from(1).bold) def test_cursor_movement(self): s = self.create_screen() s.draw('12345' * 5) s.reset_dirty() s.cursor_up(2) self.ae((s.cursor.x, s.cursor.y), (4, 2)) s.cursor_up1() self.ae((s.cursor.x, s.cursor.y), (0, 1)) s.cursor_forward(3) self.ae((s.cursor.x, s.cursor.y), (3, 1)) s.cursor_back() self.ae((s.cursor.x, s.cursor.y), (2, 1)) s.cursor_down() self.ae((s.cursor.x, s.cursor.y), (2, 2)) s.cursor_down1(5) self.ae((s.cursor.x, s.cursor.y), (0, 4)) s = self.create_screen() s.draw('12345' * 5) s.index() self.ae(str(s.line(4)), ' ' * 5) for i in range(4): self.ae(str(s.line(i)), '12345') s.draw('12345' * 5) s.cursor_up(5) s.reverse_index() self.ae(str(s.line(0)), ' ' * 5) for i in range(1, 5): self.ae(str(s.line(i)), '12345') def test_resize(self): s = self.create_screen(scrollback=6) s.draw(''.join([str(i) * s.columns for i in range(s.lines)])) s.resize(3, 10) self.ae(str(s.line(0)), '0'*5 + '1'*5) self.ae(str(s.line(1)), '2'*5 + '3'*5) self.ae(str(s.line(2)), '4'*5 + ' '*5) s.resize(5, 1) self.ae(str(s.line(0)), '4') hb = s.historybuf for i in range(s.lines): self.ae(str(hb.line(i)), '3') self.ae(str(hb.line(5)), '2') def test_tab_stops(self): # Taken from vttest/main.c s = self.create_screen(cols=80, lines=2) s.cursor_position(1, 1) s.clear_tab_stop(3) for col in range(1, s.columns - 1, 3): s.cursor_forward(3) s.set_tab_stop() s.cursor_position(1, 4) for col in range(4, s.columns - 1, 6): s.clear_tab_stop(0) s.cursor_forward(6) s.cursor_position(1, 7) s.clear_tab_stop(2) s.cursor_position(1, 1) for col in range(1, s.columns - 1, 6): s.tab() s.draw('*') s.cursor_position(2, 2) for col in range(2, s.columns - 1, 6): for i in range(5): s.draw(' ') s.draw('*') self.ae(str(s.line(0)), str(s.line(1))) def test_margins(self): # Taken from vttest/main.c s = self.create_screen(cols=80, lines=24) def nl(): s.carriage_return(), s.linefeed() for deccolm in (False, True): if deccolm: s.resize(24, 132) s.set_mode(DECCOLM) else: s.reset_mode(DECCOLM) region = s.lines - 6 s.set_margins(3, region + 3) s.set_mode(DECOM) for i in range(26): ch = chr(ord('A') + i) which = i % 4 if which == 0: s.cursor_position(region + 1, 1), s.draw(ch) s.cursor_position(region + 1, s.columns), s.draw(ch.lower()) nl() elif which == 1: # Simple wrapping s.cursor_position(region, s.columns), s.draw(chr(ord('A') + i - 1).lower() + ch) # Backspace at right margin s.cursor_position(region + 1, s.columns), s.draw(ch), s.backspace(), s.draw(ch.lower()) nl() elif which == 2: # Tab to right margin s.cursor_position(region + 1, s.columns), s.draw(ch), s.backspace(), s.backspace(), s.tab(), s.tab(), s.draw(ch.lower()) s.cursor_position(region + 1, 2), s.backspace(), s.draw(ch), nl() else: s.cursor_position(region + 1, 1), nl() s.cursor_position(region, 1), s.draw(ch) s.cursor_position(region, s.columns), s.draw(ch.lower()) for l in range(2, region + 2): c = chr(ord('I') + l - 2) self.ae(c + ' ' * (s.columns - 2) + c.lower(), str(s.line(l))) s.reset_mode(DECOM)