Bound watcher stdin draining with a deadline

This commit is contained in:
2026-09-08 18:06:43 +01:00
parent 4522fdbce1
commit f4fc81eefe
3 changed files with 82 additions and 8 deletions
+2 -1
View File
@@ -1,5 +1,6 @@
#!/bin/bash
# Run all plugin logic tests. Requires node.
# Run all plugin logic tests. Requires node and python3.
set -euo pipefail
cd "$(dirname "${BASH_SOURCE[0]}")/.."
python3 -m unittest discover -s tests -p 'test_capture.py'
exec node --test 'tests/*.mjs'
+56
View File
@@ -0,0 +1,56 @@
import os
import subprocess
import sys
import tempfile
import unittest
class WatcherInputTests(unittest.TestCase):
def helper(self, stdin, timeout=0.2):
return subprocess.Popen(
[sys.executable, "-c", """
import capture
import sys
import tracemalloc
tracemalloc.start()
drain = capture.drain_stdin
capture.drain_stdin = lambda: drain(timeout=float(sys.argv[1]))
capture.list_types = lambda: print('probed') or []
capture.main()
assert tracemalloc.get_traced_memory()[1] < 1024 * 1024
""", str(timeout)], stdin=stdin, stdout=subprocess.PIPE, stderr=subprocess.PIPE,
)
def finish(self, proc, expected, payload=None):
try:
out, err = proc.communicate(input=payload, timeout=8)
self.assertEqual(proc.returncode, 0, err.decode())
self.assertEqual(out, expected)
finally:
if proc.poll() is None:
proc.kill()
proc.wait()
def test_large_payload_is_discarded_without_accumulation(self):
with tempfile.TemporaryFile() as source:
source.truncate(64 * 1024 * 1024)
self.finish(self.helper(source, timeout=5), b"probed\n")
def test_finite_pipe_still_probes(self):
self.finish(self.helper(subprocess.PIPE, timeout=5), b"probed\n",
payload=b"clipboard content\n" * 16384)
def test_stalled_pipe_exits_without_probing(self):
reader, writer = os.pipe()
try:
with os.fdopen(reader, "rb") as source:
self.finish(self.helper(source), b"")
finally:
os.close(writer)
def test_endless_source_exits_without_probing(self):
with open('/dev/zero', 'rb') as source:
self.finish(self.helper(source), b"")
def test_empty_input_still_probes(self):
self.finish(self.helper(subprocess.DEVNULL), b"probed\n")