Image regress tests.

regress/image-fallback.sh

@@ -0,0 +1,108 @@
+#!/bin/sh
+
+# SIXEL input is visible as text on ASCII and UTF-8 clients, including
+# copy-mode selection and a graphical client with no known cell pixel size.
+
+PATH=/bin:/usr/bin
+TERM=screen
+export TERM
+[ -z "$TEST_TMUX" ] && TEST_TMUX=$(readlink -f ../tmux)
+
+python3 -B - "$TEST_TMUX" <<'PY'
+import re
+import sys
+from image_test import image_test, sixel
+from image_terminal import Terminal
+
+white = sixel(16, 16, [(100, 100, 100)], [0] * (16 * 16))
+
+for utf8 in (False, True):
+    with image_test(sys.argv[1]) as test:
+        client = test.attach(graphical=False, utf8=utf8)
+        char = b'\xe2\x96\x84' if utf8 else b'@'
+        test.emit(white)
+        assert not client.images(), 'graphical output on a fallback client'
+        assert client.data.count(char) >= 2, (utf8, bytes(client.data))
+        test.run('copy-mode')
+        test.run('send-keys', '-X', 'history-top')
+        test.run('send-keys', '-X', 'start-of-line')
+        test.run('send-keys', '-X', 'begin-selection')
+        test.run('send-keys', '-X', 'cursor-right')
+        test.redraw()
+        assert not client.images()
+        assert client.data.count(char) >= 2, ('selection lost fallback', utf8)
+        test.run('send-keys', '-X', 'cancel')
+        test.emit(b'\033[2J\033[H')
+        assert not client.images()
+        # Clearing must also remove the image from later full redraws.
+        test.redraw()
+        assert char not in client.data, ('cleared fallback reappeared', utf8)
+
+with image_test(sys.argv[1]) as test:
+    client = test.attach(graphical=True, utf8=False, cw=0, ch=0)
+    test.emit(white)
+    assert not client.images(), 'SIXEL sent without a cell pixel size'
+    # Default pixel sizes may leave padding, so the brightness character
+    # need not be '@'. Remove terminal controls before checking the text.
+    text = re.sub(rb'\033\[[0-?]*[ -/]*[@-~]', b'', client.data)
+    text = re.sub(rb'\033\][^\007]*?(?:\007|\033\\)', b'', text)
+    text = re.sub(rb'\033.', b'', text)
+    assert any(bytes([ch]) in text for ch in b'.:-=+*#%@'), (
+        'missing fallback with unknown pixel size', text)
+
+# Every ASCII brightness character is reachable, in the right grid cell.
+levels = [0, 12, 23, 34, 45, 56, 67, 78, 89, 100]
+with image_test(sys.argv[1]) as test:
+    client = test.attach(graphical=False, utf8=False)
+    test.emit(sixel(80, 16, [(v, v, v) for v in levels],
+        [x // 8 for y in range(16) for x in range(80)]))
+    test.redraw()
+    terminal = Terminal()
+    terminal.feed(client.data)
+    actual = ''.join(terminal.cells[x, 0][0] for x in range(10))
+    assert actual == ' .:-=+*#%@', actual
+
+# The lower-half character uses the upper sample as background and the
+# lower sample as foreground. Check true colour and 256-colour mapping.
+for truecolour in (False, True):
+    with image_test(sys.argv[1]) as test:
+        if truecolour:
+            test.run('set-option', '-as', 'terminal-features',
+                'screen-256color:RGB')
+        client = test.attach(graphical=False)
+        test.emit(b'\033[3;4H' + sixel(8, 16,
+            [(100, 0, 0), (0, 0, 100)],
+            [y >= 8 for y in range(16) for x in range(8)]))
+        test.redraw()
+        terminal = Terminal()
+        terminal.feed(client.data)
+        char, fg, bg, bold = terminal.cells[3, 2]
+        assert char == '\u2584' and not bold, (char, fg, bg, bold)
+        if truecolour:
+            expected = ((0, 0, 255), (255, 0, 0))
+        else:
+            expected = (21, 196)
+        assert (fg, bg) == expected, (truecolour, fg, bg)
+
+# Eight-colour UTF-8 terminals use shades, with bold supplying extra levels
+# unless Nobr disables it. Force the capabilities without extra terminfo files.
+for nobr in (False, True):
+    with image_test(sys.argv[1]) as test:
+        overrides = 'screen:colors=8:Nobr=' if nobr else 'screen:colors=8:Nobr@'
+        test.run('set-option', '-as', 'terminal-overrides', overrides)
+        client = test.attach(graphical=False, term='screen')
+        levels = [0, 14, 29, 43, 57, 71, 86, 100]
+        test.emit(sixel(64, 16, [(v, v, v) for v in levels],
+            [x // 8 for y in range(16) for x in range(64)]))
+        test.redraw()
+        terminal = Terminal()
+        terminal.feed(client.data)
+        chars = ''.join(terminal.cells[x, 0][0] for x in range(8))
+        bold = [terminal.cells[x, 0][3] for x in range(8)]
+        if nobr:
+            assert chars == ' \u2591\u2591\u2592\u2592\u2593\u2593\u2588', chars
+            assert not any(bold), bold
+        else:
+            assert chars == ' \u2591\u2591\u2592\u2592\u2593\u2588\u2588', chars
+            assert bold == [False, False, True, False, True, False, False, True], bold
+PY