tjb-tech
30 lines
939 B
Python
30 lines
939 B
Python
"""Tests for bridge helpers."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
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
|