Diffusion Ant (master) Code

ant/tests/test_mini_amp_tui_exit_restores_tty.cjs
457d5953e6c8master

test_mini_amp_tui_exit_restores_tty.cjs
View Options

			const { spawnSync } = require('child_process');

			function fail(message) {
			throw new Error(message);
			}

			if (process.platform === 'win32') {
			console.log('skipping mini amp tty restore test on win32');
			process.exit(0);
			}

			const script = `
			import os, pty, select, signal, sys, termios, time

			exec_path = sys.argv[1]
			script_path = sys.argv[2]

			pid, master = pty.fork()
			if pid == 0:
			os.execv(exec_path, [exec_path, script_path])

			sent = False
			exit_code = None
			deadline = time.time() + 8.0

			while time.time() < deadline:
			done, status = os.waitpid(pid, os.WNOHANG)
			if done == pid:
			exit_code = os.waitstatus_to_exitcode(status)
			break

			r, _, _ = select.select([master], [], [], 0.1)
			if master not in r:
			continue

			try:
			chunk = os.read(master, 4096)
			except OSError:
			break

			if not chunk:
			break

			if not sent:
			os.write(master, b'q')
			sent = True

			if exit_code is None:
			os.kill(pid, signal.SIGKILL)
			_, status = os.waitpid(pid, 0)
			exit_code = os.waitstatus_to_exitcode(status)

			attrs = termios.tcgetattr(master)
			lflag = attrs[3]

			print(f"exit={exit_code}")
			print(f"isig={1 if (lflag & termios.ISIG) else 0}")
			print(f"icanon={1 if (lflag & termios.ICANON) else 0}")
			print(f"echo={1 if (lflag & termios.ECHO) else 0}")
			`;

			const result = spawnSync('python3', ['-c', script, process.execPath, 'tests/amp/mini_amp_tui_repro.js'], {
			encoding: 'utf8',
			timeout: 10000,
			});

			if (result.error && result.error.code === 'ENOENT') {
			console.log('skipping mini amp tty restore test because `python3` is unavailable');
			process.exit(0);
			}

			if (result.error) throw result.error;

			const output = `${result.stdout \|\| ''}${result.stderr \|\| ''}`;

			if (!output.includes('exit=0')) {
			fail(`expected repro process to exit cleanly\n${output}`);
			}

			if (!output.includes('isig=1')) {
			fail(`expected ISIG restored after repro exit\n${output}`);
			}

			if (!output.includes('icanon=1')) {
			fail(`expected ICANON restored after repro exit\n${output}`);
			}

			if (!output.includes('echo=1')) {
			fail(`expected ECHO restored after repro exit\n${output}`);
			}

			console.log('mini amp tui repro exits and restores tty flags');

Owner Packages

No Owners