138 lines
4.5 KiB
Python
138 lines
4.5 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from examples.report_workflow.ops import (
|
|
ActionItem,
|
|
ExtractInput,
|
|
MarkdownInput,
|
|
ReadInput,
|
|
ReportOutput,
|
|
_extract_report,
|
|
_read_notes,
|
|
_render_markdown_report,
|
|
)
|
|
from wf_config import load_workflow_config
|
|
from wf_server.config import build_workflow_server_from_workflow_config
|
|
|
|
EXAMPLE_DIR = Path(__file__).resolve().parents[2] / "examples" / "report_workflow"
|
|
|
|
|
|
def test_report_workflow_read_notes_rejects_paths_outside_example() -> None:
|
|
with pytest.raises(ValueError, match="inside the example"):
|
|
_read_notes(ReadInput(path="../pyproject.toml"))
|
|
|
|
|
|
def test_report_workflow_markdown_renderer_round_trips_through_extractor() -> None:
|
|
report = ReportOutput(
|
|
title="Weekly Project Update",
|
|
summary="Demo summary.",
|
|
action_items=[
|
|
ActionItem(owner="Alice", task="Prepare demo config", due="Friday")
|
|
],
|
|
risks=["Quota is low"],
|
|
followups=["Render Markdown"],
|
|
)
|
|
|
|
rendered = _render_markdown_report(MarkdownInput(report=report))
|
|
extracted = _extract_report(ExtractInput(text=rendered.markdown))
|
|
|
|
assert extracted == report
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_report_workflow_python_source_loads_and_calls_capability(
|
|
tmp_path,
|
|
) -> None:
|
|
config = load_workflow_config(EXAMPLE_DIR / "wf.config.json")
|
|
config.server.store.root = tmp_path / "store"
|
|
server = build_workflow_server_from_workflow_config(config)
|
|
|
|
listed = await server.api.list_capabilities(source_id="local.report")
|
|
names = {capability["name"] for capability in listed["capabilities"]}
|
|
|
|
assert "local.report.extract_report" in names
|
|
|
|
payload = json.loads((EXAMPLE_DIR / "cap-input.json").read_text(encoding="utf-8"))
|
|
result = await server.api.call_capability(
|
|
qualified_name="local.report.extract_report",
|
|
payload=payload,
|
|
)
|
|
|
|
assert result["outcome"] == "ok"
|
|
output = result["output"]
|
|
assert output is not None
|
|
assert output["title"] == "Weekly Project Update"
|
|
assert output["action_items"][0] == {
|
|
"owner": "Alice",
|
|
"task": "Prepare demo config",
|
|
"due": "Friday",
|
|
}
|
|
assert "Google Drive MCP quota" in output["risks"][0]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_report_workflow_artifact_deployment_run_path(tmp_path) -> None:
|
|
config = load_workflow_config(EXAMPLE_DIR / "wf.config.json")
|
|
config.server.store.root = tmp_path / "store"
|
|
server = build_workflow_server_from_workflow_config(config)
|
|
|
|
plan = json.loads((EXAMPLE_DIR / "workflow.plan.json").read_text(encoding="utf-8"))
|
|
|
|
await server.api.create_artifact_from_plan(
|
|
artifact_id="report_case_study",
|
|
version=1,
|
|
title="Report Case Study",
|
|
plan=plan,
|
|
outcomes=["ok"],
|
|
source_bindings={"local.report": "local.report"},
|
|
)
|
|
await server.api.save_deployment(
|
|
{
|
|
"id": "report_case_study.default",
|
|
"artifact_id": "report_case_study",
|
|
"artifact_version": 1,
|
|
"bindings": {"local.report": "local.report"},
|
|
}
|
|
)
|
|
run_input = json.loads((EXAMPLE_DIR / "cap-input.json").read_text(encoding="utf-8"))
|
|
run = await server.api.run_deployment(
|
|
deployment_id="report_case_study.default",
|
|
workflow_input=run_input,
|
|
)
|
|
|
|
assert run["status"] == "completed"
|
|
output = run["output"]
|
|
assert output is not None
|
|
assert output["report"]["title"] == "Weekly Project Update"
|
|
assert len(output["report"]["action_items"]) == 3
|
|
assert output["markdown"].startswith("# Weekly Project Update")
|
|
|
|
|
|
def test_report_workflow_read_notes_accepts_text_by_value() -> None:
|
|
payload = json.loads((EXAMPLE_DIR / "cap-input.json").read_text(encoding="utf-8"))
|
|
|
|
notes = _read_notes(ReadInput(**payload))
|
|
|
|
assert notes.text.startswith("# Weekly Project Update")
|
|
|
|
|
|
def test_report_workflow_read_input_requires_text_or_path() -> None:
|
|
with pytest.raises(ValueError, match="text or path"):
|
|
ReadInput.model_validate({})
|
|
|
|
|
|
def test_report_workflow_read_input_rejects_text_and_path_together() -> None:
|
|
with pytest.raises(ValueError, match="exactly one"):
|
|
ReadInput(text="hello", path="input.md")
|
|
|
|
|
|
def test_report_workflow_read_input_schema_requires_text_or_path() -> None:
|
|
schema = ReadInput.model_json_schema()
|
|
|
|
assert schema["oneOf"] == [{"required": ["text"]}, {"required": ["path"]}]
|
|
assert set(schema["properties"]) == {"text", "path"}
|