mirror of
https://github.com/x1xhlol/system-prompts-and-models-of-ai-tools.git
synced 2026-06-17 23:09:35 +00:00
107 lines
3.3 KiB
Python
107 lines
3.3 KiB
Python
"""Tests for decision plane bundle (Completion Program WS2)."""
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from app.services.core_os.decision_memo import AuditMetadata, DecisionMemo, FinancialImpact
|
|
from app.services.core_os.decision_plane_contracts import (
|
|
ApprovalPacket,
|
|
EvidencePack,
|
|
ExecutionIntent,
|
|
assemble_decision_bundle,
|
|
new_evidence_pack_id,
|
|
validate_class_b_bundle,
|
|
)
|
|
|
|
|
|
def test_assemble_decision_bundle_keys():
|
|
memo = DecisionMemo(
|
|
objective="Test objective",
|
|
decision_context="ctx",
|
|
inputs_used=["a"],
|
|
assumptions=["x"],
|
|
recommendation_ar="do thing",
|
|
alternatives_considered=["b"],
|
|
expected_financial_impact=FinancialImpact(),
|
|
risk_register=[],
|
|
confidence_score=80.0,
|
|
required_approvals=["CEO"],
|
|
next_best_action="approve",
|
|
rollback_plan="revert",
|
|
evidence_links=[],
|
|
audit_metadata=AuditMetadata(agent_id="test_agent", timestamp="2026-01-01T00:00:00Z"),
|
|
)
|
|
ep = EvidencePack(
|
|
pack_id=new_evidence_pack_id(),
|
|
sources=["https://example.com/doc"],
|
|
provenance_score=70.0,
|
|
)
|
|
ap = ApprovalPacket(
|
|
approval_class="A2",
|
|
reversibility_class="R2",
|
|
sensitivity_class="S2",
|
|
actor_type="recommender_agent",
|
|
approvers_required=["CFO"],
|
|
)
|
|
ei = ExecutionIntent(
|
|
workflow_key="partner_approval_v1",
|
|
idempotency_key="idem-001",
|
|
requested_side_effect_class="internal_write",
|
|
correlation_id="corr-1",
|
|
)
|
|
bundle = assemble_decision_bundle(
|
|
evidence_pack=ep,
|
|
approval_packet=ap,
|
|
execution_intent=ei,
|
|
memo_json=memo.to_json(),
|
|
)
|
|
assert set(bundle.keys()) == {
|
|
"memo_json",
|
|
"evidence_pack_json",
|
|
"risk_register_json",
|
|
"approval_packet_json",
|
|
"execution_intent_json",
|
|
}
|
|
assert bundle["approval_packet_json"]["approval_class"] == "A2"
|
|
assert bundle["execution_intent_json"]["idempotency_key"] == "idem-001"
|
|
validate_class_b_bundle(bundle)
|
|
|
|
|
|
def test_validate_class_b_bundle_rejects_missing_memo_approvals():
|
|
memo = DecisionMemo(
|
|
objective="x",
|
|
decision_context="c",
|
|
inputs_used=["i"],
|
|
assumptions=["a"],
|
|
recommendation_ar="r",
|
|
alternatives_considered=["b"],
|
|
expected_financial_impact=FinancialImpact(),
|
|
risk_register=[],
|
|
confidence_score=50.0,
|
|
required_approvals=[], # invalid for Class B
|
|
next_best_action="n",
|
|
rollback_plan="rb",
|
|
evidence_links=[],
|
|
audit_metadata=AuditMetadata(agent_id="a", timestamp="2026-01-01T00:00:00Z"),
|
|
)
|
|
ep = EvidencePack(pack_id=new_evidence_pack_id(), provenance_score=1.0)
|
|
ap = ApprovalPacket(
|
|
approval_class="A1",
|
|
reversibility_class="R0",
|
|
sensitivity_class="S0",
|
|
actor_type="human",
|
|
)
|
|
ei = ExecutionIntent(
|
|
workflow_key="w",
|
|
idempotency_key="k",
|
|
requested_side_effect_class="none",
|
|
)
|
|
bundle = assemble_decision_bundle(
|
|
evidence_pack=ep,
|
|
approval_packet=ap,
|
|
execution_intent=ei,
|
|
memo_json=memo.to_json(),
|
|
)
|
|
with pytest.raises(ValueError, match="required_approvals"):
|
|
validate_class_b_bundle(bundle)
|