package pipeline

import "testing"

func TestBudgetModelRefinementSource(t *testing.T) {
	t.Run("uses refinement max concurrent when tighter", func(t *testing.T) {
		policy := Policy{MaxRefinementJobs: 12, RefinementMaxConcurrent: 4}
		budget := BudgetModelFromPolicy(policy)
		if budget.Refinement.Max != 4 {
			t.Fatalf("expected refinement max 4, got %d", budget.Refinement.Max)
		}
		if budget.Refinement.Source != "refinement.max_concurrent" {
			t.Fatalf("expected refinement source refinement.max_concurrent, got %s", budget.Refinement.Source)
		}
	})
	t.Run("keeps resources budget when smaller", func(t *testing.T) {
		policy := Policy{MaxRefinementJobs: 3, RefinementMaxConcurrent: 8}
		budget := BudgetModelFromPolicy(policy)
		if budget.Refinement.Max != 3 {
			t.Fatalf("expected refinement max 3, got %d", budget.Refinement.Max)
		}
		if budget.Refinement.Source != "resources.max_refinement_jobs" {
			t.Fatalf("expected refinement source resources.max_refinement_jobs, got %s", budget.Refinement.Source)
		}
	})
}