12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283 |
- #!/usr/bin/env python
- # License: GPL v3 Copyright: 2016, Kovid Goyal <kovid at kovidgoyal.net>
- from kitty.config import defaults
- from kitty.fast_data_types import DECAWM, DECCOLM, DECOM, IRM, VT_PARSER_BUFFER_SIZE, Color, ColorProfile, Cursor
- from kitty.marks import marker_from_function, marker_from_regex
- from kitty.rgb import color_names
- from kitty.window import pagerhist
- from . import BaseTest, parse_bytes
- 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)
- 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)
- self.ae(s.cursor.x, 2), self.ae(s.cursor.y, 2)
- 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)
- s.draw('ab')
- self.ae(str(s.line(0)), '0123b')
- self.ae(s.cursor.x, 5), self.ae(s.cursor.y, 0)
- # 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))
- 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)
- s.draw('ニチハ')
- self.ae(str(s.line(0)), 'ココx')
- self.ae(str(s.line(1)), 'ニチ')
- self.ae(str(s.line(2)), 'ハ')
- 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)
- 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)
- 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)
- 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)
- # 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))
- def test_rep(self):
- s = self.create_screen()
- s.draw('a')
- parse_bytes(s, b'\x1b[b')
- self.ae(str(s.line(0)), 'aa')
- parse_bytes(s, b'\x1b[3b')
- self.ae(str(s.line(0)), 'a'*5)
- s.draw(' ')
- parse_bytes(s, b'\x1b[3b')
- self.ae(str(s.line(1)), ' '*4)
- def test_emoji_skin_tone_modifiers(self):
- s = self.create_screen()
- q = chr(0x1f469) + chr(0x1f3fd)
- s.draw(q)
- self.ae(str(s.line(0)), q)
- self.ae(s.cursor.x, 2)
- def test_regional_indicators(self):
- s = self.create_screen()
- flag = '\U0001f1ee\U0001f1f3'
- s.draw(flag)
- self.ae(str(s.line(0)), flag)
- self.ae(s.cursor.x, 2)
- s = self.create_screen()
- s.draw('a'), s.draw(flag[0]), s.draw('b')
- self.ae(str(s.line(0)), 'a' + flag[0] + 'b')
- self.ae(s.cursor.x, 4)
- def test_zwj(self):
- s = self.create_screen(cols=20)
- q = '\U0001f468\u200d\U0001f469\u200d\U0001f467\u200d\U0001f466'
- s.draw(q)
- self.ae(q, str(s.line(0)))
- self.ae(s.cursor.x, 8)
- for x in '\u200b\u200c\u200d':
- s = self.create_screen()
- q = f'X{x}Y'
- s.draw(q)
- self.ae(q, str(s.line(0)))
- self.ae(s.cursor.x, 2)
- 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)
- s.cursor_back(1)
- s.insert_characters(20)
- self.ae(str(s.line(0)), '')
- s.draw('xココ')
- s.cursor_back(5)
- s.reset_dirty()
- s.insert_characters(1)
- self.ae(str(s.line(0)), ' xコ')
- 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)
- s = self.create_screen()
- s.set_margins(1, 2)
- s.cursor.y = 3
- s.draw('abcde')
- s.cursor.x = 0
- s.delete_characters(2)
- self.ae('cde', str(s.line(s.cursor.y)))
- 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)
- 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)
- init()
- s.erase_in_line(1)
- self.ae(str(s.line(0)), ' cde')
- init()
- s.erase_in_line(2)
- self.ae(str(s.line(0)), '')
- 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
- self.ae(continuations(s), (True, True, True, True, False))
- def all_lines(s):
- return tuple(str(s.line(i)) for i in range(s.lines))
- def continuations(s):
- return tuple(s.line(i).last_char_has_wrapped_flag() for i in range(s.lines))
- init()
- s.erase_in_display(0)
- self.ae(all_lines(s), ('12345', '12', '', '', ''))
- self.ae(continuations(s), (True, False, False, False, False))
- init()
- s.erase_in_display(1)
- self.ae(all_lines(s), ('', ' 45', '12345', '12345', '12345'))
- self.ae(continuations(s), (False, True, True, True, False))
- init()
- s.erase_in_display(2)
- self.ae(all_lines(s), ('', '', '', '', ''))
- self.assertTrue(s.line(0).cursor_from(1).bold)
- self.ae(continuations(s), (False, False, False, False, False))
- init()
- s.erase_in_display(2, True)
- self.ae(all_lines(s), ('', '', '', '', ''))
- self.ae(continuations(s), (False, False, False, False, False))
- 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)), '')
- 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)), '')
- for i in range(1, 5):
- self.ae(str(s.line(i)), '12345')
- def test_backspace_wide_characters(self):
- s = self.create_screen()
- s.draw('⛅')
- self.ae(s.cursor.x, 2)
- s.backspace()
- s.draw(' ')
- s.backspace()
- self.ae(s.cursor.x, 1)
- 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)
- s.resize(5, 1)
- self.ae(str(s.line(0)), '4')
- self.ae(str(s.historybuf), '3\n3\n3\n3\n3\n2')
- s = self.create_screen(scrollback=20)
- s.draw(''.join(str(i) * s.columns for i in range(s.lines*2)))
- self.ae(str(s.linebuf), '55555\n66666\n77777\n88888\n99999')
- s.resize(5, 2)
- self.ae(str(s.linebuf), '88\n88\n99\n99\n9')
- s = self.create_screen()
- s.draw('a' * s.columns)
- s.linefeed(), s.carriage_return()
- s.draw('bb')
- s.resize(s.lines, s.columns - 2)
- self.ae(str(s.linebuf), 'aaa\naa\nbb\n\n')
- s.cursor.y = s.cursor.x = 0
- s.draw('x' * len(str(s.line(0))))
- s.linefeed(), s.carriage_return()
- s.draw('x' * len(str(s.line(1))))
- s.resize(s.lines, s.columns + 4)
- self.ae(str(s.linebuf), 'xxx\nxx\nbb\n\n')
- s = self.create_screen()
- c = s.callbacks
- parse_bytes(s, b'\x1b[?2048$p') # ]
- self.ae(c.wtcbuf, b'\x1b[?2048;2$y') # ]
- c.clear()
- parse_bytes(s, b'\x1b[?2048h\x1b[?2048$p') # ]]
- self.ae(c.wtcbuf, b'\x1b[?2048;1$y') # ]
- self.ae(c.num_of_resize_events, 1)
- parse_bytes(s, b'\x1b[?2048h') # ]
- self.ae(c.num_of_resize_events, 2)
- def test_cursor_after_resize(self):
- def draw(text, end_line=True):
- s.draw(text)
- if end_line:
- s.linefeed(), s.carriage_return()
- s = self.create_screen()
- draw('123'), draw('123')
- y_before = s.cursor.y
- s.resize(s.lines, s.columns-1)
- self.ae(y_before, s.cursor.y)
- s = self.create_screen(cols=5, lines=8)
- draw('one')
- draw('two three four five |||', end_line=False)
- s.resize(s.lines + 2, s.columns + 2)
- y = s.cursor.y
- self.assertIn('|', str(s.line(y)))
- s = self.create_screen()
- draw('a')
- x_before = s.cursor.x
- s.resize(s.lines - 1, s.columns)
- self.ae(x_before, s.cursor.x)
- s = self.create_screen()
- s.draw('abc')
- b = s.cursor.x
- s.resize(7, s.columns)
- self.assertEqual(s.cursor.x, b)
- s.cursor.x = 0
- s.resize(5, s.columns)
- self.assertEqual(s.cursor.x, 0)
- def test_scrollback_fill_after_resize(self):
- def prepare_screen(content=()):
- ans = self.create_screen(options={'scrollback_fill_enlarged_window': True})
- for line in content:
- ans.draw(line)
- ans.linefeed()
- ans.carriage_return()
- return ans
- def assert_lines(*lines):
- return self.ae(lines, tuple(str(s.line(i)) for i in range(s.lines)))
- # test the reverse scroll function
- s = prepare_screen(map(str, range(6)))
- assert_lines('2', '3', '4', '5', '')
- s.reverse_scroll(2, True)
- assert_lines('0', '1', '2', '3', '4')
- # Height increased, width unchanged → pull down lines to fill new space at the top
- s = prepare_screen(map(str, range(6)))
- assert_lines('2', '3', '4', '5', '')
- dist_from_bottom = s.lines - s.cursor.y
- s.resize(7, s.columns)
- assert_lines('0', '1', '2', '3', '4', '5', '')
- self.ae(dist_from_bottom, s.lines - s.cursor.y)
- # Height increased, width increased → rewrap, pull down
- s = prepare_screen(['0', '1', '2', '3' * 15])
- assert_lines('2', '33333', '33333', '33333', '')
- s.resize(7, 12)
- assert_lines('0', '1', '2', '333333333333', '333', '', '')
- # Height increased, width decreased → rewrap, pull down if possible
- s = prepare_screen(['0', '1', '2', '3' * 5])
- assert_lines('0', '1', '2', '33333', '')
- s.resize(6, 4)
- assert_lines('0', '1', '2', '3333', '3', '')
- # Height unchanged, width increased → rewrap, pull down if possible
- s = prepare_screen(['0', '1', '2', '3' * 15])
- assert_lines('2', '33333', '33333', '33333', '')
- s.resize(s.lines, 12)
- assert_lines('1', '2', '333333333333', '333', '')
- # Height decreased, width increased → rewrap, pull down if possible
- s = prepare_screen(['0', '1', '2', '3' * 15])
- assert_lines('2', '33333', '33333', '33333', '')
- s.resize(4, 12)
- assert_lines('2', '333333333333', '333', '')
- # Height increased with large continued text
- s = self.create_screen(options={'scrollback_fill_enlarged_window': True})
- s.draw(('x' * (s.columns * s.lines * 2)) + 'abcde')
- s.carriage_return(), s.linefeed()
- s.draw('>')
- assert_lines('xxxxx', 'xxxxx', 'xxxxx', 'abcde', '>')
- s.resize(s.lines + 2, s.columns)
- assert_lines('xxxxx', 'xxxxx', 'xxxxx', 'xxxxx', 'xxxxx', 'abcde', '>')
- 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)
- self.ae(str(s.line(0)), '\t*'*13)
- s = self.create_screen(cols=4, lines=2)
- s.draw('aaaX\tbbbb')
- self.ae(str(s.line(0)) + str(s.line(1)), 'aaaXbbbb')
- 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 ln in range(2, region + 2):
- c = chr(ord('I') + ln - 2)
- before = '\t' if ln % 4 == 0 else ' '
- self.ae(c + ' ' * (s.columns - 3) + before + c.lower(), str(s.line(ln)))
- s.reset_mode(DECOM)
- # Test that moving cursor outside the margins works as expected
- s = self.create_screen(10, 10)
- s.set_margins(4, 6)
- s.cursor_position(0, 0)
- self.ae(s.cursor.y, 0)
- nl()
- self.ae(s.cursor.y, 1)
- s.cursor.y = s.lines - 1
- self.ae(s.cursor.y, 9)
- s.reverse_index()
- self.ae(s.cursor.y, 8)
- def test_sgr(self):
- s = self.create_screen()
- s.select_graphic_rendition(0, 1, 37, 42)
- s.draw('a')
- c = s.line(0).cursor_from(0)
- self.assertTrue(c.bold)
- self.ae(c.bg, (2 << 8) | 1)
- s.cursor_position(2, 1)
- s.select_graphic_rendition(0, 35)
- s.draw('b')
- c = s.line(1).cursor_from(0)
- self.ae(c.fg, (5 << 8) | 1)
- self.ae(c.bg, 0)
- s.cursor_position(2, 2)
- s.select_graphic_rendition(38, 2, 99, 1, 2, 3)
- s.draw('c')
- c = s.line(1).cursor_from(1)
- self.ae(c.fg, (1 << 24) | (2 << 16) | (3 << 8) | 2)
- def test_cursor_hidden(self):
- s = self.create_screen()
- s.toggle_alt_screen()
- s.cursor_visible = False
- s.toggle_alt_screen()
- self.assertFalse(s.cursor_visible)
- def test_dirty_lines(self):
- s = self.create_screen()
- self.assertFalse(s.linebuf.dirty_lines())
- s.draw('a' * (s.columns * 2))
- self.ae(s.linebuf.dirty_lines(), [0, 1])
- self.assertFalse(s.historybuf.dirty_lines())
- while not s.historybuf.count:
- s.draw('a' * (s.columns * 2))
- self.ae(s.historybuf.dirty_lines(), list(range(s.historybuf.count)))
- self.ae(s.linebuf.dirty_lines(), list(range(s.lines)))
- s.cursor.x, s.cursor.y = 0, 1
- s.insert_lines(2)
- self.ae(s.linebuf.dirty_lines(), [0, 3, 4])
- s.draw('a' * (s.columns * s.lines))
- self.ae(s.linebuf.dirty_lines(), list(range(s.lines)))
- s.cursor.x, s.cursor.y = 0, 1
- s.delete_lines(2)
- self.ae(s.linebuf.dirty_lines(), [0, 1, 2])
- s = self.create_screen()
- self.assertFalse(s.linebuf.dirty_lines())
- s.erase_in_line(0, False)
- self.ae(s.linebuf.dirty_lines(), [0])
- s.index(), s.index()
- s.erase_in_display(0, False)
- self.ae(s.linebuf.dirty_lines(), [0, 2, 3, 4])
- s = self.create_screen()
- self.assertFalse(s.linebuf.dirty_lines())
- s.insert_characters(2)
- self.ae(s.linebuf.dirty_lines(), [0])
- s.cursor.y = 1
- s.delete_characters(2)
- self.ae(s.linebuf.dirty_lines(), [0, 1])
- s.cursor.y = 2
- s.erase_characters(2)
- self.ae(s.linebuf.dirty_lines(), [0, 1, 2])
- def test_selection_as_text(self):
- s = self.create_screen()
- for i in range(2 * s.lines):
- if i != 0:
- s.carriage_return(), s.linefeed()
- s.draw(str(i) * s.columns)
- s.start_selection(0, 0)
- s.update_selection(4, 4)
- def ts(*args):
- return ''.join(s.text_for_selection(*args))
- expected = ''.join(('55555', '\n66666', '\n77777', '\n88888', '\n99999'))
- self.ae(ts(), expected)
- s.scroll(2, True)
- self.ae(ts(), expected)
- s.reset()
- s.draw('ab cd')
- s.start_selection(0, 0)
- s.update_selection(1, 3)
- self.ae(ts(), ''.join(('ab ', 'cd')))
- self.ae(ts(False, True), ''.join(('ab', 'cd')))
- s.reset()
- s.draw('ab cd')
- s.start_selection(0, 0)
- s.update_selection(3, 4)
- self.ae(s.text_for_selection(), ('ab ', ' ', 'cd'))
- self.ae(s.text_for_selection(False, True), ('ab', '\n', 'cd'))
- s.reset()
- s.draw('a')
- s.select_graphic_rendition(32)
- s.draw('b')
- s.select_graphic_rendition(39)
- s.draw('c xy')
- s.start_selection(0, 0)
- s.update_selection(1, 3)
- self.ae(s.text_for_selection(), ('abc ', 'xy'))
- self.ae(s.text_for_selection(True), ('a\x1b[32mb\x1b[39mc ', 'xy', '\x1b[m'))
- self.ae(s.text_for_selection(True, True), ('a\x1b[32mb\x1b[39mc', 'xy', '\x1b[m'))
- def test_soft_hyphen(self):
- s = self.create_screen()
- s.draw('a\u00adb')
- self.ae(s.cursor.x, 2)
- s.start_selection(0, 0)
- s.update_selection(2, 0)
- self.ae(s.text_for_selection(), ('a\u00adb',))
- def test_variation_selectors(self):
- s = self.create_screen()
- s.draw('\U0001f610')
- self.ae(s.cursor.x, 2)
- s.carriage_return(), s.linefeed()
- s.draw('\U0001f610\ufe0e')
- self.ae(s.cursor.x, 1)
- s.carriage_return(), s.linefeed()
- s.draw('\u25b6')
- self.ae(s.cursor.x, 1)
- s.carriage_return(), s.linefeed()
- s.draw('\u25b6\ufe0f')
- self.ae(s.cursor.x, 2)
- def test_writing_with_cursor_on_trailer_of_wide_character(self):
- s = self.create_screen()
- def r(x, pos, expected):
- s.reset()
- s.draw('😸')
- self.ae(s.cursor.x, 2)
- s.cursor.x = 1
- s.draw(x)
- self.ae(s.cursor.x, pos)
- self.ae(str(s.line(0)), expected)
- r('a', 2, ' a')
- r('😸', 3, ' 😸')
- r('\u0304', 1, '😸\u0304')
- r('\r', 0, '😸')
- def test_serialize(self):
- from kitty.window import as_text
- s = self.create_screen()
- parse_bytes(s, b'\x1b[1;91m')
- s.draw('X')
- parse_bytes(s, b'\x1b[0m\x1b[2m')
- s.draw('Y')
- self.ae(as_text(s, True), '\x1b[m\x1b[22;1;91mX\x1b[22;2;39mY\n\n\n\n')
- s.reset()
- s.draw('ab' * s.columns)
- s.carriage_return(), s.linefeed()
- s.draw('c')
- self.ae(as_text(s), 'ababababab\nc\n\n')
- self.ae(as_text(s, True), '\x1b[mababa\x1b[mbabab\n\x1b[mc\n\n')
- s = self.create_screen(cols=2, lines=2, scrollback=2)
- for i in range(1, 7):
- s.select_graphic_rendition(30 + i)
- s.draw(f'{i}' * s.columns)
- self.ae(as_text(s, True, True), '\x1b[m\x1b[31m11\x1b[m\x1b[32m22\x1b[m\x1b[33m33\x1b[m\x1b[34m44\x1b[m\x1b[m\x1b[35m55\x1b[m\x1b[36m66')
- def set_link(url=None, id=None):
- parse_bytes(s, '\x1b]8;id={};{}\x1b\\'.format(id or '', url or '').encode('utf-8'))
- s = self.create_screen()
- s.draw('a')
- set_link('moo', 'foo')
- s.draw('bcdef')
- self.ae(as_text(s, True), '\x1b[ma\x1b]8;id=foo;moo\x1b\\bcde\x1b[mf\n\n\n\x1b]8;;\x1b\\')
- set_link()
- s.draw('gh')
- self.ae(as_text(s, True), '\x1b[ma\x1b]8;id=foo;moo\x1b\\bcde\x1b[mf\x1b]8;;\x1b\\gh\n\n\n')
- s = self.create_screen()
- s.draw('a')
- set_link('moo')
- s.draw('bcdef')
- self.ae(as_text(s, True), '\x1b[ma\x1b]8;;moo\x1b\\bcde\x1b[mf\n\n\n\x1b]8;;\x1b\\')
- def test_wrapping_serialization(self):
- from kitty.window import as_text
- s = self.create_screen(cols=2, lines=2, scrollback=2, options={'scrollback_pager_history_size': 128})
- s.draw('aabbccddeeff')
- self.ae(as_text(s, add_history=True), 'aabbccddeeff')
- self.assertNotIn('\n', as_text(s, add_history=True, as_ansi=True))
- s = self.create_screen(cols=2, lines=2, scrollback=2, options={'scrollback_pager_history_size': 128})
- s.draw('1'), s.carriage_return(), s.linefeed()
- s.draw('2'), s.carriage_return(), s.linefeed()
- s.draw('3'), s.carriage_return(), s.linefeed()
- s.draw('4'), s.carriage_return(), s.linefeed()
- s.draw('5'), s.carriage_return(), s.linefeed()
- s.draw('6'), s.carriage_return(), s.linefeed()
- s.draw('7')
- self.ae(as_text(s, add_history=True), '1\n2\n3\n4\n5\n6\n7')
- s = self.create_screen(cols=2, lines=2, scrollback=2, options={'scrollback_pager_history_size': 128})
- s.draw('aabb')
- s.cursor.y = 0
- s.carriage_return(), s.linefeed()
- self.ae(as_text(s, add_history=True), 'aabb')
- s = self.create_screen(cols=2, lines=2, scrollback=2, options={'scrollback_pager_history_size': 128})
- s.draw('a'), s.carriage_return(), s.linefeed()
- s.cursor.y = 0
- s.draw('aabb')
- self.ae(as_text(s), 'aabb')
- s = self.create_screen(cols=2, lines=2, scrollback=2, options={'scrollback_pager_history_size': 128})
- s.draw('a😀')
- self.ae(as_text(s), 'a😀')
- def test_pagerhist(self):
- hsz = 8
- s = self.create_screen(cols=2, lines=2, scrollback=2, options={'scrollback_pager_history_size': hsz})
- def contents():
- return s.historybuf.pagerhist_as_text()
- def line(i):
- q.append('\x1b[m' + f'{i}' * s.columns + '\r')
- def w(x):
- s.historybuf.pagerhist_write(x)
- def test():
- expected = ''.join(q)
- maxlen = hsz
- extra = len(expected) - maxlen
- if extra > 0:
- expected = expected[extra:]
- got = contents()
- self.ae(got, expected)
- q = []
- for i in range(4):
- s.draw(f'{i}' * s.columns)
- self.ae(contents(), '')
- s.draw('4' * s.columns), line(0), test()
- s.draw('5' * s.columns), line(1), test()
- s.draw('6' * s.columns), line(2), test()
- s.draw('7' * s.columns), line(3), test()
- s.draw('8' * s.columns), line(4), test()
- s.draw('9' * s.columns), line(5), test()
- s = self.create_screen(options={'scrollback_pager_history_size': 2048})
- text = '\x1b[msoft\r\x1b[mbreak\nnext😼cat'
- w(text)
- self.ae(contents(), text)
- s.historybuf.pagerhist_rewrap(2)
- self.ae(contents(), '\x1b[mso\rft\x1b[m\rbr\rea\rk\nne\rxt\r😼\rca\rt')
- s = self.create_screen(options={'scrollback_pager_history_size': 8})
- w('😼')
- self.ae(contents(), '😼')
- w('abcd')
- self.ae(contents(), '😼abcd')
- w('e')
- self.ae(contents(), 'abcde')
- def test_user_marking(self):
- def cells(*a, y=0, mark=3):
- return [(x, y, mark) for x in a]
- s = self.create_screen()
- s.draw('abaa')
- s.carriage_return(), s.linefeed()
- s.draw('xyxyx')
- s.set_marker(marker_from_regex('a', 3))
- self.ae(s.marked_cells(), cells(0, 2, 3))
- s.set_marker()
- self.ae(s.marked_cells(), [])
- def mark_x(text):
- col = 0
- for i, c in enumerate(text):
- if c == 'x':
- col += 1
- yield i, i, col
- s.set_marker(marker_from_function(mark_x))
- self.ae(s.marked_cells(), [(0, 1, 1), (2, 1, 2), (4, 1, 3)])
- s = self.create_screen(lines=5, scrollback=10)
- for i in range(15):
- s.draw(str(i))
- if i != 14:
- s.carriage_return(), s.linefeed()
- s.set_marker(marker_from_regex(r'\d+', 3))
- for i in range(10):
- self.assertTrue(s.scroll_to_next_mark())
- self.ae(s.scrolled_by, i + 1)
- self.ae(s.scrolled_by, 10)
- for i in range(10):
- self.assertTrue(s.scroll_to_next_mark(0, False))
- self.ae(s.scrolled_by, 10 - i - 1)
- self.ae(s.scrolled_by, 0)
- s = self.create_screen()
- s.draw('🐈ab')
- s.set_marker(marker_from_regex('🐈', 3))
- self.ae(s.marked_cells(), cells(0, 1))
- s.set_marker(marker_from_regex('🐈a', 3))
- self.ae(s.marked_cells(), cells(0, 1, 2))
- s.set_marker(marker_from_regex('a', 3))
- self.ae(s.marked_cells(), cells(2))
- s = self.create_screen(cols=20)
- s.tab()
- s.draw('ab')
- s.set_marker(marker_from_regex('a', 3))
- self.ae(s.marked_cells(), cells(8))
- s.set_marker(marker_from_regex('\t', 3))
- self.ae(s.marked_cells(), cells(*range(8)))
- s = self.create_screen()
- s.cursor.x = 2
- s.draw('x')
- s.cursor.x += 1
- s.draw('x')
- s.set_marker(marker_from_function(mark_x))
- self.ae(s.marked_cells(), [(2, 0, 1), (4, 0, 2)])
- def test_hyperlinks(self):
- s = self.create_screen()
- self.ae(s.line(0).hyperlink_ids(), tuple(0 for x in range(s.columns)))
- def set_link(url=None, id=None):
- parse_bytes(s, '\x1b]8;id={};{}\x1b\\'.format(id or '', url or '').encode('utf-8'))
- set_link('url-a', 'a')
- self.ae(s.line(0).hyperlink_ids(), tuple(0 for x in range(s.columns)))
- s.draw('a')
- self.ae(s.line(0).hyperlink_ids(), (1,) + tuple(0 for x in range(s.columns - 1)))
- s.draw('bc')
- self.ae(s.line(0).hyperlink_ids(), (1, 1, 1, 0, 0))
- set_link()
- s.draw('d')
- self.ae(s.line(0).hyperlink_ids(), (1, 1, 1, 0, 0))
- set_link('url-a', 'a')
- s.draw('efg')
- self.ae(s.line(0).hyperlink_ids(), (1, 1, 1, 0, 1))
- self.ae(s.line(1).hyperlink_ids(), (1, 1, 0, 0, 0))
- set_link('url-b')
- s.draw('hij')
- self.ae(s.line(1).hyperlink_ids(), (1, 1, 2, 2, 2))
- set_link()
- self.ae({('a:url-a', 1), (':url-b', 2)}, s.hyperlinks_as_set())
- s.garbage_collect_hyperlink_pool()
- self.ae({('a:url-a', 1), (':url-b', 2)}, s.hyperlinks_as_set())
- for i in range(s.lines + 2):
- s.linefeed()
- s.garbage_collect_hyperlink_pool()
- self.ae({('a:url-a', 1), (':url-b', 2)}, s.hyperlinks_as_set())
- for i in range(s.lines * 2):
- s.linefeed()
- s.garbage_collect_hyperlink_pool()
- self.assertFalse(s.hyperlinks_as_set())
- set_link('url-a', 'x')
- s.draw('a')
- set_link('url-a', 'y')
- s.draw('a')
- set_link()
- self.ae({('x:url-a', 1), ('y:url-a', 2)}, s.hyperlinks_as_set())
- s = self.create_screen()
- set_link('u' * 2048)
- s.draw('a')
- self.ae({(':' + 'u' * 2045, 1)}, s.hyperlinks_as_set())
- s = self.create_screen()
- set_link('u' * 2048, 'i' * 300)
- s.draw('a')
- self.ae({('i'*256 + ':' + 'u' * (2045 - 256), 1)}, s.hyperlinks_as_set())
- s = self.create_screen()
- set_link('1'), s.draw('1')
- set_link('2'), s.draw('2')
- set_link('3'), s.draw('3')
- s.cursor.x = 1
- set_link(), s.draw('X')
- self.ae(s.line(0).hyperlink_ids(), (1, 0, 3, 0, 0))
- self.ae({(':1', 1), (':2', 2), (':3', 3)}, s.hyperlinks_as_set())
- s.garbage_collect_hyperlink_pool()
- self.ae({(':1', 1), (':3', 2)}, s.hyperlinks_as_set())
- set_link('3'), s.draw('3')
- self.ae({(':1', 1), (':3', 2)}, s.hyperlinks_as_set())
- set_link('4'), s.draw('4')
- self.ae({(':1', 1), (':3', 2), (':4', 3)}, s.hyperlinks_as_set())
- s = self.create_screen()
- set_link('1'), s.draw('1')
- set_link('2'), s.draw('2')
- set_link('1'), s.draw('1')
- self.ae({(':2', 2), (':1', 1)}, s.hyperlinks_as_set())
- s = self.create_screen()
- set_link('1'), s.draw('12'), set_link(), s.draw('X'), set_link('1'), s.draw('3')
- s.linefeed(), s.carriage_return()
- s.draw('abc')
- s.linefeed(), s.carriage_return()
- set_link(), s.draw('Z ')
- set_link('1'), s.draw('xyz')
- s.linefeed(), s.carriage_return()
- set_link('2'), s.draw('Z Z')
- self.assertIsNone(s.current_url_text())
- self.assertIsNone(s.hyperlink_at(0, 4))
- self.assertIsNone(s.current_url_text())
- self.ae(s.hyperlink_at(0, 0), '1')
- self.ae(s.current_url_text(), '123abcxyz')
- self.ae('1', s.hyperlink_at(3, 2))
- self.ae(s.current_url_text(), '123abcxyz')
- self.ae('2', s.hyperlink_at(1, 3))
- self.ae(s.current_url_text(), 'Z Z')
- def test_bottom_margin(self):
- s = self.create_screen(cols=80, lines=6, scrollback=4)
- s.set_margins(0, 5)
- for i in range(8):
- s.draw(str(i))
- s.linefeed()
- s.carriage_return()
- self.ae(str(s.linebuf), '4\n5\n6\n7\n\n')
- self.ae(str(s.historybuf), '3\n2\n1\n0')
- def test_top_margin(self):
- s = self.create_screen(cols=80, lines=6, scrollback=4)
- s.set_margins(2, 6)
- for i in range(8):
- s.draw(str(i))
- s.linefeed()
- s.carriage_return()
- self.ae(str(s.linebuf), '0\n4\n5\n6\n7\n')
- self.ae(str(s.historybuf), '')
- def test_top_and_bottom_margin(self):
- s = self.create_screen(cols=80, lines=6, scrollback=4)
- s.set_margins(2, 5)
- for i in range(8):
- s.draw(str(i))
- s.linefeed()
- s.carriage_return()
- self.ae(str(s.linebuf), '0\n5\n6\n7\n\n')
- self.ae(str(s.historybuf), '')
- def test_osc_52(self):
- s = self.create_screen()
- c = s.callbacks
- def send(what: str):
- return parse_bytes(s, f'\033]52;p;{what}\a'.encode('ascii'))
- def t(q, *expected):
- c.clear()
- send(q)
- del q
- t.ex = list(expected)
- del expected
- try:
- self.ae(tuple(map(len, c.cc_buf)), tuple(map(len, t.ex)))
- self.ae(c.cc_buf, t.ex)
- finally:
- del t.ex
- t('XYZ', ('p;XYZ', False))
- t('a' * VT_PARSER_BUFFER_SIZE, ('p;' + 'a' * (VT_PARSER_BUFFER_SIZE - 8), True), (';' + 'a' * 8, False))
- t('', ('p;', False))
- t('!', ('p;!', False))
- def test_key_encoding_flags_stack(self):
- s = self.create_screen()
- c = s.callbacks
- def w(code, p1='', p2=''):
- p = f'{p1}'
- if p2:
- p += f';{p2}'
- return parse_bytes(s, f'\033[{code}{p}u'.encode('ascii'))
- def ac(flags):
- parse_bytes(s, b'\033[?u')
- self.ae(c.wtcbuf, f'\033[?{flags}u'.encode('ascii'))
- c.clear()
- ac(0)
- w('=', 0b1001)
- ac(0b1001)
- w('=', 0b0011, 2)
- ac(0b1011)
- w('=', 0b0110, 3)
- ac(0b1001)
- s.reset()
- ac(0)
- w('>', 0b0011)
- ac(0b0011)
- w('=', 0b1111)
- ac(0b1111)
- w('>', 0b10)
- ac(0b10)
- w('<')
- ac(0b1111)
- for i in range(10):
- w('<')
- ac(0)
- s.reset()
- for i in range(1, 16):
- w('>', i)
- ac(15)
- w('<'), ac(14), w('<'), ac(13)
- def test_color_stack(self):
- s = self.create_screen()
- c = s.callbacks
- def w(code):
- return parse_bytes(s, ('\033[' + code).encode('ascii'))
- def ac(idx, count):
- self.ae(c.wtcbuf, f'\033[{idx};{count}#Q'.encode('ascii'))
- c.clear()
- w('#R')
- ac(0, 0)
- w('#P')
- w('#R')
- ac(0, 1)
- w('10#P')
- w('#R')
- ac(0, 1)
- w('#Q')
- w('#R')
- ac(0, 0)
- for i in range(20):
- w('#P')
- w('#R')
- ac(9, 10)
- def test_detect_url(self):
- s = self.create_screen(cols=30)
- def ae(expected, x=3, y=0):
- s.detect_url(x, y)
- url = ''.join(s.text_for_marked_url())
- self.assertEqual(expected, url)
- def t(url, x=0, y=0, before='', after='', expected=''):
- s.reset()
- s.cursor.x = x
- s.cursor.y = y
- s.draw(before + url + after)
- ae(expected or url, x=x + 1 + len(before), y=y)
- t('http://moo.com')
- t('http://moo.com/something?else=+&what-')
- t('http://moo.com#fragme')
- for (st, e) in '() {} [] <>'.split():
- t('http://moo.com', before=st, after=e)
- for trailer in ')-=':
- t('http://moo.com' + trailer)
- for trailer in '{}([<>':
- t('http://moo.com', after=trailer)
- t('http://moo.com', x=s.columns - 9)
- t('https://wraps-by-one-char.com', before='[', after=']')
- t('http://[::1]:8080')
- t('https://wr[aps-by-one-ch]ar.com')
- t('http://[::1]:8080/x', after='[')
- t('http://[::1]:8080/x]y34', expected='http://[::1]:8080/x')
- t('https://wraps-by-one-char.com[]/x', after='[')
- def test_prompt_marking(self):
- s = self.create_screen()
- def mark_prompt():
- parse_bytes(s, b'\033]133;A\007')
- def mark_output():
- parse_bytes(s, b'\033]133;C\007')
- for i in range(4):
- mark_prompt()
- s.draw(f'$ {i}')
- s.carriage_return()
- s.index(), s.index()
- self.ae(s.scrolled_by, 0)
- self.assertTrue(s.scroll_to_prompt())
- self.ae(str(s.visual_line(0)), '$ 1')
- self.assertTrue(s.scroll_to_prompt())
- self.ae(str(s.visual_line(0)), '$ 0')
- self.assertFalse(s.scroll_to_prompt())
- self.assertTrue(s.scroll_to_prompt(1))
- self.ae(str(s.visual_line(0)), '$ 1')
- self.assertTrue(s.scroll_to_prompt(1))
- self.ae(str(s.visual_line(0)), '$ 2')
- self.assertFalse(s.scroll_to_prompt(1))
- s = self.create_screen()
- mark_prompt(), s.draw('$ 0')
- s.carriage_return(), s.index()
- mark_prompt(), s.draw('$ 1')
- for i in range(s.lines):
- s.carriage_return(), s.index()
- s.draw(str(i))
- self.assertTrue(s.scroll_to_prompt())
- self.ae(str(s.visual_line(0)), '$ 1')
- def lco(as_ansi=False, which=0):
- a = []
- if s.cmd_output(which, a.append, as_ansi):
- pht = pagerhist(s, as_ansi=as_ansi, upto_output_start=True)
- if pht:
- a.insert(0, pht)
- return ''.join(a)
- def fco():
- a = []
- s.cmd_output(1, a.append)
- return ''.join(a)
- def lvco():
- a = []
- s.cmd_output(2, a.append)
- return ''.join(a)
- s = self.create_screen()
- s.draw('abcd'), s.index(), s.carriage_return()
- s.draw('12'), s.index(), s.carriage_return()
- self.ae(fco(), '')
- self.ae(lco(), 'abcd\n12\n')
- s = self.create_screen()
- mark_prompt(), s.draw('$ 0')
- s.carriage_return(), s.index()
- mark_output()
- s.draw('abcd'), s.index(), s.carriage_return()
- s.draw('12'), s.index(), s.carriage_return()
- mark_prompt(), s.draw('$ 1')
- self.ae(fco(), 'abcd\n12')
- self.ae(lco(), 'abcd\n12')
- self.ae(lco(as_ansi=True), '\x1b[m\x1b]133;C\x1b\\abcd\n\x1b[m12')
- def draw_prompt(x):
- mark_prompt(), s.draw(f'$ {x}'), s.carriage_return(), s.index()
- def draw_output(n, x='', m=True):
- if m:
- mark_output()
- for i in range(n):
- s.draw(f'{i}{x}'), s.index(), s.carriage_return()
- s = self.create_screen(cols=5, lines=5, scrollback=15)
- draw_output(1, 'start', False)
- draw_prompt('0'), draw_output(3)
- draw_prompt('1')
- draw_prompt('2'), draw_output(2, 'x')
- # last cmd output
- # test: find upwards
- self.ae(s.scrolled_by, 0)
- self.ae(lco(), '0x\n1x\n')
- # get output after scroll up
- s.scroll_to_prompt()
- self.ae(s.scrolled_by, 4)
- self.ae(str(s.visual_line(0)), '$ 0')
- self.ae(lco(), '0x\n1x\n')
- # first cmd output on screen
- # test: find around
- self.ae(fco(), '0\n1\n2')
- s.scroll(2, False)
- self.ae(s.scrolled_by, 2)
- self.ae(str(s.visual_line(0)), '1')
- self.ae(fco(), '0x\n1x\n')
- # test: find downwards
- s.scroll(2, False)
- self.ae(str(s.visual_line(0)), '$ 1')
- self.ae(fco(), '0x\n1x\n')
- # resize
- # get last cmd output with continued output mark
- draw_prompt('3'), draw_output(1, 'long_line'), draw_output(2, 'l', False)
- s.resize(4, 5)
- s.scroll_to_prompt(-4)
- self.ae(str(s.visual_line(0)), '$ 0')
- self.ae(lco(), '0long_line\n0l\n1l\n')
- # last visited cmd output
- self.ae(lvco(), '0\n1\n2')
- s.scroll_to_prompt(1)
- self.ae(lvco(), '0x\n1x')
- # last command output without line break
- s = self.create_screen(cols=10, lines=3)
- draw_prompt('p1')
- mark_output(), s.draw('running')
- self.ae(lco(), 'running')
- s.index(), s.carriage_return()
- self.ae(lco(), 'running\n')
- # last command output from pager history
- s = self.create_screen()
- draw_prompt('p1')
- draw_output(30)
- self.ae(tuple(map(int, lco().split())), tuple(range(0, 30)))
- # last non empty command output
- s = self.create_screen()
- draw_prompt('a'), draw_output(2, 'a')
- draw_prompt('b'), mark_output()
- self.ae(lco(), '')
- self.ae(lco(which=3), '0a\n1a')
- s.draw('running'), s.index(), s.carriage_return()
- self.ae(lco(which=3), 'running\n')
- s = self.create_screen()
- draw_prompt('p1')
- draw_output(30)
- self.ae(tuple(map(int, lco(which=3).split())), tuple(range(0, 30)))
- s = self.create_screen()
- draw_prompt('p1')
- draw_output(2, 'a')
- draw_prompt('p1')
- draw_prompt('p1')
- self.ae(lco(which=3), '0a\n1a')
- def test_pointer_shapes(self):
- from kitty.window import set_pointer_shape
- s = self.create_screen()
- c = s.callbacks
- response = ''
- def cb(data):
- nonlocal response
- response = set_pointer_shape(s, str(data, 'utf-8'))
- c.set_pointer_shape = cb
- def send(a):
- nonlocal response
- response = ''
- parse_bytes(s, f'\x1b]22;{a}\x1b\\'.encode())
- return response
- self.ae(send('?__current__'), '0')
- self.ae(send('?__default__,__grabbed__,default,ne-resize,crosshair,XXX'), 'text,default,1,1,1,0')
- def t(q, e=None):
- self.ae(send(q), '')
- self.ae(send('?__current__'), e)
- t('default', 'default')
- s.reset()
- self.ae(send('?__current__'), '0')
- t('=crosshair', 'crosshair')
- t('<', '0')
- t('=crosshair', 'crosshair')
- t('', '0')
- t('>help', 'help')
- t('>wait', 'wait')
- t('<', 'help')
- t('<', '0')
- t('default,help', 'help')
- t('<', '0')
- t('>default,help', 'help')
- t('<', 'default')
- t('<', '0')
- t('=left_ptr', 'default')
- t('=fleur', 'move')
- def test_color_profile(self):
- c = ColorProfile(defaults)
- for i in range(8):
- col = getattr(defaults, f'color{i}')
- self.ae(c.as_color(i << 8 | 1), col)
- self.ae(c.as_color(255 << 8 | 1), Color(0xee, 0xee, 0xee))
- s = self.create_screen()
- s.color_profile.reload_from_opts(defaults)
- def q(send, expected=None):
- s.callbacks.clear()
- parse_bytes(s, b'\x1b]21;' + ';'.join(f'{k}={v}' for k, v in send.items()).encode() + b'\a')
- self.ae(s.callbacks.color_control_responses, [expected] if expected else [])
- q({k: '?' for k in 'background foreground 213 unknown'.split()}, {
- 'background': defaults.background, 'foreground': defaults.foreground, '213': defaults.color213, 'unknown': '?'})
- q({'background':'aquamarine'})
- q({'background':'?', 'selection_background': '?'}, {'background': color_names['aquamarine'], 'selection_background': s.color_profile.highlight_bg})
- q({'selection_background': ''})
- self.assertIsNone(s.color_profile.highlight_bg)
- q({'selection_background': '?'}, {'selection_background': ''})
|