import json
import tempfile
import unittest
from pathlib import Path

from test_cli import _read_final_answer_from_raw, _read_final_answer_from_report


class CliHelperTests(unittest.TestCase):
    def test_read_final_answer_from_raw_returns_last_iteration_answer(self):
        with tempfile.TemporaryDirectory() as tmpdir:
            path = Path(tmpdir) / "result.json"
            path.write_text(json.dumps({
                "iterations": [
                    {"t": 1, "answer": "first"},
                    {"t": 2, "answer": "final answer"},
                ]
            }), encoding="utf-8")

            self.assertEqual(_read_final_answer_from_raw(str(path)), "final answer")

    def test_read_final_answer_from_raw_handles_missing_or_invalid_files(self):
        self.assertEqual(_read_final_answer_from_raw("/tmp/does-not-exist.json"), "")

        with tempfile.TemporaryDirectory() as tmpdir:
            path = Path(tmpdir) / "broken.json"
            path.write_text("{not json", encoding="utf-8")
            self.assertEqual(_read_final_answer_from_raw(str(path)), "")

    def test_read_final_answer_from_report_extracts_blockquoted_answer(self):
        report = (
            "## Summary\n\n"
            "Something.\n\n"
            "## Generated Answer\n\n"
            "> Line one\n"
            "> Line two\n\n"
            "## Next Section\n\n"
            "Trailing content."
        )
        self.assertEqual(_read_final_answer_from_report(report), "Line one\nLine two")

    def test_read_final_answer_from_report_returns_empty_when_missing(self):
        self.assertEqual(_read_final_answer_from_report("## Summary\n\nNothing here."), "")