Kitty image input path.

regress/image-kitty-lifecycle.sh

@@ -0,0 +1,109 @@
+#!/bin/sh
+
+# Kitty placement recovery, reflow, alternate screens and pane-local IDs.
+
+PATH=/bin:/usr/bin
+TERM=screen
+export TERM
+[ -z "$TEST_TMUX" ] && TEST_TMUX=$(readlink -f ../tmux)
+command -v python3 >/dev/null || exit 101
+
+python3 -B - "$TEST_TMUX" <<'PY'
+import sys
+from image_test import image_test, kitty
+from image_terminal import Terminal
+
+
+def snapshot(test, client, expected, sx=40, sy=16):
+    """Check visible image pixels after an independent redraw."""
+    test.redraw()
+    terminal = Terminal(sx=sx, sy=sy)
+    terminal.feed(client.data)
+    terminal.check(expected)
+
+
+red, green, blue = (255, 0, 0), (0, 255, 0), (0, 0, 255)
+raw = bytes(c for y in range(32) for x in range(32)
+    for c in (*(red if x < 16 else blue), 255))
+upload = kitty('a=t,i=7,f=32,s=32,v=32', raw)
+place = kitty('a=p,i=7,p=9,C=1')
+expected = {(x, y): red if x < 16 else blue
+    for y in range(32) for x in range(32)}
+
+with image_test(sys.argv[1]) as test:
+    client = test.attach()
+    test.emit(upload + place)
+    for width in (1, 40, 2, 40):
+        test.resize(client, width, 16)
+        snapshot(test, client,
+            {point: colour for point, colour in expected.items()
+                if point[0] < width * 8}, sx=width)
+    # Recovering a clipped named placement must still allow its replacement.
+    test.emit(b'\033[4;5H' + place)
+    snapshot(test, client,
+        {(x + 32, y + 48): colour for (x, y), colour in expected.items()})
+
+    # An alternate screen has its own grid, while the original grid is retained.
+    test.emit(b'\033[?1049h')
+    snapshot(test, client, {})
+    test.emit(b'\033[H' + kitty('a=T,i=8,p=9,f=32,s=1,v=1,C=1,c=1,r=1',
+        bytes((*green, 255))))
+    snapshot(test, client,
+        {(x, y): green for y in range(16) for x in range(8)})
+    test.emit(b'\033[?1049l')
+    snapshot(test, client,
+        {(x + 32, y + 48): colour for (x, y), colour in expected.items()})
+
+# Kitty image rows retain their width while adjacent wrapped text reflows.
+with image_test(sys.argv[1], sx=16, sy=20) as test:
+    client = test.attach()
+    text = b'abcdefghijklmnopqrstuvwx'
+    test.emit(text + b'\r\n' + upload + kitty('a=p,i=7,p=9') +
+        b'\r\n' + text + b'\r\n')
+    for width in (8, 16, 8, 16):
+        test.resize(client, width, 20)
+        top = (len(text) + width - 1) // width - int(test.run(
+            'display-message', '-p', '#{history_size}'))
+        assert top >= 0, top
+        snapshot(test, client,
+            {(x, y + top * 16): colour for (x, y), colour in expected.items()},
+            sx=width, sy=20)
+        captured = test.run('capture-pane', '-p', '-S', '-').replace('\n', '')
+        assert captured == text.decode() * 2, captured
+
+# Equal application image and placement IDs in separate panes are independent.
+with image_test(sys.argv[1]) as test:
+    client = test.attach()
+    first = test.run('display-message', '-p', '#{pane_id}')
+    second = test.run('split-window', '-d', '-P', '-F', '#{pane_id}',
+        "stty raw -echo; printf '\033]2;SECOND\007'; exec cat")
+    test.wait(lambda: test.run('display-message', '-pt', second,
+        '#{pane_title}') == 'SECOND')
+    placed = {}
+    for pane, colour in [(first, red), (second, blue)]:
+        test.run('select-pane', '-t', pane)
+        test.emit(kitty('a=T,i=7,p=9,f=32,s=1,v=1,C=1,c=1,r=1',
+            bytes((*colour, 255))))
+        x, y = map(int, test.run('display-message', '-pt', pane,
+            '#{pane_left},#{pane_top}').split(','))
+        placed.update({(x * 8 + px, y * 16 + py): colour
+            for py in range(16) for px in range(8)})
+    snapshot(test, client, placed)
+    test.run('select-pane', '-t', first)
+    test.emit(kitty('a=T,i=7,p=9,f=32,s=1,v=1,C=1,c=1,r=1',
+        bytes((*green, 255))))
+    placed = {point: green if colour == red else colour
+        for point, colour in placed.items()}
+    snapshot(test, client, placed)
+    test.run('kill-pane', '-t', second)
+    snapshot(test, client,
+        {(x, y): green for y in range(16) for x in range(8)})
+    # Respawning removes the previous application's placements and sources.
+    test.run('respawn-pane', '-k', '-t', first,
+        "stty raw -echo; printf '\033]2;RESPAWNED\007'; exec cat")
+    test.wait(lambda: test.run('display-message', '-p',
+        '#{pane_title}') == 'RESPAWNED')
+    snapshot(test, client, {})
+    test.emit(kitty('a=p,i=7,p=9,C=1'))
+    snapshot(test, client, {})
+PY