fix: flaky Kademlia test (#7002)

Author: hanabi1224

scripts/tests/calibnet_kademlia_check.sh

@@ -18,7 +18,7 @@ echo "Stateless node peer id: $STATELESS_NODE_PEER_ID"
 CONFIG_PATH="./forest_config.toml"
 cat <<- EOF > $CONFIG_PATH
 	[network]
-	listening_multiaddrs = ["/ip4/127.0.0.1/tcp/0"]
+	listening_multiaddrs = ["/ip4/127.0.0.1/tcp/0", "/ip4/127.0.0.1/udp/0/quic-v1"]
 	bootstrap_peers = ["$STATELESS_NODE_ADDRESS"]
 	mdns = false
 	kademlia = true

scripts/tests/calibnet_stateless_mode_check.sh

@@ -11,14 +11,14 @@ forest_init_stateless
 STATELESS_NODE_ADDRESS=$($FOREST_CLI_PATH net listen | tail -n 1)
 echo "Stateless node address: $STATELESS_NODE_ADDRESS"
 # Example format: 12D3KooWAB9z7vZ1x1v9t4BViVkX1Hy1ScoRnWV2GgGy5ec6pfUZ
-STATELESS_NODE_PEER_ID=$(echo "$STATELESS_NODE_ADDRESS" | cut --delimiter="/" --fields=7 --zero-terminated)
+STATELESS_NODE_PEER_ID=$(echo "$STATELESS_NODE_ADDRESS" | rev | cut --delimiter="/" --fields=1 --zero-terminated | rev)
 echo "Stateless node peer id: $STATELESS_NODE_PEER_ID"
 
 # Run a normal forest node that only connects to the stateless node
 CONFIG_PATH="./forest_config.toml"
 cat <<- EOF > $CONFIG_PATH
 	[network]
-	listening_multiaddrs = ["/ip4/127.0.0.1/tcp/0"]
+	listening_multiaddrs = ["/ip4/127.0.0.1/tcp/0", "/ip4/127.0.0.1/udp/0/quic-v1"]
 	bootstrap_peers = ["$STATELESS_NODE_ADDRESS"]
 	mdns = false
 	kademlia = false

scripts/tests/harness.sh

@@ -83,7 +83,7 @@ function forest_run_node_stateless_detached {
 		data_dir = "/tmp/stateless_forest_data"
 
 		[network]
-		listening_multiaddrs = ["/ip4/127.0.0.1/tcp/0"]
+		listening_multiaddrs = ["/ip4/127.0.0.1/tcp/0", "/ip4/127.0.0.1/udp/0/quic-v1"]
 	EOF
 
   $FOREST_PATH --chain calibnet --encrypt-keystore false --config "$CONFIG_PATH" --log-dir "$LOG_DIRECTORY" --save-token ./stateless_admin_token --stateless &

src/libp2p/peer_manager.rs

@@ -162,6 +162,7 @@ impl PeerManager {
         let peer_stats = peers.full_peers.entry(*peer).or_default();
         peer_stats.successes += 1;
         log_time(peer_stats, dur);
+        metrics::FULL_PEERS.set(peers.full_peers.len() as i64);
     }
 
     /// Logs a failure for the given peer, and updates the average request
@@ -177,6 +178,7 @@ impl PeerManager {
         let peer_stats = peers.full_peers.entry(*peer).or_default();
         peer_stats.failures += 1;
         log_time(peer_stats, dur);
+        metrics::FULL_PEERS.set(peers.full_peers.len() as i64);
     }
 
     /// Removes a peer from the set and returns true if the value was present
@@ -316,7 +318,7 @@ impl PeerManager {
 
 fn remove_peer(peers: &mut PeerSets, peer_id: &PeerId) {
     if peers.full_peers.remove(peer_id).is_some() {
-        metrics::FULL_PEERS.set(peers.full_peers.len() as _);
+        metrics::FULL_PEERS.set(peers.full_peers.len() as i64);
     }
     trace!(
         "removing peer {peer_id}, remaining chain exchange peers: {}",