38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
from types import SimpleNamespace
|
|
|
|
from src.agents.protocol import AgentSessionStatus
|
|
from src.services.decision_manager import _phase2_fallback_reason
|
|
|
|
|
|
def _package(**kwargs):
|
|
data = {
|
|
"session_status": AgentSessionStatus.COMPLETED,
|
|
"all_agents_degraded": False,
|
|
"recommended_action": "kubectl rollout restart deployment/awoooi-api -n awoooi-prod",
|
|
"requires_human_approval": False,
|
|
}
|
|
data.update(kwargs)
|
|
return SimpleNamespace(**data)
|
|
|
|
|
|
def test_phase2_timeout_continues_to_fallback() -> None:
|
|
pkg = _package(
|
|
session_status=AgentSessionStatus.TIMEOUT,
|
|
recommended_action=None,
|
|
requires_human_approval=True,
|
|
)
|
|
|
|
assert _phase2_fallback_reason(pkg) == "session_timeout"
|
|
|
|
|
|
def test_phase2_empty_manual_gate_continues_to_fallback() -> None:
|
|
pkg = _package(recommended_action="", requires_human_approval=True)
|
|
|
|
assert _phase2_fallback_reason(pkg) == "empty_action_requires_human"
|
|
|
|
|
|
def test_phase2_actionable_package_stays_primary() -> None:
|
|
pkg = _package()
|
|
|
|
assert _phase2_fallback_reason(pkg) is None
|