Restore YG rules generation flow
This commit is contained in:
32
tests/test_schema_routes.py
Normal file
32
tests/test_schema_routes.py
Normal file
@@ -0,0 +1,32 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user