From 9dde7e5ebc0304c7683cb15edd401623360aa325 Mon Sep 17 00:00:00 2001
From: nazeh <ar.nazeh@gmail.com>
Date: Mon, 16 Dec 2024 18:12:03 +0300
Subject: [PATCH] test(pkarr): add unit test for maximum_ttl = 0

---
 pkarr/src/client/dht.rs | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/pkarr/src/client/dht.rs b/pkarr/src/client/dht.rs
index 075f245..c28887a 100644
--- a/pkarr/src/client/dht.rs
+++ b/pkarr/src/client/dht.rs
@@ -772,4 +772,38 @@ mod tests {
 
         assert_eq!(resolved, Some(signed_packet));
     }
+
+    #[tokio::test]
+    async fn ttl_0_test() {
+        let testnet = Testnet::new(10).unwrap();
+
+        let client = Client::builder()
+            .testnet(&testnet)
+            .maximum_ttl(0)
+            .build()
+            .unwrap();
+
+        let keypair = Keypair::random();
+        let signed_packet = SignedPacket::builder().sign(&keypair).unwrap();
+
+        client.publish(&signed_packet).await.unwrap();
+
+        // First Call
+        let resolved = client
+            .resolve(&signed_packet.public_key())
+            .await
+            .unwrap()
+            .unwrap();
+
+        assert_eq!(resolved.encoded_packet(), signed_packet.encoded_packet());
+
+        thread::sleep(Duration::from_secs(1));
+
+        let second = client
+            .resolve(&signed_packet.public_key())
+            .await
+            .unwrap()
+            .unwrap();
+        assert_eq!(second.encoded_packet(), signed_packet.encoded_packet());
+    }
 }