From 0fb26b4e8dcece944e8defee6dbd6286b299e6a9 Mon Sep 17 00:00:00 2001
From: FranciscaOrtegaG <franciscaortega713@gmail.com>
Date: Sat, 11 Nov 2023 11:40:58 -0300
Subject: [PATCH] Added test max number of retry

---
 src/resolver/async_resolver.rs | 37 ++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/src/resolver/async_resolver.rs b/src/resolver/async_resolver.rs
index 25d8aee2..16e229ab 100644
--- a/src/resolver/async_resolver.rs
+++ b/src/resolver/async_resolver.rs
@@ -351,6 +351,43 @@ mod async_resolver_test {
 
 
     //TODO: test max number of retry
+    #[tokio::test]
+    async fn max_number_of_retry() {
+        let mut config = ResolverConfig::default();
+        let max_retries = 6;
+        config.set_retry(max_retries);
+        let mut resolver = AsyncResolver::new(config);
+
+        // Realiza una resolución de DNS que sabes que fallará
+        //let result = resolver.lookup_ip("nonexistent-domain.com", "UDP").await;
+
+        let mut retries_attempted = 0;
+
+        // Realiza la resolución de DNS que sabes que fallará
+        while retries_attempted < max_retries {
+            let result = resolver.lookup_ip("nonexistent-domain.com", "UDP").await;
+             retries_attempted += 1;
+
+            if result.is_ok() {
+                break; // La resolución tuvo éxito, sal del bucle
+            }
+        }
+        if retries_attempted == max_retries {
+            assert!(retries_attempted == max_retries, "Número incorrecto de reintentos");
+        } else {
+            panic!("La resolución DNS tuvo éxito antes de lo esperado");
+        }
+        // // Verifica que se hayan intentado los reintentos especificados
+        // match result {
+        //     Ok(_) => {
+        //         panic!("Se esperaba un error de timeout, pero se resolvió exitosamente");
+        //     }
+        //     Err(_) => {
+        //         panic!("El timeout no se manejó correctamente");
+        //     }
+        // }
+        
+    }
 
     //TODO: use UDP