Function phase_emits_budget_exhausted_when_capped

core/src/orchestration/workflow.rs:540–575

()

    #[tokio::test]
    async fn phase_emits_budget_exhausted_when_capped() {
        use crate::budget::BudgetGuard;
        use crate::llm::TokenUsage;

        let budget = Arc::new(WorkflowBudget::new(Some(10)));
        // Spend past the cap before the phase runs (simulates child loops having
        // already recorded usage into the shared ledger).
        budget
            .record_after_llm(
                "s",
                &TokenUsage {
                    total_tokens: 12,
                    ..Default::default()
                },
            )
            .await;

        let wf = Workflow::builder(Arc::new(EchoExecutor::new()))
            .with_budget(Arc::clone(&budget))
            .build();
        let mut rx = wf.subscribe();
        wf.phase("p", vec![spec("a", "explore", "p")]).await;

        let mut saw_exhausted = false;
        while let Ok(ev) = rx.try_recv() {
            if let WorkflowEvent::BudgetExhausted { resource, .. } = ev {
                assert_eq!(resource, "workflow_tokens");
                saw_exhausted = true;
            }
        }
        assert!(
            saw_exhausted,
            "phase boundary emits BudgetExhausted once capped"
        );
        assert_eq!(wf.budget_snapshot().unwrap().consumed_tokens, 12);
    }

    #[tokio::test]
    async fn no_budget_means_no_snapshot_and_no_exhausted_event() {