3

2026-05-18 21:33:39 +03:00
parent 5096155d83
commit 469fa0e860
94 changed files with 8761 additions and 8003 deletions
@@ -221,6 +221,125 @@ func TestPeerConnectionProbeTargetKeepsPeerForLocalRelayReverseQUIC(t *testing.T
 	}
 }

+func TestPeerConnectionProbeTargetsFallsBackToBestPeerCertSHA256(t *testing.T) {
+	intent := PeerConnectionIntent{
+		NodeID:             "node-b",
+		BestPeerCertSHA256: "intent-cert",
+	}
+	cacheEntry := PeerCacheEntry{
+		NodeID:              "node-b",
+		BestPeerCertSHA256:  "cache-cert",
+		BestCandidateID:     "node-b-best",
+		BestTransport:       "direct_quic",
+		Endpoint:            "quic://94.141.118.222:19199",
+		EndpointCandidates: []PeerEndpointCandidate{
+			{
+				EndpointID:       "node-b-public",
+				NodeID:           "node-b",
+				Transport:        "direct_quic",
+				Address:          "quic://94.141.118.222:19199",
+				Reachability:     "public",
+				ConnectivityMode: "direct",
+				Priority:         1,
+			},
+		},
+	}
+
+	targets := peerConnectionProbeTargets(intent, cacheEntry)
+	if len(targets) != 1 {
+		t.Fatalf("target count = %d, want 1", len(targets))
+	}
+	for _, target := range targets {
+		if target.Endpoint != "quic://94.141.118.222:19199" {
+			continue
+		}
+		if target.PeerCertSHA256 != "cache-cert" {
+			t.Fatalf("peer cert = %q, want cache-cert", target.PeerCertSHA256)
+		}
+	}
+}
+
+func TestPeerConnectionProbeTargetsUpgradeRelayReadyPeerToDirectQUIC(t *testing.T) {
+	now := time.Date(2026, 5, 18, 12, 0, 0, 0, time.UTC)
+	current := now
+	tlsConfig := testQUICTLSConfig(t)
+	server, err := StartQUICFabricServer(context.Background(), QUICFabricServerConfig{
+		ListenAddr: "127.0.0.1:0",
+		TLSConfig:  tlsConfig,
+	})
+	if err != nil {
+		t.Fatalf("start quic fabric server: %v", err)
+	}
+	defer server.Close()
+
+	local := PeerIdentity{ClusterID: "cluster-1", NodeID: "node-a"}
+	certSHA256 := testQUICCertSHA256(t, tlsConfig)
+	leases := []PeerRendezvousLease{{
+		LeaseID:          "lease-node-b-via-node-r",
+		PeerNodeID:       "node-b",
+		RelayNodeID:      "node-r",
+		RelayEndpoint:    "quic://127.0.0.1:1",
+		Transport:        "relay_quic",
+		ConnectivityMode: "relay_required",
+		Priority:         10,
+		ControlPlaneOnly: true,
+		IssuedAt:         now.Add(-time.Minute),
+		ExpiresAt:        now.Add(time.Minute),
+	}}
+	cache := NewPeerCache(PeerCacheConfig{
+		Local: local,
+		PeerEndpointCandidates: map[string][]PeerEndpointCandidate{
+			"node-b": {
+				{
+					EndpointID:       "node-b-direct",
+					NodeID:           "node-b",
+					Transport:        "direct_quic",
+					Address:          "quic://" + server.Addr().String(),
+					Reachability:     "public",
+					ConnectivityMode: "direct",
+					Priority:         1,
+					Metadata:         peerConnectionProbeMetadata(t, certSHA256),
+				},
+			},
+		},
+		RendezvousLeases: leases,
+		WarmPeerLimit:    1,
+		Now:              now,
+	})
+	tracker := NewPeerConnectionTracker(cache.Snapshot(), now)
+	manager := NewPeerConnectionManager(PeerConnectionManagerConfig{
+		Local:            local,
+		PeerCache:        cache,
+		Tracker:          tracker,
+		RendezvousLeases: leases,
+		QUICTransport:    NewQUICFabricTransport(nil),
+		ProbeTimeout:     time.Second,
+		Now: func() time.Time {
+			current = current.Add(10 * time.Millisecond)
+			return current
+		},
+	})
+
+	cycle := manager.ProbeOnce(context.Background())
+	if cycle.Attempted != 1 || cycle.Succeeded != 1 || len(cycle.Results) != 1 {
+		t.Fatalf("unexpected cycle: %+v", cycle)
+	}
+	result := cycle.Results[0]
+	if result.SelectedCandidateID != "node-b-direct" || result.SelectedEndpoint != "quic://"+server.Addr().String() {
+		t.Fatalf("relay-ready peer did not upgrade to direct candidate: %+v", result)
+	}
+	if result.ConnectionState.State != PeerConnectionReady {
+		t.Fatalf("connection state = %q, want ready", result.ConnectionState.State)
+	}
+	if len(result.CandidateResults) == 0 || result.CandidateResults[0].Transport != "direct_quic" || result.CandidateResults[0].LinkStatus != PeerConnectionProbeReachable {
+		t.Fatalf("candidate trail missing direct probe success: %+v", result.CandidateResults)
+	}
+	snapshot := tracker.Snapshot()
+	if snapshot.Ready != 1 || snapshot.RelayReady != 0 {
+		t.Fatalf("unexpected tracker snapshot after direct upgrade: %+v", snapshot)
+	}
+}
+
 func TestPeerConnectionManagerFallsBackAcrossEndpointCandidates(t *testing.T) {
 	now := time.Date(2026, 4, 30, 12, 0, 0, 0, time.UTC)
 	current := now