hkuds--openharness
yl-jiang
593dcbcaeb
将子进程 stderr=PIPE 改为 stderr=STDOUT，确保 stderr 输出不会因无人读取而阻塞子进程，同时简化调用方对输出流的处理 Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
62 行
1.7 KiB
Python
62 行
1.7 KiB
Python
"""Tests for bridge helpers."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
import asyncio
|
|
|
|
import pytest
|
|
|
|
from openharness.bridge import WorkSecret, build_sdk_url, decode_work_secret, encode_work_secret, spawn_session
|
|
|
|
|
|
def test_work_secret_roundtrip():
|
|
secret = WorkSecret(version=1, session_ingress_token="tok", api_base_url="https://api.example.com")
|
|
encoded = encode_work_secret(secret)
|
|
decoded = decode_work_secret(encoded)
|
|
assert decoded == secret
|
|
|
|
|
|
def test_build_sdk_url():
|
|
assert build_sdk_url("https://api.example.com", "abc").startswith("wss://")
|
|
assert build_sdk_url("http://localhost:3000", "abc").startswith("ws://")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_spawn_session_and_kill(tmp_path: Path):
|
|
handle = await spawn_session(session_id="s1", command="sleep 30", cwd=tmp_path)
|
|
assert handle.process.returncode is None
|
|
await handle.kill()
|
|
assert handle.process.returncode is not None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_spawn_session_merges_stderr_into_stdout(monkeypatch, tmp_path: Path):
|
|
seen_kwargs = {}
|
|
|
|
class FakeProcess:
|
|
returncode = 0
|
|
|
|
def terminate(self):
|
|
pass
|
|
|
|
def kill(self):
|
|
pass
|
|
|
|
async def wait(self):
|
|
return 0
|
|
|
|
async def fake_create_shell_subprocess(*args, **kwargs):
|
|
seen_kwargs.update(kwargs)
|
|
return FakeProcess()
|
|
|
|
monkeypatch.setattr(
|
|
"openharness.bridge.session_runner.create_shell_subprocess",
|
|
fake_create_shell_subprocess,
|
|
)
|
|
|
|
await spawn_session(session_id="s1", command="printf err >&2", cwd=tmp_path)
|
|
|
|
assert seen_kwargs["stdout"] is asyncio.subprocess.PIPE
|
|
assert seen_kwargs["stderr"] is asyncio.subprocess.STDOUT
|