From d849da29a347c211172e398b7b16b465477f6eeb Mon Sep 17 00:00:00 2001 From: Matt Brubeck Date: Wed, 27 Mar 2024 12:33:42 -0700 Subject: [PATCH] Add a retry loop to index_page_unix test --- tests/tests.rs | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/tests/tests.rs b/tests/tests.rs index 3af9011..5a33e0b 100644 --- a/tests/tests.rs +++ b/tests/tests.rs @@ -6,6 +6,8 @@ use std::net::{SocketAddr, ToSocketAddrs}; use std::path::PathBuf; use std::process::{Command, Stdio}; use std::sync::atomic::{AtomicU16, Ordering}; +use std::thread::sleep; +use std::time::Duration; use tokio_rustls::rustls; use url::Url; @@ -201,8 +203,13 @@ fn index_page_unix() { "example.com".try_into().unwrap(), ) .unwrap(); - let mut unix = - std::os::unix::net::UnixStream::connect(sock_path).expect("could not connect unix socket"); + + let mut unix = loop { + if let Ok(sock) = std::os::unix::net::UnixStream::connect(&sock_path) { + break sock; + } + sleep(Duration::from_millis(10)); + }; let mut tls = rustls::Stream::new(&mut session, &mut unix); write!(tls, "gemini://example.com\r\n").unwrap();