fix incorrect reuse of UDP connection in handshake benchmark

quic-go · May 5, 2024 · 3d7e516 · 3d7e516
1 parent b4210f8
commit 3d7e516
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/integrationtests/self/benchmark_test.go b/integrationtests/self/benchmark_test.go
@@ -33,10 +33,13 @@ func BenchmarkHandshake(b *testing.B) {
 	if err != nil {
 		b.Fatal(err)
 	}
+	defer conn.Close()
+	tr := &quic.Transport{Conn: conn}
+	defer tr.Close()
 
 	b.ResetTimer()
 	for i := 0; i < b.N; i++ {
-		c, err := quic.Dial(context.Background(), conn, ln.Addr(), tlsClientConfig, nil)
+		c, err := tr.Dial(context.Background(), ln.Addr(), tlsClientConfig, nil)
 		if err != nil {
 			b.Fatal(err)
 		}