homelab/traefik
1
0
Fork 0
Code Activity

Mitigate TestShutdownUDPConn flakyness

Browse source
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
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
pkg/server/server_entrypoint_udp_test.go
View file

@ -53,12 +53,18 @@ func TestShutdownUDPConn(t *testing.T) {
// Start sending packets, to create a "session" with the server. // Start sending packets, to create a "session" with the server.
requireEcho(t, "TEST", conn, time.Second) requireEcho(t, "TEST", conn, time.Second)
shutdownStartedChan := make(chan struct{})
doneChan := make(chan struct{}) doneChan := make(chan struct{})
go func() { go func() {
close(shutdownStartedChan)
entryPoint.Shutdown(t.Context()) entryPoint.Shutdown(t.Context())
close(doneChan) 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. // Make sure that our session is still live even after the shutdown.
requireEcho(t, "TEST2", conn, time.Second) requireEcho(t, "TEST2", conn, time.Second)