lint

Author: d4l3k

src/interceptor.rs

@@ -1,4 +1,4 @@
-use tonic::{service::Interceptor, metadata::MetadataValue, Request, Status};
+use tonic::{metadata::MetadataValue, service::Interceptor, Request, Status};
 
 /// Attaches user-assigned room-id header to every outbound RPC
 #[derive(Clone)]

src/lighthouse.rs

@@ -977,7 +977,7 @@ mod tests {
             quorum_tick_ms: 10,
             heartbeat_timeout_ms: 5000,
         };
-        let lighthouse = Lighthouse::new(opt).await?;
+        let lighthouse = Lighthouse::new("".to_string(), opt).await?;
 
         let lighthouse_task = tokio::spawn(lighthouse.clone().run());
 
@@ -1135,7 +1135,7 @@ mod tests {
         };
 
         // Start the lighthouse service
-        let lighthouse = Lighthouse::new(opt).await?;
+        let lighthouse = Lighthouse::new("".to_string(), opt).await?;
         let lighthouse_task = tokio::spawn(lighthouse.clone().run());
 
         // Create client to interact with lighthouse
@@ -1242,7 +1242,7 @@ mod tests {
         };
 
         // Start the lighthouse service
-        let lighthouse = Lighthouse::new(opt).await?;
+        let lighthouse = Lighthouse::new("".to_string(), opt).await?;
         let lighthouse_task = tokio::spawn(lighthouse.clone().run());
 
         // Create client to interact with lighthouse

src/manager.rs

@@ -544,13 +544,16 @@ mod tests {
 
     #[tokio::test]
     async fn test_should_commit() -> Result<()> {
-        let lighthouse = Lighthouse::new(LighthouseOpt {
-            bind: "[::]:0".to_string(),
-            join_timeout_ms: 100,
-            min_replicas: 1,
-            quorum_tick_ms: 100,
-            heartbeat_timeout_ms: 5000,
-        })
+        let lighthouse = Lighthouse::new(
+            "".to_string(),
+            LighthouseOpt {
+                bind: "[::]:0".to_string(),
+                join_timeout_ms: 100,
+                min_replicas: 1,
+                quorum_tick_ms: 100,
+                heartbeat_timeout_ms: 5000,
+            },
+        )
         .await?;
         let lighthouse_fut = tokio::spawn(lighthouse.clone().run());
 
@@ -591,13 +594,16 @@ mod tests {
 
     #[tokio::test]
     async fn test_get_quorum() -> Result<()> {
-        let lighthouse = Lighthouse::new(LighthouseOpt {
-            bind: "[::]:0".to_string(),
-            join_timeout_ms: 100,
-            min_replicas: 1,
-            quorum_tick_ms: 100,
-            heartbeat_timeout_ms: 5000,
-        })
+        let lighthouse = Lighthouse::new(
+            "".to_string(),
+            LighthouseOpt {
+                bind: "[::]:0".to_string(),
+                join_timeout_ms: 100,
+                min_replicas: 1,
+                quorum_tick_ms: 100,
+                heartbeat_timeout_ms: 5000,
+            },
+        )
         .await?;
         let lighthouse_fut = tokio::spawn(lighthouse.clone().run());
 
@@ -646,13 +652,16 @@ mod tests {
 
     #[tokio::test]
     async fn test_get_quorum_heal_first_step() -> Result<()> {
-        let lighthouse = Lighthouse::new(LighthouseOpt {
-            bind: "[::]:0".to_string(),
-            join_timeout_ms: 100,
-            min_replicas: 2,
-            quorum_tick_ms: 100,
-            heartbeat_timeout_ms: 5000,
-        })
+        let lighthouse = Lighthouse::new(
+            "".to_string(),
+            LighthouseOpt {
+                bind: "[::]:0".to_string(),
+                join_timeout_ms: 100,
+                min_replicas: 2,
+                quorum_tick_ms: 100,
+                heartbeat_timeout_ms: 5000,
+            },
+        )
         .await?;
         let lighthouse_fut = tokio::spawn(lighthouse.clone().run());
 
@@ -718,13 +727,16 @@ mod tests {
 
     #[tokio::test]
     async fn test_checkpoint_metadata() -> Result<()> {
-        let lighthouse = Lighthouse::new(LighthouseOpt {
-            bind: "[::]:0".to_string(),
-            join_timeout_ms: 100,
-            min_replicas: 1,
-            quorum_tick_ms: 100,
-            heartbeat_timeout_ms: 5000,
-        })
+        let lighthouse = Lighthouse::new(
+            "".to_string(),
+            LighthouseOpt {
+                bind: "[::]:0".to_string(),
+                join_timeout_ms: 100,
+                min_replicas: 1,
+                quorum_tick_ms: 100,
+                heartbeat_timeout_ms: 5000,
+            },
+        )
         .await?;
         let lighthouse_fut = tokio::spawn(lighthouse.clone().run());
 

torchft/_torchft.pyi

@@ -89,6 +89,7 @@ class Quorum:
 class LighthouseClient:
     addr: str
     connect_timeout: timedelta
+    room_id: Optional[str] = None
 
     def quorum(
         self,