from __future__ import annotations

from nanobot.agent.verification_state import (
    analyze_verification_result,
    append_verification_feedback,
)


def test_analyze_pytest_failure_extracts_actionable_summary():
    output = """\
FAILED ../tests/test_outputs.py::test_regex_matches_dates - AssertionError: Expected dates
E       AssertionError: Expected ['2025-01-09'], but got ['bad']
E       FileNotFoundError: [Errno 2] No such file or directory: '/app/out.txt'
============================== 1 failed in 0.05s ===============================
Exit code: 1
"""

    analysis = analyze_verification_result(
        command="pytest /tests/test_outputs.py",
        output=output,
        exit_code=1,
    )

    assert analysis is not None
    assert analysis.status == "failed"
    assert analysis.failed_tests == ("../tests/test_outputs.py::test_regex_matches_dates",)
    assert any("AssertionError" in item for item in analysis.primary_errors)
    assert "/app/out.txt" in analysis.missing_artifacts


def test_append_verification_feedback_tells_agent_not_to_finish():
    analysis = analyze_verification_result(
        command="python /app/test_outputs.py",
        output="FAILED test_outputs.py::test_file\nAssertionError: missing\nExit code: 1",
        exit_code=1,
    )

    feedback = append_verification_feedback("raw output\nExit code: 1", analysis)

    assert "[Verification Feedback]" in feedback
    assert "Do not call complete_goal" in feedback
    assert "Next action" in feedback


def test_analyze_passing_test_records_success_without_feedback():
    analysis = analyze_verification_result(
        command="pytest",
        output="============================== 3 passed in 0.10s ==============================\nExit code: 0",
        exit_code=0,
    )

    assert analysis is not None
    assert analysis.status == "passed"
    assert append_verification_feedback("ok", analysis) == "ok"