wehub-resource-sync
121 lines
4.0 KiB
Python
121 lines
4.0 KiB
Python
"""
|
|
Unit tests for the ``sys_session_send`` tool schema.
|
|
|
|
These cover the ``file_ids`` field added to the OBJECT form of the
|
|
``args`` parameter (U1). The schema is validated with ``jsonschema``
|
|
against the ``args`` sub-schema so the back-compat contract (plain
|
|
string form, object form without ``file_ids``) is exercised directly.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import jsonschema
|
|
import pytest
|
|
|
|
from omnigent.spec.types import AgentSpec
|
|
from omnigent.tools.builtins.spawn import _build_sys_session_send_schema
|
|
|
|
|
|
def _args_schema() -> dict:
|
|
""":returns: The ``anyOf`` schema for the ``args`` parameter."""
|
|
schema = _build_sys_session_send_schema({})
|
|
return schema["function"]["parameters"]["properties"]["args"]
|
|
|
|
|
|
def _schema_with_subagent() -> dict:
|
|
""":returns: The schema for a tool with one named sub-agent."""
|
|
return _build_sys_session_send_schema(
|
|
{"researcher": AgentSpec(spec_version=1, name="researcher", description="d")}
|
|
)
|
|
|
|
|
|
def _object_branch() -> dict:
|
|
""":returns: The object branch of the ``args`` ``anyOf``."""
|
|
for branch in _args_schema()["anyOf"]:
|
|
if branch.get("type") == "object":
|
|
return branch
|
|
raise AssertionError("no object branch in args anyOf")
|
|
|
|
|
|
def _validate(args: object) -> None:
|
|
"""Validate ``args`` against the ``args`` parameter schema."""
|
|
jsonschema.validate(instance=args, schema=_args_schema())
|
|
|
|
|
|
# ── Schema structure ──────────────────────────────────
|
|
|
|
|
|
def test_file_ids_present_and_optional() -> None:
|
|
branch = _object_branch()
|
|
assert branch["properties"]["file_ids"] == {
|
|
"type": "array",
|
|
"items": {"type": "string", "minLength": 1},
|
|
"minItems": 1,
|
|
"uniqueItems": True,
|
|
"description": branch["properties"]["file_ids"]["description"],
|
|
}
|
|
assert "file_ids" not in branch["required"]
|
|
assert branch["required"] == ["input"]
|
|
assert branch["additionalProperties"] is False
|
|
|
|
|
|
def test_description_mentions_file_ids() -> None:
|
|
schema = _schema_with_subagent()
|
|
assert "file_ids" in schema["function"]["description"]
|
|
|
|
|
|
def test_file_ids_description_mentions_fresh_named_spawn_only() -> None:
|
|
schema = _schema_with_subagent()
|
|
description = schema["function"]["description"]
|
|
file_ids_description = _object_branch()["properties"]["file_ids"]["description"]
|
|
|
|
assert "first named" in description
|
|
assert "session_id" in description
|
|
assert "continuing an existing named session" in description
|
|
assert "first named" in file_ids_description
|
|
assert "session_id" in file_ids_description
|
|
assert "continuing an existing named session" in file_ids_description
|
|
|
|
|
|
# ── Happy paths ───────────────────────────────────────
|
|
|
|
|
|
def test_object_args_with_file_ids_validate() -> None:
|
|
_validate({"input": "go", "file_ids": ["file_abc"]})
|
|
|
|
|
|
def test_object_args_without_file_ids_validate() -> None:
|
|
_validate({"input": "go"})
|
|
|
|
|
|
def test_plain_string_args_validate() -> None:
|
|
_validate("just a message")
|
|
|
|
|
|
# ── Errors ────────────────────────────────────────────
|
|
|
|
|
|
def test_file_ids_string_rejected() -> None:
|
|
with pytest.raises(jsonschema.ValidationError):
|
|
_validate({"input": "go", "file_ids": "file_abc"})
|
|
|
|
|
|
def test_empty_file_ids_rejected() -> None:
|
|
with pytest.raises(jsonschema.ValidationError):
|
|
_validate({"input": "go", "file_ids": []})
|
|
|
|
|
|
def test_duplicate_file_ids_rejected() -> None:
|
|
with pytest.raises(jsonschema.ValidationError):
|
|
_validate({"input": "go", "file_ids": ["file_abc", "file_abc"]})
|
|
|
|
|
|
def test_empty_file_id_rejected() -> None:
|
|
with pytest.raises(jsonschema.ValidationError):
|
|
_validate({"input": "go", "file_ids": [""]})
|
|
|
|
|
|
def test_file_ids_non_string_item_rejected() -> None:
|
|
with pytest.raises(jsonschema.ValidationError):
|
|
_validate({"input": "go", "file_ids": [123]})
|