[bfops/test-flakes]: Allow custom smoketest subscribe timeout

Author: bfops

crates/smoketests/src/lib.rs

@@ -1609,21 +1609,30 @@ log = "0.4"
     /// This matches Python's subscribe semantics - start subscription first,
     /// perform actions, then call the handle to collect results.
     pub fn subscribe_background(&self, queries: &[&str], n: usize) -> Result<SubscriptionHandle> {
-        self.subscribe_background_opts(queries, n, None)
+        self.subscribe_background_opts(queries, n, Duration::from_secs(30), None)
+    }
+
+    pub fn subscribe_background_with_timeout(
+        &self,
+        queries: &[&str],
+        n: usize,
+        timeout: Duration,
+    ) -> Result<SubscriptionHandle> {
+        self.subscribe_background_opts(queries, n, timeout, None)
     }
 
     pub fn subscribe_background_on(&self, database: &str, queries: &[&str], n: usize) -> Result<SubscriptionHandle> {
-        self.subscribe_background_on_opts(database, queries, n, Some(false))
+        self.subscribe_background_on_opts(database, queries, n, Duration::from_secs(30), Some(false))
     }
 
     /// Starts a subscription in the background with --confirmed flag.
     pub fn subscribe_background_confirmed(&self, queries: &[&str], n: usize) -> Result<SubscriptionHandle> {
-        self.subscribe_background_opts(queries, n, Some(true))
+        self.subscribe_background_opts(queries, n, Duration::from_secs(30), Some(true))
     }
 
     /// Starts a subscription in the background with --confirmed flag.
     pub fn subscribe_background_unconfirmed(&self, queries: &[&str], n: usize) -> Result<SubscriptionHandle> {
-        self.subscribe_background_opts(queries, n, Some(false))
+        self.subscribe_background_opts(queries, n, Duration::from_secs(30), Some(false))
     }
 
     pub fn subscribe_background_on_confirmed(
@@ -1632,14 +1641,15 @@ log = "0.4"
         queries: &[&str],
         n: usize,
     ) -> Result<SubscriptionHandle> {
-        self.subscribe_background_on_opts(database, queries, n, Some(true))
+        self.subscribe_background_on_opts(database, queries, n, Duration::from_secs(30), Some(true))
     }
 
     /// Internal helper for background subscribe with options.
     fn subscribe_background_opts(
         &self,
         queries: &[&str],
         n: usize,
+        timeout: Duration,
         confirmed: Option<bool>,
     ) -> Result<SubscriptionHandle> {
         let identity = self
@@ -1648,24 +1658,26 @@ log = "0.4"
             .context("No database published")?
             .clone();
 
-        self.subscribe_background_on_impl(&identity, queries, n, confirmed)
+        self.subscribe_background_on_impl(&identity, queries, n, timeout, confirmed)
     }
 
     fn subscribe_background_on_opts(
         &self,
         database: &str,
         queries: &[&str],
         n: usize,
+        timeout: Duration,
         confirmed: Option<bool>,
     ) -> Result<SubscriptionHandle> {
-        self.subscribe_background_on_impl(database, queries, n, confirmed)
+        self.subscribe_background_on_impl(database, queries, n, timeout, confirmed)
     }
 
     fn subscribe_background_on_impl(
         &self,
         database: &str,
         queries: &[&str],
         n: usize,
+        timeout: Duration,
         confirmed: Option<bool>,
     ) -> Result<SubscriptionHandle> {
         let cli_path = ensure_binaries_built();
@@ -1680,7 +1692,7 @@ log = "0.4"
             self.server_url.clone(),
             database.to_string(),
             "-t".to_string(),
-            "30".to_string(),
+            timeout.as_secs().to_string(),
             "-n".to_string(),
             n.to_string(),
             "--print-initial-update".to_string(),