agate

commit d849da29a347c211172e398b7b16b465477f6eeb
parent f5dbb97e2973847e46e297af805a113f3ac7607c
Author: Matt Brubeck <mbrubeck@limpet.net>
Date:   Wed, 27 Mar 2024 12:33:42 -0700

Add a retry loop to index_page_unix test

Diffstat:
M
tests/tests.rs
 | 
11
+++++++++--

1 file changed, 9 insertions(+), 2 deletions(-)
diff --git 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();