Mitigate TestShutdownUDPConn flakyness

This commit is contained in:
Romain 2025-10-23 11:28:04 +02:00 committed by GitHub
parent 822f349fa1
commit 0880cc672f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -53,12 +53,18 @@ func TestShutdownUDPConn(t *testing.T) {
// Start sending packets, to create a "session" with the server.
requireEcho(t, "TEST", conn, time.Second)
shutdownStartedChan := make(chan struct{})
doneChan := make(chan struct{})
go func() {
close(shutdownStartedChan)
entryPoint.Shutdown(t.Context())
close(doneChan)
}()
// Wait until shutdown has started, and hopefully after 100 ms the listener has stopped accepting new sessions.
<-shutdownStartedChan
time.Sleep(100 * time.Millisecond)
// Make sure that our session is still live even after the shutdown.
requireEcho(t, "TEST2", conn, time.Second)