Fix sporadic connect failure in Linux tests.

sharjeel · Oct 28, 2020 · bbe37c0 · bbe37c0
1 parent ccadcf9
commit bbe37c0
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/src/device/integration_tests/mod.rs b/src/device/integration_tests/mod.rs
@@ -179,9 +179,24 @@ mod tests {
             self.container_name = Some(peer_config_file);
         }
 
-        fn get_request(&self) -> String {
+        fn connect(&self) -> std::net::TcpStream {
             let http_addr = SocketAddr::new(self.allowed_ips[0].ip, 80);
-            let mut tcp_conn = std::net::TcpStream::connect(http_addr).unwrap();
+            for _i in 0..5 {
+                let res = std::net::TcpStream::connect(http_addr);
+                if let Err(err) = res {
+                    println!("failed to connect: {:?}", err);
+                    std::thread::sleep(std::time::Duration::from_millis(100));
+                    continue;
+                }
+
+                return res.unwrap();
+            }
+
+            panic!("failed to connect");
+        }
+
+        fn get_request(&self) -> String {
+            let mut tcp_conn = self.connect();
 
             write!(
                 tcp_conn,