72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
|
"""Schema tests: Plan.from_dict, Plan.to_dict, and related deserialization."""
|
||
|
|
|
||
|
|
from fusionagi.schemas.plan import Plan, PlanStep
|
||
|
|
|
||
|
|
|
||
|
|
def test_plan_from_dict_valid() -> None:
|
||
|
|
d = {
|
||
|
|
"steps": [
|
||
|
|
{"id": "s1", "description": "Step 1", "dependencies": []},
|
||
|
|
{"id": "s2", "description": "Step 2", "dependencies": ["s1"]},
|
||
|
|
],
|
||
|
|
"fallback_paths": [],
|
||
|
|
"metadata": {},
|
||
|
|
}
|
||
|
|
plan = Plan.from_dict(d)
|
||
|
|
assert len(plan.steps) == 2
|
||
|
|
assert plan.steps[0].id == "s1"
|
||
|
|
assert plan.steps[1].id == "s2"
|
||
|
|
assert plan.steps[1].dependencies == ["s1"]
|
||
|
|
assert plan.fallback_paths == []
|
||
|
|
assert plan.metadata == {}
|
||
|
|
|
||
|
|
|
||
|
|
def test_plan_from_dict_extra_keys() -> None:
|
||
|
|
d = {
|
||
|
|
"steps": [{"id": "s1", "description": "Step 1", "dependencies": []}],
|
||
|
|
"fallback_paths": [],
|
||
|
|
"metadata": {},
|
||
|
|
"extra_key": "ignored",
|
||
|
|
}
|
||
|
|
plan = Plan.from_dict(d)
|
||
|
|
assert len(plan.steps) == 1
|
||
|
|
assert plan.steps[0].id == "s1"
|
||
|
|
|
||
|
|
|
||
|
|
def test_plan_from_dict_empty_steps() -> None:
|
||
|
|
d = {"steps": [], "fallback_paths": [], "metadata": {}}
|
||
|
|
plan = Plan.from_dict(d)
|
||
|
|
assert plan.steps == []
|
||
|
|
assert plan.step_ids() == []
|
||
|
|
|
||
|
|
|
||
|
|
def test_plan_from_dict_invalid_input() -> None:
|
||
|
|
"""Plan.from_dict with non-dict raises TypeError."""
|
||
|
|
try:
|
||
|
|
Plan.from_dict(None) # type: ignore[arg-type]
|
||
|
|
assert False, "expected TypeError"
|
||
|
|
except TypeError as e:
|
||
|
|
assert "expects dict" in str(e)
|
||
|
|
try:
|
||
|
|
Plan.from_dict("not a dict") # type: ignore[arg-type]
|
||
|
|
assert False, "expected TypeError"
|
||
|
|
except TypeError as e:
|
||
|
|
assert "expects dict" in str(e)
|
||
|
|
|
||
|
|
|
||
|
|
def test_plan_from_dict_to_dict_roundtrip() -> None:
|
||
|
|
plan = Plan(
|
||
|
|
steps=[
|
||
|
|
PlanStep(id="a", description="A", dependencies=[]),
|
||
|
|
PlanStep(id="b", description="B", dependencies=["a"]),
|
||
|
|
],
|
||
|
|
fallback_paths=[["a"]],
|
||
|
|
metadata={"k": "v"},
|
||
|
|
)
|
||
|
|
d = plan.to_dict()
|
||
|
|
restored = Plan.from_dict(d)
|
||
|
|
assert len(restored.steps) == 2
|
||
|
|
assert restored.steps[0].id == plan.steps[0].id
|
||
|
|
assert restored.fallback_paths == plan.fallback_paths
|
||
|
|
assert restored.metadata == plan.metadata
|