import unittest
from unittest.mock import Mock, patch

from ocr_dashboard_service import export_evidences, get_comparison, get_summary, list_batches, list_results
from task_service import TaskError


class OcrDashboardServiceTest(unittest.TestCase):
    @patch("ocr_dashboard_service.SESSION.get")
    def test_list_batches_unwraps_framework_response(self, get):
        response = Mock()
        response.json.return_value = {"success": True, "code": 200, "data": [{"batch_key": "b1"}]}
        get.return_value = response
        self.assertEqual(list_batches()[0]["batch_key"], "b1")

    @patch("ocr_dashboard_service.SESSION.get")
    def test_summary_forwards_scope(self, get):
        response = Mock()
        response.json.return_value = {"success": True, "code": 200, "data": {"questions": []}}
        get.return_value = response
        get_summary("b1", "c1")
        self.assertEqual(get.call_args.kwargs["params"], {"batch_key": "b1", "collection_key": "c1"})

    @patch("ocr_dashboard_service.SESSION.get")
    def test_comparison_forwards_scope(self, get):
        response = Mock()
        response.json.return_value = {
            "success": True, "code": 200,
            "data": {"current": {}, "previous": {}},
        }
        get.return_value = response

        result = get_comparison("b1", "c1")

        self.assertIn("current", result)
        self.assertEqual(
            get.call_args.kwargs["params"],
            {"batch_key": "b1", "collection_key": "c1"},
        )

    def test_results_rejects_oversized_page(self):
        with self.assertRaises(TaskError):
            list_results({"page_size": "201"})

    @patch("ocr_dashboard_service.SESSION.get")
    @patch("ocr_dashboard_service.list_evidences")
    def test_export_evidences_builds_zip(self, list_evidences, get):
        list_evidences.return_value = {"items": [{
            "question_code": "Q1", "search_keyword": "山姆会员卡", "device_code": "DEV-1",
            "city": "厦门", "sort_order": 1, "source_file_name": "shot.jpg",
            "cos_url": "https://bucket.cos.ap-guangzhou.myqcloud.com/shot.jpg",
        }]}
        response = Mock(content=b"image-bytes")
        get.return_value = response

        payload, filename, count = export_evidences({"batch_key": "b1", "collection_key": "c1"})

        self.assertEqual(filename, "holy-crab-evidences-all.zip")
        self.assertEqual(count, 1)
        self.assertTrue(payload.read(2) == b"PK")


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