Unpublicize get_obfuscator()

Since RelaySelector::get_obfuscator() is only ever called from tests, it
only makes sense to only compile it when testing.

1 files changed, 11 insertions, 9 deletions

diff --git a/mullvad-relay-selector/src/lib.rs b/mullvad-relay-selector/src/lib.rs
index e62c63f3db..2093ccf07e 100644
--- a/mullvad-relay-selector/src/lib.rs
+++ b/mullvad-relay-selector/src/lib.rs
@@ -928,15 +928,6 @@ impl RelaySelector {
         })
     }
 
-    pub fn get_obfuscator(
-        &self,
-        relay: &Relay,
-        endpoint: &MullvadWireguardEndpoint,
-        retry_attempt: u32,
-    ) -> Result<Option<SelectedObfuscator>, Error> {
-        self.get_obfuscator_inner(&self.config.lock(), relay, endpoint, retry_attempt)
-    }
-
     fn get_obfuscator_inner(
         &self,
         config: &MutexGuard<'_, SelectorConfig>,
@@ -1311,6 +1302,17 @@ mod test {
     use std::collections::HashSet;
     use talpid_types::net::{wireguard::PublicKey, Endpoint};
 
+    impl RelaySelector {
+        fn get_obfuscator(
+            &self,
+            relay: &Relay,
+            endpoint: &MullvadWireguardEndpoint,
+            retry_attempt: u32,
+        ) -> Result<Option<SelectedObfuscator>, Error> {
+            self.get_obfuscator_inner(&self.config.lock(), relay, endpoint, retry_attempt)
+        }
+    }
+
     lazy_static::lazy_static! {
         static ref RELAYS: RelayList = RelayList {
             etag: None,