dd30a70f84
* feat!: add MRTR behavior support * feat: harden SEP-2322 MRTR support * ci: diff public API only on features common to base and head cargo public-api builds both revisions with the same feature set, so a feature introduced (or removed) by a PR broke the build of the other revision. Restrict the all-features diff to features present in both the base and head revisions.
91 lines
2.5 KiB
Rust
91 lines
2.5 KiB
Rust
//! SEP-2164: the resource-not-found error code follows the negotiated protocol version.
|
|
//!
|
|
//! `2026-07-28` and newer get the standard `INVALID_PARAMS` (-32602); older versions
|
|
//! keep the legacy `RESOURCE_NOT_FOUND` (-32002).
|
|
#![cfg(not(feature = "local"))]
|
|
#![cfg(feature = "client")]
|
|
|
|
use rmcp::{
|
|
ClientHandler, RoleServer, ServerHandler, ServiceError, ServiceExt,
|
|
model::{
|
|
ClientInfo, ErrorCode, ErrorData, ProtocolVersion, ReadResourceRequestParams,
|
|
ReadResourceResponse,
|
|
},
|
|
service::RequestContext,
|
|
};
|
|
|
|
#[derive(Debug, Clone, Default)]
|
|
struct ResourceServer;
|
|
|
|
impl ServerHandler for ResourceServer {
|
|
async fn read_resource(
|
|
&self,
|
|
_request: ReadResourceRequestParams,
|
|
_context: RequestContext<RoleServer>,
|
|
) -> Result<ReadResourceResponse, ErrorData> {
|
|
Err(ErrorData::resource_not_found("resource not found", None))
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
struct VersionedClient {
|
|
protocol_version: ProtocolVersion,
|
|
}
|
|
|
|
impl ClientHandler for VersionedClient {
|
|
fn get_info(&self) -> ClientInfo {
|
|
let mut info = ClientInfo::default();
|
|
info.protocol_version = self.protocol_version.clone();
|
|
info
|
|
}
|
|
}
|
|
|
|
async fn not_found_code(client_version: ProtocolVersion) -> ErrorCode {
|
|
let (server_transport, client_transport) = tokio::io::duplex(4096);
|
|
|
|
let server_handle = tokio::spawn(async move {
|
|
ResourceServer
|
|
.serve(server_transport)
|
|
.await?
|
|
.waiting()
|
|
.await?;
|
|
anyhow::Ok(())
|
|
});
|
|
|
|
let client = VersionedClient {
|
|
protocol_version: client_version,
|
|
}
|
|
.serve(client_transport)
|
|
.await
|
|
.expect("client should connect");
|
|
|
|
let error = client
|
|
.read_resource(ReadResourceRequestParams::new("missing://resource"))
|
|
.await
|
|
.expect_err("missing resource should error");
|
|
|
|
let code = match error {
|
|
ServiceError::McpError(data) => data.code,
|
|
other => panic!("expected McpError, got: {other:?}"),
|
|
};
|
|
|
|
client.cancel().await.expect("client should cancel");
|
|
server_handle.await.expect("server task").expect("server");
|
|
code
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn legacy_version_gets_resource_not_found_code() {
|
|
assert_eq!(
|
|
not_found_code(ProtocolVersion::V_2025_11_25).await,
|
|
ErrorCode::RESOURCE_NOT_FOUND,
|
|
);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn sep_2164_version_gets_invalid_params_code() {
|
|
assert_eq!(
|
|
not_found_code(ProtocolVersion::V_2026_07_28).await,
|
|
ErrorCode::INVALID_PARAMS,
|
|
);
|
|
}
|