71 lines
3.1 KiB
Python
71 lines
3.1 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""Real terminal persistence/interrupt/reset via controld's Cua action route.
|
||
|
|
Run inside a disposable computer container. Files are test oracles, never an
|
||
|
|
agent-side output channel. Every application/input action goes through Cua.
|
||
|
|
"""
|
||
|
|
import base64
|
||
|
|
import importlib.machinery
|
||
|
|
import importlib.util
|
||
|
|
import time
|
||
|
|
import os
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
loader = importlib.machinery.SourceFileLoader("adapter", "/usr/local/bin/lazyboy-cua-adapter-test")
|
||
|
|
spec = importlib.util.spec_from_loader(loader.name, loader)
|
||
|
|
adapter = importlib.util.module_from_spec(spec)
|
||
|
|
loader.exec_module(adapter)
|
||
|
|
adapter.wait_health()
|
||
|
|
root = Path('/tmp/lazyboy/cua-terminal-test')
|
||
|
|
root.mkdir(exist_ok=True)
|
||
|
|
title = 'LazyBoy terminal cua-persistence-' + str(time.monotonic_ns())
|
||
|
|
for filename in ['unicode.txt', 'persistence.txt', 'interrupt.txt', 'reset.txt', 'background.pid']:
|
||
|
|
(root / filename).unlink(missing_ok=True)
|
||
|
|
|
||
|
|
|
||
|
|
def act(actions, name, settle=1000):
|
||
|
|
result = adapter.api('/act', {'actions': actions, 'observe': True, 'settle_ms': settle})
|
||
|
|
frame = base64.b64decode(result['png_base64'])
|
||
|
|
assert frame.startswith(b'\x89PNG'), 'missing shared-desktop frame'
|
||
|
|
(root / (name + '.png')).write_bytes(frame)
|
||
|
|
return result
|
||
|
|
|
||
|
|
|
||
|
|
def key(value):
|
||
|
|
return {'kind': 'key', 'key': value}
|
||
|
|
|
||
|
|
|
||
|
|
def encoded_command(value):
|
||
|
|
encoded = ''.join('\\\\' if b == 92 else "\\'" if b == 39 else chr(b) if 32 <= b <= 126 else f'\\x{b:02x}' for b in value.encode())
|
||
|
|
return "eval $'" + encoded + "'"
|
||
|
|
|
||
|
|
|
||
|
|
def command(value, name):
|
||
|
|
value = encoded_command(value)
|
||
|
|
return act([{'kind':'focus', 'title':title}, {'kind':'clipboard','text':value}, key('return')], name)
|
||
|
|
|
||
|
|
|
||
|
|
first_command = "export CUA_TERMINAL_TEST=retained; cd /tmp/lazyboy/cua-terminal-test; printf '中文 hello-cua\\n' > unicode.txt"
|
||
|
|
act([{'kind':'launch','application':'terminal','uri':'--title='+title},
|
||
|
|
{'kind':'clipboard','text':encoded_command(first_command)}, key('return')], 'first')
|
||
|
|
assert (root/'unicode.txt').read_text() == '中文 hello-cua\n'
|
||
|
|
command("printf '%s\\n' \"$CUA_TERMINAL_TEST\" \"$PWD\" > persistence.txt", 'persist')
|
||
|
|
assert (root/'persistence.txt').read_text() == 'retained\n/tmp/lazyboy/cua-terminal-test\n'
|
||
|
|
command("sleep 30", 'running')
|
||
|
|
act([{'kind':'focus','title':title},key('ctrl+c')], 'interrupt', 300)
|
||
|
|
command("printf 'interrupted-ok\\n' > interrupt.txt", 'after-interrupt')
|
||
|
|
assert (root/'interrupt.txt').read_text() == 'interrupted-ok\n'
|
||
|
|
command("sleep 60 & echo $! > background.pid", 'background')
|
||
|
|
background_pid = int((root/'background.pid').read_text())
|
||
|
|
os.kill(background_pid, 0)
|
||
|
|
command("exec /usr/local/bin/lazyboy-terminal-reset", 'reset')
|
||
|
|
try:
|
||
|
|
os.kill(background_pid, 0)
|
||
|
|
except ProcessLookupError:
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
raise AssertionError('reset left its background job running')
|
||
|
|
command("printf '%s\\n' \"${CUA_TERMINAL_TEST-unset}\" \"$PWD\" > /tmp/lazyboy/cua-terminal-test/reset.txt", 'after-reset')
|
||
|
|
assert (root/'reset.txt').read_text() == 'unset\n/home/lazyboy\n'
|
||
|
|
command("printf 'CUA terminal: 中文, persistence, interrupt and reset passed\\n'", 'complete')
|
||
|
|
print('Cua terminal persistence, Unicode, interrupt and reset passed', flush=True)
|