fix race condition in multiplex integration test (#4136)

quic-go · Oct 28, 2023 · 6eb0cac · 6eb0cac
1 parent 5311f81
commit 6eb0cac
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/integrationtests/self/multiplex_test.go b/integrationtests/self/multiplex_test.go
@@ -251,6 +251,9 @@ var _ = Describe("Multiplexing", func() {
 				b := make([]byte, packetLen)
 				rand.Read(b[1:]) // keep the first byte set to 0, so it's not classified as a QUIC packet
 				_, err := tr1.WriteTo(b, tr2.Conn.LocalAddr())
+				if ctx.Err() != nil { // ctx canceled while Read was executing
+					return
+				}
 				Expect(err).ToNot(HaveOccurred())
 				sentPackets.Add(1)
 			}