Fix imports

Author: Danielle Jenkins

src/lib.rs

@@ -1,5 +1,3 @@
 pub mod tools;
 pub mod transport;
 
-// Re-export key components for easier access
-pub use tools::DocRouter;

src/tools/docs/mod.rs

@@ -1,4 +1,6 @@
 pub mod docs;
-mod tests;
 
-pub use docs::DocRouter;
+pub use docs::DocRouter;
+
+#[cfg(test)]
+mod tests;

src/tools/docs/tests.rs

@@ -1,5 +1,5 @@
 use crate::tools::DocCache;
-use crate::DocRouter;
+use crate::tools::DocRouter;
 use mcp_core::{Content, ToolError};
 use serde_json::json;
 use mcp_server::Router;

src/transport/http_sse_server/mod.rs

@@ -1,4 +1,6 @@
 mod http_sse_server;
-mod tests;
 
-pub use http_sse_server::*;
+pub use http_sse_server::*;
+
+#[cfg(test)]
+mod tests;

src/transport/http_sse_server/tests.rs

@@ -23,12 +23,6 @@ async fn test_router_setup() {
     assert!(true);
 }
 
-// Test removed since session_id is private
-// #[tokio::test]
-// async fn test_session_id_generation() {
-//     // Test removed
-// }
-
 #[tokio::test]
 async fn test_post_event_handler_not_found() {
     let app = App::new();

src/transport/jsonrpc_frame_codec/mod.rs

@@ -1,4 +1,5 @@
 mod jsonrpc_frame_codec;
-mod tests;
+pub use jsonrpc_frame_codec::JsonRpcFrameCodec;
 
-pub use jsonrpc_frame_codec::JsonRpcFrameCodec;
+#[cfg(test)]
+mod tests;