make test fail instead of hang

2026-02-28 22:08:01 +08:00 · 2014-10-14 17:46:11 -07:00 · 2014-10-14 17:46:11 -07:00 · 1a7fac4551
commit 1a7fac4551
parent 00516299dd
1 changed files with 11 additions and 2 deletions
--- a/routing/dht/dht_test.go
+++ b/routing/dht/dht_test.go
@ -431,6 +431,15 @@ func TestConnectCollision(t *testing.T) {
 		done <- struct{}{}
 	}()

-	<-done
-	<-done
+	timeout := time.After(time.Second * 5)
+	select {
+	case <-done:
+	case <-timeout:
+		t.Fatal("Timeout received!")
+	}
+	select {
+	case <-done:
+	case <-timeout:
+		t.Fatal("Timeout received!")
+	}
 }