Phase 7: Directory restructuring (messaging/ and tests/)

- Create messaging/platforms/ (base, discord, telegram, factory)
- Create messaging/rendering/ (discord_markdown, telegram_markdown)
- Create messaging/trees/ (data, repository, processor, queue_manager)
- Organize tests/ into api/, providers/, messaging/, cli/, config/
- Add backward-compatible re-exports at old locations
- Update handler.py and test_messaging_factory.py imports
- Fix Telegram type hints for TELEGRAM_AVAILABLE=False case
- Fix Python 3 except syntax in discord_markdown

Co-authored-by: Ali Khokhar <alishahryar2@gmail.com>

tests/api/test_routes_optimizations.py

@@ -0,0 +1,148 @@
+import pytest
+from unittest.mock import AsyncMock, MagicMock, patch
+from fastapi.testclient import TestClient
+from api.app import app
+from api.dependencies import get_settings
+from config.settings import Settings
+
+
+@pytest.fixture
+def client():
+    return TestClient(app)
+
+
+@pytest.fixture
+def mock_settings():
+    settings = Settings()
+    settings.fast_prefix_detection = True
+    settings.enable_network_probe_mock = True
+    settings.enable_title_generation_skip = True
+    return settings
+
+
+def test_create_message_fast_prefix_detection(client, mock_settings):
+    app.dependency_overrides[get_settings] = lambda: mock_settings
+
+    payload = {
+        "model": "claude-3-sonnet",
+        "max_tokens": 100,
+        "messages": [{"role": "user", "content": "What is the prefix?"}],
+    }
+
+    with patch(
+        "api.optimization_handlers.is_prefix_detection_request",
+        return_value=(True, "/ask"),
+    ):
+        with patch(
+            "api.optimization_handlers.extract_command_prefix",
+            return_value="/ask",
+        ):
+            response = client.post("/v1/messages", json=payload)
+
+    assert response.status_code == 200
+    data = response.json()
+    assert "/ask" in data["content"][0]["text"]
+
+    app.dependency_overrides.clear()
+
+
+def test_create_message_quota_check_mock(client, mock_settings):
+    app.dependency_overrides[get_settings] = lambda: mock_settings
+
+    payload = {
+        "model": "claude-3-sonnet",
+        "max_tokens": 100,
+        "messages": [{"role": "user", "content": "quota check"}],
+    }
+
+    with patch("api.optimization_handlers.is_quota_check_request", return_value=True):
+        response = client.post("/v1/messages", json=payload)
+
+    assert response.status_code == 200
+    assert "Quota check passed" in response.json()["content"][0]["text"]
+
+    app.dependency_overrides.clear()
+
+
+def test_create_message_title_generation_skip(client, mock_settings):
+    app.dependency_overrides[get_settings] = lambda: mock_settings
+
+    payload = {
+        "model": "claude-3-sonnet",
+        "max_tokens": 100,
+        "messages": [{"role": "user", "content": "generate title"}],
+    }
+
+    with patch(
+        "api.optimization_handlers.is_title_generation_request", return_value=True
+    ):
+        response = client.post("/v1/messages", json=payload)
+
+    assert response.status_code == 200
+    assert "Conversation" in response.json()["content"][0]["text"]
+
+    app.dependency_overrides.clear()
+
+
+def test_count_tokens_endpoint(client):
+    payload = {
+        "model": "claude-3-sonnet",
+        "messages": [{"role": "user", "content": "hello"}],
+    }
+
+    with patch("api.routes.get_token_count", return_value=5):
+        response = client.post("/v1/messages/count_tokens", json=payload)
+
+    assert response.status_code == 200
+    assert response.json()["input_tokens"] == 5
+
+
+def test_count_tokens_error_returns_500(client):
+    """When get_token_count raises, count_tokens returns 500."""
+    payload = {
+        "model": "claude-3-sonnet",
+        "messages": [{"role": "user", "content": "hello"}],
+    }
+
+    with patch("api.routes.get_token_count", side_effect=RuntimeError("token error")):
+        response = client.post("/v1/messages/count_tokens", json=payload)
+
+    assert response.status_code == 500
+    assert "token error" in response.json()["detail"]
+
+
+def test_stop_cli_with_handler(client):
+    mock_handler = MagicMock()
+    # Mock the async method to return a completed future or just mock it since TestClient
+    # will run the app in a way that respects it?
+    # Actually, we need to mock it as an async function.
+    mock_handler.stop_all_tasks = AsyncMock(return_value=3)
+    app.state.message_handler = mock_handler
+
+    response = client.post("/stop")
+
+    assert response.status_code == 200
+    assert response.json()["cancelled_count"] == 3
+    mock_handler.stop_all_tasks.assert_called_once()
+
+    # Cleanup state
+    if hasattr(app.state, "message_handler"):
+        del app.state.message_handler
+
+
+def test_stop_cli_fallback_to_manager(client):
+    if hasattr(app.state, "message_handler"):
+        del app.state.message_handler
+
+    mock_manager = MagicMock()
+    mock_manager.stop_all = AsyncMock()
+    app.state.cli_manager = mock_manager
+
+    response = client.post("/stop")
+
+    assert response.status_code == 200
+    assert response.json()["source"] == "cli_manager"
+    mock_manager.stop_all.assert_called_once()
+
+    if hasattr(app.state, "cli_manager"):
+        del app.state.cli_manager