import tempfile
import unittest
from pathlib import Path

from config import Config
from database import connection, init_database, now_iso
from ocr_dashboard_service import _overlay_acceptance_questions


class OcrDashboardOverlayTest(unittest.TestCase):
    def setUp(self):
        self.temp_dir = tempfile.TemporaryDirectory()
        self.original_database_path = Config.DATABASE_PATH
        Config.DATABASE_PATH = Path(self.temp_dir.name) / "holy.db"
        init_database()
        now = now_iso()
        with connection() as conn:
            task_id = conn.execute(
                """INSERT INTO acceptance_tasks
                   (public_task_id,name,acceptance_method,status,progress_percent,device_count,
                    created_by_id,created_by_name,created_at,updated_at,dashboard_batch_key)
                   VALUES ('root','民生银行','combined','completed',100,1,'u','u',?,?,?)""",
                (now, now, "batch-1"),
            ).lastrowid
            conn.executemany(
                "INSERT INTO acceptance_keywords(task_id,keyword_order,keyword,created_at) VALUES (?,?,?,?)",
                [(task_id, 0, "修改后的第一题", now), (task_id, 1, "第二题", now)],
            )

    def tearDown(self):
        Config.DATABASE_PATH = self.original_database_path
        self.temp_dir.cleanup()

    def test_current_acceptance_questions_replace_historical_list(self):
        data = {"questions": [
            {"question_code": "Q1", "search_keyword": "旧的第一题", "result_count": 10},
            {"question_code": "Q2", "search_keyword": "第二题", "result_count": 8},
        ]}

        result = _overlay_acceptance_questions(data, "batch-1")

        self.assertEqual([item["search_keyword"] for item in result["questions"]], ["修改后的第一题", "第二题"])
        self.assertEqual(result["questions"][0]["result_count"], 0)
        self.assertEqual(result["questions"][1]["result_count"], 8)


if __name__ == "__main__":
    unittest.main()