From 9b52f76cf54468e4dda0719f1d086170df4ffa48 Mon Sep 17 00:00:00 2001
From: Danielle Jenkins <danielle@megacrit.com>
Date: Wed, 12 Mar 2025 18:28:54 -0700
Subject: [PATCH] Fix imports

---
 src/lib.rs                               | 2 --
 src/tools/docs/mod.rs                    | 6 ++++--
 src/tools/docs/tests.rs                  | 2 +-
 src/transport/http_sse_server/mod.rs     | 6 ++++--
 src/transport/http_sse_server/tests.rs   | 6 ------
 src/transport/jsonrpc_frame_codec/mod.rs | 5 +++--
 6 files changed, 12 insertions(+), 15 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index dff5956..4eb7aeb 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,5 +1,3 @@
 pub mod tools;
 pub mod transport;
 
-// Re-export key components for easier access
-pub use tools::DocRouter;
\ No newline at end of file
diff --git a/src/tools/docs/mod.rs b/src/tools/docs/mod.rs
index e6c5da1..3fbc501 100644
--- a/src/tools/docs/mod.rs
+++ b/src/tools/docs/mod.rs
@@ -1,4 +1,6 @@
 pub mod docs;
-mod tests;
 
-pub use docs::DocRouter;
\ No newline at end of file
+pub use docs::DocRouter;
+
+#[cfg(test)]
+mod tests;
\ No newline at end of file
diff --git a/src/tools/docs/tests.rs b/src/tools/docs/tests.rs
index df89cec..645f056 100644
--- a/src/tools/docs/tests.rs
+++ b/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;
diff --git a/src/transport/http_sse_server/mod.rs b/src/transport/http_sse_server/mod.rs
index 2a9abc6..2d05eac 100644
--- a/src/transport/http_sse_server/mod.rs
+++ b/src/transport/http_sse_server/mod.rs
@@ -1,4 +1,6 @@
 mod http_sse_server;
-mod tests;
 
-pub use http_sse_server::*;
\ No newline at end of file
+pub use http_sse_server::*;
+
+#[cfg(test)]
+mod tests;
\ No newline at end of file
diff --git a/src/transport/http_sse_server/tests.rs b/src/transport/http_sse_server/tests.rs
index 8ae95ae..701fc87 100644
--- a/src/transport/http_sse_server/tests.rs
+++ b/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();
diff --git a/src/transport/jsonrpc_frame_codec/mod.rs b/src/transport/jsonrpc_frame_codec/mod.rs
index e7fb1a2..5dd29d5 100644
--- a/src/transport/jsonrpc_frame_codec/mod.rs
+++ b/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;
\ No newline at end of file
+#[cfg(test)]
+mod tests;
\ No newline at end of file