Jeff (CTO)
18482b938f
CI / typecheck (push) Waiting to run
CI / lint (push) Waiting to run
CI / security (push) Waiting to run
CI / quality (push) Waiting to run
CI / unit_tests (push) Waiting to run
CI / integration_tests (push) Waiting to run
CI / coverage (push) Blocked by required conditions
CI / build (push) Waiting to run
CI / docker (push) Blocked by required conditions
CI / lint (pull_request) Successful in 14s
CI / typecheck (pull_request) Successful in 27s
CI / security (pull_request) Successful in 22s
CI / quality (pull_request) Successful in 15s
CI / integration_tests (pull_request) Successful in 4m32s
CI / build (pull_request) Successful in 16s
CI / unit_tests (pull_request) Successful in 9m45s
CI / coverage (pull_request) Successful in 6m53s
CI / docker (pull_request) Successful in 39s
85 lines
2.2 KiB
Python
85 lines
2.2 KiB
Python
"""Helper script for Robot Framework tool runtime smoke tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
# Ensure src is importable when run from workspace root
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "src"))
|
|
|
|
from cleveragents.tool.registry import ToolRegistry
|
|
from cleveragents.tool.runner import ToolRunner
|
|
from cleveragents.tool.runtime import ToolSpec
|
|
|
|
|
|
def _echo(inputs: dict[str, Any]) -> dict[str, Any]:
|
|
return dict(inputs)
|
|
|
|
|
|
def _fail(inputs: dict[str, Any]) -> dict[str, Any]:
|
|
raise RuntimeError("boom")
|
|
|
|
|
|
def test_register() -> None:
|
|
registry = ToolRegistry()
|
|
spec = ToolSpec(
|
|
name="smoke/echo",
|
|
description="Echo tool",
|
|
handler=_echo,
|
|
)
|
|
registry.register(spec)
|
|
found = registry.get("smoke/echo")
|
|
assert found is not None
|
|
assert found.name == "smoke/echo"
|
|
print("registry-ok")
|
|
|
|
|
|
def test_lifecycle() -> None:
|
|
registry = ToolRegistry()
|
|
spec = ToolSpec(
|
|
name="smoke/adder",
|
|
description="Adder tool",
|
|
handler=lambda inp: {"sum": inp.get("a", 0) + inp.get("b", 0)},
|
|
)
|
|
registry.register(spec)
|
|
runner = ToolRunner(registry)
|
|
discovered = runner.discover()
|
|
assert len(discovered) == 1
|
|
runner.activate("smoke/adder")
|
|
result = runner.execute("smoke/adder", {"a": 1, "b": 2})
|
|
assert result.success
|
|
assert result.output["sum"] == 3
|
|
runner.deactivate("smoke/adder")
|
|
print("lifecycle-ok")
|
|
|
|
|
|
def test_error() -> None:
|
|
registry = ToolRegistry()
|
|
spec = ToolSpec(
|
|
name="smoke/fail",
|
|
description="Failing tool",
|
|
handler=_fail,
|
|
)
|
|
registry.register(spec)
|
|
runner = ToolRunner(registry)
|
|
result = runner.execute("smoke/fail", {})
|
|
assert not result.success
|
|
assert result.error is not None
|
|
assert "RuntimeError" in result.error
|
|
print("error-ok")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
cmd = sys.argv[1] if len(sys.argv) > 1 else "register"
|
|
if cmd == "register":
|
|
test_register()
|
|
elif cmd == "lifecycle":
|
|
test_lifecycle()
|
|
elif cmd == "error":
|
|
test_error()
|
|
else:
|
|
print(f"Unknown command: {cmd}", file=sys.stderr)
|
|
sys.exit(1)
|