Retry db connection

tests/utils/setup.rs

@@ -89,14 +89,28 @@ async fn setup_database() -> (ContainerAsync<Postgres>, DbConn) {
         container.get_host_port_ipv4(5432).await.unwrap(),
     );
 
-    sleep(Duration::from_secs(1)).await;
-
     tracing::debug!("Connecting to Postgres at {:?}", connection_string);
 
-    let db = Database::connect(connection_string)
-        .await
-        .inspect_err(|err| tracing::error!("postgres connect err: {err:?}"))
-        .unwrap();
+    let mut i = 0;
+    let db = loop {
+        let res = Database::connect(connection_string)
+            .await
+            .inspect_err(|err| tracing::error!("postgres connect err: {err:?}"));
+
+        match res {
+            Ok(db) => break db,
+            Err(_) => {
+                sleep(Duration::from_secs(1)).await;
+
+                if i > 3 {
+                    tracing::error!("Can't connect to database");
+                    panic!();
+                }
+
+                i += 1;
+            }
+        }
+    };
 
     tracing::debug!("Running migrations");