From 0a6eb4162a7ace71f0afd5f7c245a583c374c3e9 Mon Sep 17 00:00:00 2001
From: nk_ysg <nk_ysg@163.com>
Date: Sat, 24 Aug 2024 05:22:37 +0800
Subject: [PATCH] add add_and_connect add tests

---
 crates/net/network/src/manager.rs |  2 +-
 crates/net/network/src/peers.rs   | 10 +++++-----
 crates/net/network/src/state.rs   |  4 ++--
 3 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/crates/net/network/src/manager.rs b/crates/net/network/src/manager.rs
index 7ff183759a4db..bd6c0d41c0fce 100644
--- a/crates/net/network/src/manager.rs
+++ b/crates/net/network/src/manager.rs
@@ -589,7 +589,7 @@ impl NetworkManager {
                 self.swarm.sessions_mut().disconnect(peer_id, reason);
             }
             NetworkHandleMessage::ConnectPeer(peer_id, kind, addr) => {
-                self.swarm.state_mut().connect_peer_kind(peer_id, kind, addr);
+                self.swarm.state_mut().add_and_connect(peer_id, kind, addr);
             }
             NetworkHandleMessage::SetNetworkState(net_state) => {
                 // Sets network connection state between Active and Hibernate.
diff --git a/crates/net/network/src/peers.rs b/crates/net/network/src/peers.rs
index bf9be90fe505f..eb5cb3929f438 100644
--- a/crates/net/network/src/peers.rs
+++ b/crates/net/network/src/peers.rs
@@ -732,17 +732,17 @@ impl PeersManager {
     /// Connect to the given peer. NOTE: if the maximum number out outbound sessions is reached,
     /// this won't do anything. See `reth_network::SessionManager::dial_outbound`.
     #[allow(dead_code)]
-    pub(crate) fn connect_peer(
+    pub(crate) fn add_and_connect(
         &mut self,
         peer_id: PeerId,
         addr: PeerAddr,
         fork_id: Option<ForkId>,
     ) {
-        self.connect_peer_kind(peer_id, PeerKind::Basic, addr, fork_id)
+        self.add_and_connect_kind(peer_id, PeerKind::Basic, addr, fork_id)
     }
 
     ///  Connects a peer and its address with the given kind.
-    pub(crate) fn connect_peer_kind(
+    pub(crate) fn add_and_connect_kind(
         &mut self,
         peer_id: PeerId,
         kind: PeerKind,
@@ -2428,7 +2428,7 @@ mod tests {
         let peer = PeerId::random();
         let socket_addr = SocketAddr::new(IpAddr::V4(Ipv4Addr::new(127, 0, 1, 2)), 8008);
         let mut peers = PeersManager::default();
-        peers.connect_peer(peer, PeerAddr::from_tcp(socket_addr), None);
+        peers.add_and_connect(peer, PeerAddr::from_tcp(socket_addr), None);
 
         match event!(peers) {
             PeerAction::Connect { peer_id, remote_addr } => {
@@ -2443,7 +2443,7 @@ mod tests {
         assert_eq!(record.udp_addr(), socket_addr);
 
         // connect again
-        peers.connect_peer(peer, PeerAddr::from_tcp(socket_addr), None);
+        peers.add_and_connect(peer, PeerAddr::from_tcp(socket_addr), None);
 
         let (record, _) = peers.peer_by_id(peer).unwrap();
         assert_eq!(record.tcp_addr(), socket_addr);
diff --git a/crates/net/network/src/state.rs b/crates/net/network/src/state.rs
index c85e466ee47ae..f55375fe78985 100644
--- a/crates/net/network/src/state.rs
+++ b/crates/net/network/src/state.rs
@@ -297,8 +297,8 @@ impl NetworkState {
     }
 
     /// Connects a peer and its address with the given kind
-    pub(crate) fn connect_peer_kind(&mut self, peer_id: PeerId, kind: PeerKind, addr: PeerAddr) {
-        self.peers_manager.connect_peer_kind(peer_id, kind, addr, None)
+    pub(crate) fn add_and_connect(&mut self, peer_id: PeerId, kind: PeerKind, addr: PeerAddr) {
+        self.peers_manager.add_and_connect_kind(peer_id, kind, addr, None)
     }
 
     /// Removes a peer and its address with the given kind from the peerset.