From 033ebf5038ed473e397312f5f7622762f66b57f4 Mon Sep 17 00:00:00 2001
From: Alexey <247128645+axkurcom@users.noreply.github.com>
Date: Fri, 24 Apr 2026 15:51:19 +0300
Subject: [PATCH] Relays Tests Fixes

---
 src/proxy/middle_relay.rs | 1 +
 src/proxy/relay.rs        | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/proxy/middle_relay.rs b/src/proxy/middle_relay.rs
index aff2eb7..f1f6584 100644
--- a/src/proxy/middle_relay.rs
+++ b/src/proxy/middle_relay.rs
@@ -2201,6 +2201,7 @@ enum MeWriterResponseOutcome {
     Close,
 }
 
+#[cfg(test)]
 async fn process_me_writer_response<W>(
     response: MeResponse,
     client_writer: &mut CryptoWriter<W>,
diff --git a/src/proxy/relay.rs b/src/proxy/relay.rs
index c9e6a98..117d158 100644
--- a/src/proxy/relay.rs
+++ b/src/proxy/relay.rs
@@ -230,6 +230,7 @@ struct RateWaitState {
 }
 
 impl<S> StatsIo<S> {
+    #[cfg(test)]
     fn new(
         inner: S,
         counters: Arc<SharedCounters>,