tests: add new tests for client runtime

Author: hashemix

crates/rust-mcp-sdk/tests/common/common.rs

@@ -0,0 +1,24 @@
+use async_trait::async_trait;
+use rust_mcp_schema::{
+    ClientCapabilities, Implementation, InitializeRequestParams, JSONRPC_VERSION,
+};
+use rust_mcp_sdk::mcp_client::ClientHandler;
+
+pub const NPX_SERVER_EVERYTHING: &str = "@modelcontextprotocol/server-everything";
+pub const UVX_SERVER_GIT: &str = "mcp-server-git";
+
+pub fn test_client_info() -> InitializeRequestParams {
+    InitializeRequestParams {
+        capabilities: ClientCapabilities::default(),
+        client_info: Implementation {
+            name: "test-rust-mcp-client".into(),
+            version: "0.1.0".into(),
+        },
+        protocol_version: JSONRPC_VERSION.into(),
+    }
+}
+
+pub struct TestClientHandler;
+
+#[async_trait]
+impl ClientHandler for TestClientHandler {}

crates/rust-mcp-sdk/tests/test_client_runtime.rs

@@ -0,0 +1,52 @@
+use common::{test_client_info, TestClientHandler, NPX_SERVER_EVERYTHING, UVX_SERVER_GIT};
+use rust_mcp_sdk::{mcp_client::client_runtime, McpClient, StdioTransport, TransportOptions};
+
+#[path = "common/common.rs"]
+pub mod common;
+
+#[tokio::test]
+async fn tets_client_launch_npx_server() {
+    // NPM based MCP servers should launch successfully using `npx`
+    let transport = StdioTransport::create_with_server_launch(
+        "npx",
+        vec!["-y".into(), NPX_SERVER_EVERYTHING.into()],
+        None,
+        TransportOptions::default(),
+    )
+    .unwrap();
+
+    let client = client_runtime::create_client(test_client_info(), transport, TestClientHandler {});
+
+    client.clone().start().await.unwrap();
+
+    let server_capabilities = client.server_capabilities().unwrap();
+    let server_info = client.server_info().unwrap();
+
+    assert!(server_info.server_info.name.len() > 0);
+    assert!(server_info.server_info.version.len() > 0);
+    assert!(server_capabilities.tools.is_some());
+}
+
+#[tokio::test]
+async fn tets_client_launch_uvx_server() {
+    // The Python-based MCP server should launch successfully
+    // provided that `uvx` is installed and accessible in the system's PATH
+    let transport = StdioTransport::create_with_server_launch(
+        "uvx",
+        vec![UVX_SERVER_GIT.into()],
+        None,
+        TransportOptions::default(),
+    )
+    .unwrap();
+
+    let client = client_runtime::create_client(test_client_info(), transport, TestClientHandler {});
+
+    client.clone().start().await.unwrap();
+
+    let server_capabilities = client.server_capabilities().unwrap();
+    let server_info = client.server_info().unwrap();
+
+    assert!(server_info.server_info.name.len() > 0);
+    assert!(server_info.server_info.version.len() > 0);
+    assert!(server_capabilities.tools.is_some());
+}