mcp-stdio-proxy 0.2.9

MCP (Model Context Protocol) proxy server and CLI tool for protocol conversion and remote service access
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90

//! OpenAPI 文档路由测试：Swagger UI 与 Scalar 双风格并存、spec 完整性。

use axum::body::Body;
use axum::http::{Request, StatusCode};
use tower::ServiceExt;

use crate::AppConfig;
use crate::model::AppState;
use crate::server::get_router;

async fn build_router() -> axum::Router {
    let config = AppConfig::load_config().expect("加载默认配置失败");
    let state = AppState::new(config).await;
    get_router(state).await.expect("构建路由失败")
}

#[tokio::test]
async fn scalar_docs_page_returns_html_with_embedded_spec() {
    let app = build_router().await;
    let response = app
        .oneshot(
            Request::get("/api/docs/scalar")
                .body(Body::empty())
                .unwrap(),
        )
        .await
        .unwrap();
    assert_eq!(response.status(), StatusCode::OK);
    let content_type = response
        .headers()
        .get("content-type")
        .and_then(|v| v.to_str().ok())
        .unwrap_or_default()
        .to_string();
    assert!(
        content_type.starts_with("text/html"),
        "Scalar 页面应为 HTML，实际 content-type: {content_type}"
    );
    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .unwrap();
    let html = String::from_utf8(body.to_vec()).unwrap();
    // utoipa-scalar 将完整 OpenAPI spec 内嵌进 HTML
    assert!(
        html.contains("\"openapi\""),
        "Scalar HTML 应内嵌 OpenAPI spec"
    );
    assert!(html.contains("MCP Proxy API"));
}

#[tokio::test]
async fn swagger_openapi_json_covers_all_documented_paths() {
    let app = build_router().await;
    let response = app
        .oneshot(
            Request::get("/api/docs/openapi.json")
                .body(Body::empty())
                .unwrap(),
        )
        .await
        .unwrap();
    assert_eq!(response.status(), StatusCode::OK);
    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .unwrap();
    let spec: serde_json::Value = serde_json::from_slice(&body).unwrap();
    let paths = spec["paths"].as_object().expect("paths 应为对象");
    for expected in [
        "/health",
        "/ready",
        "/mcp/sse/add",
        "/mcp/config/delete/{mcp_id}",
        "/mcp/check/status/{mcp_id}",
        "/mcp/sse/check_status",
        "/mcp/stream/check_status",
        "/api/run_code_with_log",
    ] {
        assert!(paths.contains_key(expected), "文档缺少路径 {expected}");
    }
}

#[tokio::test]
async fn swagger_ui_still_served_alongside_scalar() {
    let app = build_router().await;
    let response = app
        .oneshot(Request::get("/api/docs/").body(Body::empty()).unwrap())
        .await
        .unwrap();
    assert_eq!(response.status(), StatusCode::OK);
}