fix test and dispatcher for pings

This commit is contained in:
KingKDot
2026-04-07 10:43:27 -05:00
parent 0ad98e3f24
commit acc010be81
2 changed files with 10 additions and 2 deletions
@@ -45,8 +45,14 @@ func TestHandlePing(t *testing.T) {
t.Fatalf("HandlePing failed: %v", err)
}
if len(writer.msgs) < 1 {
t.Fatal("Expected at least 1 message")
deadline := time.After(2 * time.Second)
for len(writer.msgs) < 1 {
select {
case <-deadline:
t.Fatal("Timed out waiting for pong message")
default:
time.Sleep(5 * time.Millisecond)
}
}
var pong wire.Pong