from __future__ import annotations

from opensquilla.engine.outcome import outcome_from_error


def test_max_iterations_is_partial_not_failed() -> None:
    outcome = outcome_from_error(code="max_iterations", message="hit cap")

    assert outcome.kind == "partial"
    assert outcome.reason == "max_iterations"
    assert outcome.error_class == "max_iterations"
    assert outcome.error_message == "hit cap"


def test_provider_request_budget_is_budget_limited_and_retryable() -> None:
    outcome = outcome_from_error(code="provider_request_too_large")

    assert outcome.kind == "budgetLimited"
    assert outcome.retryable is True


def test_unknown_error_remains_failed() -> None:
    outcome = outcome_from_error(code="boom")

    assert outcome.kind == "failed"
    assert outcome.retryable is False