fix(daemon): attempt to fix idle timeout test

charmbracelet · Dec 6, 2024 · 50ed2b8 · 50ed2b8
1 parent 06bc8cf
commit 50ed2b8
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/pkg/daemon/daemon_test.go b/pkg/daemon/daemon_test.go
@@ -75,11 +75,21 @@ func TestMain(m *testing.M) {
 }
 
 func TestIdleTimeout(t *testing.T) {
-	c, err := net.Dial("tcp", testDaemon.addr)
-	if err != nil {
-		t.Fatal(err)
+	var err error
+	var c net.Conn
+	var tries int
+	for {
+		c, err = net.Dial("tcp", testDaemon.addr)
+		if err != nil && tries >= 3 {
+			t.Fatal(err)
+		}
+		tries++
+		if testDaemon.conns.Size() != 0 {
+			break
+		}
+		time.Sleep(10 * time.Millisecond)
 	}
-	time.Sleep(time.Second)
+	time.Sleep(2 * time.Second)
 	_, err = readPktline(c)
 	if err == nil {
 		t.Errorf("expected error, got nil")