33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
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()
|