Spaces:
Running
Running
| import json | |
| import sys | |
| from unittest.mock import MagicMock | |
| sys.modules['sentence_transformers'] = MagicMock() | |
| sys.modules['torch'] = MagicMock() | |
| sys.modules['pathspec'] = MagicMock() | |
| sys.modules['watchdog'] = MagicMock() | |
| sys.modules['watchdog.observers'] = MagicMock() | |
| sys.modules['helpers.call_llm'] = MagicMock() | |
| sys.modules['helpers.persist_chat'] = MagicMock() | |
| sys.modules['models'] = MagicMock() | |
| sys.modules['helpers.mcp_server'] = MagicMock() | |
| from helpers.ui_server import UiServerRuntime, UiRouteHandlers | |
| def test_health_endpoint(): | |
| server = UiServerRuntime.create() | |
| server.register_http_routes() | |
| server.register_transport_handlers() | |
| client = server.webapp.test_client() | |
| response = client.get('/health') | |
| assert response.status_code == 200 | |
| data = json.loads(response.data) | |
| assert data.get("status") == "ok" | |
| def test_api_docs_endpoint(): | |
| server = UiServerRuntime.create() | |
| server.register_http_routes() | |
| server.register_transport_handlers() | |
| client = server.webapp.test_client() | |
| response = client.get('/api-docs') | |
| assert response.status_code == 200 | |
| data = json.loads(response.data) | |
| assert "endpoints" in data | |
| assert data["title"] == "Open Operator API Documentation" | |
| def test_api_docs_handler(): | |
| from api.api_docs import ApiDocs | |
| handler = ApiDocs(app=None, thread_lock=None) | |
| assert handler.requires_auth() is False | |
| assert handler.requires_csrf() is False | |
| assert "GET" in handler.get_methods() | |
| if __name__ == '__main__': | |
| test_health_endpoint() | |
| test_api_docs_endpoint() | |
| test_api_docs_handler() | |
| print("ALL TESTS PASSED SUCCESSFULLY!") | |