rust-sdk/crates/rmcp/tests/test_with_python.rs
adam jones eb5a7f7408
feat!: remove SSE transport support (#562)
SSE transport has been removed from the MCP specification in favor of
streamable HTTP. This removes all SSE-specific transport code:

- Remove `transport-sse-client` and `transport-sse-server` features
- Remove `SseClientTransport` and `SseServer` types
- Remove SSE-specific examples (`counter_sse`, `counter_sse_directly`)
- Migrate auth examples from SSE to streamable HTTP
- Update tests to remove SSE transport usage
- Update documentation

BREAKING CHANGE: The following have been removed:
- `transport-sse-client` feature
- `transport-sse-client-reqwest` feature
- `transport-sse-server` feature
- `SseClientTransport` type
- `SseServer` type
- `sse_client` and `sse_server` modules

Users should migrate to streamable HTTP transport which provides
equivalent functionality. See `StreamableHttpClientTransport` and
`StreamableHttpService` for the replacement APIs.

Ref: https://github.com/modelcontextprotocol/rust-sdk/pull/561#issuecomment-3576551699
2025-12-01 20:40:13 -05:00

77 lines
2.2 KiB
Rust

use std::process::Stdio;
use rmcp::{
ServiceExt,
transport::{ConfigureCommandExt, TokioChildProcess},
};
use tokio::io::AsyncReadExt;
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
mod common;
async fn init() -> anyhow::Result<()> {
let _ = tracing_subscriber::registry()
.with(
tracing_subscriber::EnvFilter::try_from_default_env()
.unwrap_or_else(|_| "debug".to_string().into()),
)
.with(tracing_subscriber::fmt::layer())
.try_init();
tokio::process::Command::new("uv")
.args(["sync"])
.current_dir("tests/test_with_python")
.spawn()?
.wait()
.await?;
Ok(())
}
#[tokio::test]
async fn test_with_python_server() -> anyhow::Result<()> {
init().await?;
let transport = TokioChildProcess::new(tokio::process::Command::new("uv").configure(|cmd| {
cmd.arg("run")
.arg("server.py")
.current_dir("tests/test_with_python");
}))?;
let client = ().serve(transport).await?;
let resources = client.list_all_resources().await?;
tracing::info!("{:#?}", resources);
let tools = client.list_all_tools().await?;
tracing::info!("{:#?}", tools);
client.cancel().await?;
Ok(())
}
#[tokio::test]
async fn test_with_python_server_stderr() -> anyhow::Result<()> {
init().await?;
let (transport, stderr) =
TokioChildProcess::builder(tokio::process::Command::new("uv").configure(|cmd| {
cmd.arg("run")
.arg("server.py")
.current_dir("tests/test_with_python");
}))
.stderr(Stdio::piped())
.spawn()?;
let mut stderr = stderr.expect("stderr must be piped");
let stderr_task = tokio::spawn(async move {
let mut buffer = String::new();
stderr.read_to_string(&mut buffer).await?;
Ok::<_, std::io::Error>(buffer)
});
let client = ().serve(transport).await?;
let _ = client.list_all_resources().await?;
let _ = client.list_all_tools().await?;
client.cancel().await?;
let stderr_output = stderr_task.await??;
assert!(stderr_output.contains("server starting up..."));
Ok(())
}