From 13dd2b33acfd73695b40ed0ed6099ff19de9cb7f Mon Sep 17 00:00:00 2001
From: guillaumemichel <guillaume@michel.id>
Date: Fri, 7 Feb 2025 17:15:44 +0100
Subject: [PATCH] fix: flaky TestClientModeConnect

---
 dht_test.go | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/dht_test.go b/dht_test.go
index 99ead547..09bbfbaf 100644
--- a/dht_test.go
+++ b/dht_test.go
@@ -1459,9 +1459,14 @@ func TestClientModeConnect(t *testing.T) {
 	a.ProviderStore().AddProvider(ctx, c.Hash(), peer.AddrInfo{ID: p})
 	time.Sleep(time.Millisecond * 5) // just in case...
 
-	provs, err := b.FindProviders(ctx, c)
-	if err != nil {
-		t.Fatal(err)
+	maxRetries := 3
+	var provs []peer.AddrInfo
+	var err error
+	for i := 0; i < maxRetries && len(provs) == 0; i++ {
+		provs, err = b.FindProviders(ctx, c)
+		if err != nil {
+			t.Fatal(err)
+		}
 	}
 
 	if len(provs) == 0 {