fmt

Author: Hendler

crates/rmcp/tests/test_tool_macros.rs

@@ -1,21 +1,14 @@
 //cargo test --test test_tool_macros --features "client server"
 
-use std::sync::Arc;
 use rmcp::{
-    ClientHandler, 
-    Peer, 
-    RoleClient, 
-    ServerHandler, 
-    ServiceExt,
-    model::{
-        CallToolRequestParam,
-        ClientInfo,
-    },
+    ClientHandler, Peer, RoleClient, ServerHandler, ServiceExt,
+    model::{CallToolRequestParam, ClientInfo},
 };
 use rmcp::{handler::server::tool::ToolCallContext, tool};
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
 use serde_json;
+use std::sync::Arc;
 
 #[derive(Serialize, Deserialize, JsonSchema)]
 pub struct GetWeatherRequest {
@@ -143,10 +136,7 @@ impl OptionalSchemaTester {
 
     // Tool function to test optional i64 handling
     #[tool(description = "A tool to test optional i64 schema generation")]
-    async fn test_optional_i64_aggr(
-        &self,
-        #[tool(aggr)] req: OptionalI64TestSchema,
-    ) -> String {
+    async fn test_optional_i64_aggr(&self, #[tool(aggr)] req: OptionalI64TestSchema) -> String {
         match req.count {
             Some(c) => format!("Received count: {}", c),
             None => "Received null count".to_string(),
@@ -255,9 +245,7 @@ async fn test_optional_i64_field_with_null_input() -> anyhow::Result<()> {
     // Server setup
     let server = OptionalSchemaTester::default();
     let server_handle = tokio::spawn(async move {
-        server.serve(server_transport).await?
-            .waiting()
-            .await?;
+        server.serve(server_transport).await?.waiting().await?;
         anyhow::Ok(())
     });
 
@@ -266,52 +254,62 @@ async fn test_optional_i64_field_with_null_input() -> anyhow::Result<()> {
     let client = client_handler.serve(client_transport).await?;
 
     // Test null case
-    let result = client.call_tool(
-        CallToolRequestParam {
+    let result = client
+        .call_tool(CallToolRequestParam {
             name: "test_optional_i64_aggr".into(),
-            arguments: Some(serde_json::json!({
-                "count": null,
-                "mandatory_field": "test_null"
-            }).as_object().unwrap().clone()),
-        }
-    ).await?;
-
-    let result_text = result.content
+            arguments: Some(
+                serde_json::json!({
+                    "count": null,
+                    "mandatory_field": "test_null"
+                })
+                .as_object()
+                .unwrap()
+                .clone(),
+            ),
+        })
+        .await?;
+
+    let result_text = result
+        .content
         .first()
         .and_then(|content| content.raw.as_text())
         .map(|text| text.text.as_str())
         .expect("Expected text content");
 
     assert_eq!(
-        result_text,
-        "Received null count",
+        result_text, "Received null count",
         "Null case should return expected message"
     );
 
     // Test Some case
-    let some_result = client.call_tool(
-        CallToolRequestParam {
+    let some_result = client
+        .call_tool(CallToolRequestParam {
             name: "test_optional_i64_aggr".into(),
-            arguments: Some(serde_json::json!({
-                "count": 42,
-                "mandatory_field": "test_some"
-            }).as_object().unwrap().clone()),
-        }
-    ).await?;
-
-    let some_result_text = some_result.content
+            arguments: Some(
+                serde_json::json!({
+                    "count": 42,
+                    "mandatory_field": "test_some"
+                })
+                .as_object()
+                .unwrap()
+                .clone(),
+            ),
+        })
+        .await?;
+
+    let some_result_text = some_result
+        .content
         .first()
         .and_then(|content| content.raw.as_text())
         .map(|text| text.text.as_str())
         .expect("Expected text content");
 
     assert_eq!(
-        some_result_text,
-        "Received count: 42",
+        some_result_text, "Received count: 42",
         "Some case should return expected message"
     );
 
     client.cancel().await?;
     server_handle.await??;
     Ok(())
-}
+}