wehub-resource-sync
b4fbd6fe9f
Deploy Site / deploy-vercel (push) Has been skipped
Deploy Site / deploy-docs (push) Has been skipped
Build Skills Index / build-index (push) Has been skipped
CI / Deny unrelated histories (push) Has been skipped
CI / Detect affected areas (push) Successful in 27m35s
CI / OSV scan (push) Failing after 4s
CI / Build&Test Docker image (push) Successful in 9s
CI / Supply-chain scan (push) Has been skipped
CI / Lint Docker scripts (push) Failing after 5m13s
CI / Check contributors (push) Failing after 12m8s
CI / Docs Site (push) Failing after 12m8s
CI / TypeScript (push) Failing after 12m8s
CI / Python lints (push) Failing after 12m9s
CI / Python tests (push) Failing after 12m9s
CI / Check uv.lock (push) Failing after 23m22s
CI / CI timing report (push) Has been cancelled
Build Skills Index / trigger-deploy (push) Has been cancelled
CI / All required checks pass (push) Has been cancelled
145 lines
3.6 KiB
Python
145 lines
3.6 KiB
Python
"""Tests for _ThreadedProcessHandle — the adapter for SDK backends."""
|
|
|
|
import threading
|
|
import time
|
|
|
|
from tools.environments.base import _ThreadedProcessHandle
|
|
|
|
|
|
class TestBasicExecution:
|
|
def test_successful_execution(self):
|
|
def exec_fn():
|
|
return ("hello world", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
|
|
assert handle.returncode == 0
|
|
output = handle.stdout.read()
|
|
assert "hello world" in output
|
|
|
|
def test_nonzero_exit_code(self):
|
|
def exec_fn():
|
|
return ("error occurred", 42)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
|
|
assert handle.returncode == 42
|
|
output = handle.stdout.read()
|
|
assert "error occurred" in output
|
|
|
|
def test_exception_in_exec_fn(self):
|
|
def exec_fn():
|
|
raise RuntimeError("boom")
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
|
|
assert handle.returncode == 1
|
|
|
|
def test_empty_output(self):
|
|
def exec_fn():
|
|
return ("", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
|
|
assert handle.returncode == 0
|
|
output = handle.stdout.read()
|
|
assert output == ""
|
|
|
|
|
|
class TestPolling:
|
|
def test_poll_returns_none_while_running(self):
|
|
event = threading.Event()
|
|
|
|
def exec_fn():
|
|
event.wait(timeout=5)
|
|
return ("done", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
assert handle.poll() is None
|
|
|
|
event.set()
|
|
handle.wait(timeout=5)
|
|
assert handle.poll() == 0
|
|
|
|
def test_poll_returns_returncode_when_done(self):
|
|
def exec_fn():
|
|
return ("ok", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
assert handle.poll() == 0
|
|
|
|
|
|
class TestCancelFn:
|
|
def test_cancel_fn_called_on_kill(self):
|
|
called = threading.Event()
|
|
|
|
def cancel():
|
|
called.set()
|
|
|
|
def exec_fn():
|
|
time.sleep(10)
|
|
return ("", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn, cancel_fn=cancel)
|
|
handle.kill()
|
|
assert called.is_set()
|
|
|
|
def test_cancel_fn_none_is_safe(self):
|
|
def exec_fn():
|
|
return ("ok", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn, cancel_fn=None)
|
|
handle.kill() # should not raise
|
|
handle.wait(timeout=5)
|
|
assert handle.returncode == 0
|
|
|
|
def test_cancel_fn_exception_swallowed(self):
|
|
def cancel():
|
|
raise RuntimeError("cancel failed")
|
|
|
|
def exec_fn():
|
|
return ("ok", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn, cancel_fn=cancel)
|
|
handle.kill() # should not raise despite cancel raising
|
|
handle.wait(timeout=5)
|
|
|
|
|
|
class TestStdoutPipe:
|
|
def test_stdout_is_readable(self):
|
|
def exec_fn():
|
|
return ("line1\nline2\nline3\n", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
|
|
lines = handle.stdout.readlines()
|
|
assert len(lines) == 3
|
|
assert lines[0] == "line1\n"
|
|
|
|
def test_stdout_iterable(self):
|
|
def exec_fn():
|
|
return ("a\nb\nc\n", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
|
|
collected = list(handle.stdout)
|
|
assert len(collected) == 3
|
|
|
|
def test_unicode_output(self):
|
|
def exec_fn():
|
|
return ("hello 世界 🌍\n", 0)
|
|
|
|
handle = _ThreadedProcessHandle(exec_fn)
|
|
handle.wait(timeout=5)
|
|
|
|
output = handle.stdout.read()
|
|
assert "世界" in output
|
|
assert "🌍" in output
|