Reuse QUIC fabric connections

agents/rap-node-agent/internal/mesh/fabric_quic_transport_test.go

@@ -155,6 +155,48 @@ func TestQUICFabricTransportRejectsPinnedCertificateMismatch(t *testing.T) {
 	}
 }
 
+func TestQUICFabricTransportReusesConnectionForPeerEndpoint(t *testing.T) {
+	server, err := StartQUICFabricServer(context.Background(), QUICFabricServerConfig{
+		ListenAddr: "127.0.0.1:0",
+		TLSConfig:  testQUICTLSConfig(t),
+	})
+	if err != nil {
+		t.Fatalf("start quic fabric server: %v", err)
+	}
+	defer server.Close()
+
+	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
+	defer cancel()
+	transport := NewQUICFabricTransport(nil)
+	defer transport.Close()
+	target := FabricTransportTarget{
+		PeerID:   "node-b",
+		Endpoint: server.Addr().String(),
+		TLSConfig: &tls.Config{
+			InsecureSkipVerify: true,
+			NextProtos:         []string{fabricQUICNextProto},
+		},
+		Timeout:       time.Second,
+		InboundBuffer: 4,
+		ErrorBuffer:   4,
+	}
+	first, err := transport.Connect(ctx, target)
+	if err != nil {
+		t.Fatalf("first connect: %v", err)
+	}
+	defer first.Close()
+	second, err := transport.Connect(ctx, target)
+	if err != nil {
+		t.Fatalf("second connect: %v", err)
+	}
+	defer second.Close()
+
+	snapshot := transport.Snapshot()
+	if snapshot.ActiveCount != 1 || snapshot.Stats.Opens != 1 || snapshot.Stats.Reuses != 1 {
+		t.Fatalf("unexpected quic transport snapshot: %+v", snapshot)
+	}
+}
+
 func TestQUICFabricServerHandlesFabricFrames(t *testing.T) {
 	var events []FabricSessionEventLogEntry
 	server, err := StartQUICFabricServer(context.Background(), QUICFabricServerConfig{