test: split server init tests into single-assertion cases

Author: DaleSeo

crates/rmcp/tests/test_server_initialization.rs

@@ -54,12 +54,11 @@ async fn do_initialize(client: &mut impl Transport<rmcp::RoleClient>) {
     let _response = client.receive().await.unwrap();
 }
 
-/// Server accepts logging/setLevel before initialized and responds with EmptyResult.
+// Server responds with EmptyResult to setLevel received before initialized.
 #[tokio::test]
-async fn server_init_accepts_set_level_before_initialized() {
+async fn server_init_set_level_response_is_empty_result() {
     let (server_transport, client_transport) = tokio::io::duplex(4096);
-    let server_handle =
-        tokio::spawn(async move { TestServer::new().serve(server_transport).await });
+    let _server = tokio::spawn(async move { TestServer::new().serve(server_transport).await });
     let mut client = IntoTransport::<rmcp::RoleClient, _, _>::into_transport(client_transport);
 
     do_initialize(&mut client).await;
@@ -74,8 +73,21 @@ async fn server_init_accepts_set_level_before_initialized() {
         ),
         "expected EmptyResult for setLevel, got: {response:?}"
     );
+}
 
+// Server initializes successfully when setLevel is sent before the initialized notification.
+#[tokio::test]
+async fn server_init_succeeds_after_set_level_before_initialized() {
+    let (server_transport, client_transport) = tokio::io::duplex(4096);
+    let server_handle =
+        tokio::spawn(async move { TestServer::new().serve(server_transport).await });
+    let mut client = IntoTransport::<rmcp::RoleClient, _, _>::into_transport(client_transport);
+
+    do_initialize(&mut client).await;
+    client.send(set_level_request(2)).await.unwrap();
+    let _response = client.receive().await.unwrap();
     client.send(initialized_notification()).await.unwrap();
+
     let result = server_handle.await.unwrap();
     assert!(
         result.is_ok(),
@@ -84,12 +96,11 @@ async fn server_init_accepts_set_level_before_initialized() {
     result.unwrap().cancel().await.unwrap();
 }
 
-/// Server accepts ping before initialized and responds with EmptyResult.
+// Server responds with EmptyResult to ping received before initialized.
 #[tokio::test]
-async fn server_init_accepts_ping_before_initialized() {
+async fn server_init_ping_response_is_empty_result() {
     let (server_transport, client_transport) = tokio::io::duplex(4096);
-    let server_handle =
-        tokio::spawn(async move { TestServer::new().serve(server_transport).await });
+    let _server = tokio::spawn(async move { TestServer::new().serve(server_transport).await });
     let mut client = IntoTransport::<rmcp::RoleClient, _, _>::into_transport(client_transport);
 
     do_initialize(&mut client).await;
@@ -104,8 +115,21 @@ async fn server_init_accepts_ping_before_initialized() {
         ),
         "expected EmptyResult for ping, got: {response:?}"
     );
+}
 
+// Server initializes successfully when ping is sent before the initialized notification.
+#[tokio::test]
+async fn server_init_succeeds_after_ping_before_initialized() {
+    let (server_transport, client_transport) = tokio::io::duplex(4096);
+    let server_handle =
+        tokio::spawn(async move { TestServer::new().serve(server_transport).await });
+    let mut client = IntoTransport::<rmcp::RoleClient, _, _>::into_transport(client_transport);
+
+    do_initialize(&mut client).await;
+    client.send(ping_request(2)).await.unwrap();
+    let _response = client.receive().await.unwrap();
     client.send(initialized_notification()).await.unwrap();
+
     let result = server_handle.await.unwrap();
     assert!(
         result.is_ok(),
@@ -114,7 +138,7 @@ async fn server_init_accepts_ping_before_initialized() {
     result.unwrap().cancel().await.unwrap();
 }
 
-/// Server rejects any message other than setLevel/ping/initialized before initialized.
+// Server returns ExpectedInitializedNotification for any other message before initialized.
 #[tokio::test]
 async fn server_init_rejects_unexpected_message_before_initialized() {
     let (server_transport, client_transport) = tokio::io::duplex(4096);