Merge pull request #1466 from adespawn/add-translation-policies-helpers

Add a UntranslatedPeer constructor

Author: wprzytula

.github/workflows/rust.yml

@@ -50,6 +50,10 @@ jobs:
     - name: Cargo check with cpp_rust_unstable cfg
       run: RUSTFLAGS="--cfg cpp_rust_unstable -Dwarnings" cargo clippy --all-targets --all-features
 
+# nodejs_rs_unstable special cfg
+    - name: Cargo check with nodejs_rs_unstable cfg
+      run: RUSTFLAGS="--cfg nodejs_rs_unstable -Dwarnings" cargo clippy --all-targets --all-features
+
 # Features checks.
     # No features.
     - name: Cargo check without features

scylla/Cargo.toml

@@ -157,6 +157,7 @@ unexpected_cfgs = { level = "warn", check-cfg = [
     'cfg(scylla_cloud_tests)',
     'cfg(cassandra_tests)',
     'cfg(cpp_rust_unstable)',
+    'cfg(nodejs_rs_unstable)',
     'cfg(ccm_tests)',
 ] }
 

scylla/src/policies/address_translator.rs

@@ -49,6 +49,18 @@ impl UntranslatedPeer<'_> {
     pub fn rack(&self) -> Option<&str> {
         self.rack
     }
+
+    /// Creates a new untranslated peer. Intended to be used only in nodejs-rs driver only.
+    /// This specific constructor is intended for unit tests only.
+    #[cfg(nodejs_rs_unstable)]
+    pub fn new(address: SocketAddr) -> Self {
+        UntranslatedPeer {
+            host_id: Uuid::nil(),
+            untranslated_address: address,
+            datacenter: None,
+            rack: None,
+        }
+    }
 }
 
 /// Translates IP addresses received from ScyllaDB nodes into locally reachable addresses.