modelcontextprotocol/rust-sdk
Watch
1
0
Fork
You've already forked rust-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

feat(model): add ServerConfig and ClientConfig (#1266)

Browse source
This commit is contained in:
Dale Seo 2026-09-14 15:30:41 -04:00 • committed by GitHub
commit 165ff92df1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 331 additions and 303 deletions
Download patch file Download diff file Expand all files Collapse all files

30
crates/rmcp/tests/test_custom_headers.rs
View file

@ -719,7 +719,7 @@ async fn test_server_rejects_unsupported_protocol_version() {
use http_body_util::Full;
use rmcp::{
handler::server::ServerHandler,
model::{InitializeResult, ServerCapabilities},
model::{ServerCapabilities, ServerConfig},
transport::streamable_http_server::{
StreamableHttpServerConfig, StreamableHttpService, session::local::LocalSessionManager,
},
@ -730,8 +730,8 @@ async fn test_server_rejects_unsupported_protocol_version() {
struct TestHandler;
impl ServerHandler for TestHandler {
fn get_info(&self) -> InitializeResult {
InitializeResult::new(ServerCapabilities::builder().build())
fn get_info(&self) -> ServerConfig {
ServerConfig::new(ServerCapabilities::builder().build())
}
}
@ -892,7 +892,7 @@ async fn test_server_validates_host_when_origin_validation_is_disabled_by_defaul
use http_body_util::Full;
use rmcp::{
handler::server::ServerHandler,
model::{InitializeResult, ServerCapabilities},
model::{ServerCapabilities, ServerConfig},
transport::streamable_http_server::{
StreamableHttpServerConfig, StreamableHttpService, session::local::LocalSessionManager,
},
@ -903,8 +903,8 @@ async fn test_server_validates_host_when_origin_validation_is_disabled_by_defaul
struct TestHandler;
impl ServerHandler for TestHandler {
fn get_info(&self) -> InitializeResult {
InitializeResult::new(ServerCapabilities::builder().build())
fn get_info(&self) -> ServerConfig {
ServerConfig::new(ServerCapabilities::builder().build())
}
}
@ -975,7 +975,7 @@ async fn test_server_validates_host_header_port_for_dns_rebinding_protection() {
use http_body_util::Full;
use rmcp::{
handler::server::ServerHandler,
model::{InitializeResult, ServerCapabilities},
model::{ServerCapabilities, ServerConfig},
transport::streamable_http_server::{
StreamableHttpServerConfig, StreamableHttpService, session::local::LocalSessionManager,
},
@ -986,8 +986,8 @@ async fn test_server_validates_host_header_port_for_dns_rebinding_protection() {
struct TestHandler;
impl ServerHandler for TestHandler {
fn get_info(&self) -> InitializeResult {
InitializeResult::new(ServerCapabilities::builder().build())
fn get_info(&self) -> ServerConfig {
ServerConfig::new(ServerCapabilities::builder().build())
}
}
@ -1046,7 +1046,7 @@ async fn test_server_falls_back_to_uri_authority_when_host_header_missing() {
use http_body_util::Full;
use rmcp::{
handler::server::ServerHandler,
model::{InitializeResult, ServerCapabilities},
model::{ServerCapabilities, ServerConfig},
transport::streamable_http_server::{
StreamableHttpServerConfig, StreamableHttpService, session::local::LocalSessionManager,
},
@ -1057,8 +1057,8 @@ async fn test_server_falls_back_to_uri_authority_when_host_header_missing() {
struct TestHandler;
impl ServerHandler for TestHandler {
fn get_info(&self) -> InitializeResult {
InitializeResult::new(ServerCapabilities::builder().build())
fn get_info(&self) -> ServerConfig {
ServerConfig::new(ServerCapabilities::builder().build())
}
}
@ -1132,7 +1132,7 @@ mod origin_validation {
use http_body_util::Full;
use rmcp::{
handler::server::ServerHandler,
model::{InitializeResult, ServerCapabilities},
model::{ServerCapabilities, ServerConfig},
transport::streamable_http_server::{
StreamableHttpServerConfig, StreamableHttpService, session::local::LocalSessionManager,
},
@ -1143,8 +1143,8 @@ mod origin_validation {
struct TestHandler;
impl ServerHandler for TestHandler {
fn get_info(&self) -> InitializeResult {
InitializeResult::new(ServerCapabilities::builder().build())
fn get_info(&self) -> ServerConfig {
ServerConfig::new(ServerCapabilities::builder().build())
}
}