import unittest
from unittest.mock import patch

from flask import Flask

from app.routes.schema import schema_bp


class SchemaRouteTest(unittest.TestCase):
    def setUp(self):
        self.app = Flask(__name__)
        self.app.register_blueprint(schema_bp)
        self.client = self.app.test_client()

    def test_schema_status_returns_processing_summary(self):
        with patch("app.routes.schema.SchemaStorage") as storage_cls:
            storage_cls.return_value.status.return_value = {
                "processing_status": "done",
                "module_count": 2,
                "uploaded_at": "2026-06-10 15:50:00",
            }

            response = self.client.get("/api/schema/status")

        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.get_json()["message"], "查询成功")
        self.assertEqual(response.get_json()["data"]["processing_status"], "done")
        self.assertEqual(response.get_json()["data"]["module_count"], 2)


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