use grin_core as core;
use grin_p2p as p2p;
use grin_store as store;
use grin_util as util;
use grin_util::{Mutex, StopState};
use std::net::{SocketAddr, TcpListener, TcpStream};
use std::sync::Arc;
use std::{thread, time};
use crate::core::core::hash::Hash;
use crate::core::pow::Difficulty;
use crate::p2p::Peer;
fn open_port() -> u16 {
let listener = TcpListener::bind("127.0.0.1:0").unwrap();
listener.local_addr().unwrap().port()
}
#[test]
fn peer_handshake() {
util::init_test_logger();
let p2p_config = p2p::P2PConfig {
host: "0.0.0.0".parse().unwrap(),
port: open_port(),
peers_allow: None,
peers_deny: None,
..p2p::P2PConfig::default()
};
let net_adapter = Arc::new(p2p::DummyAdapter {});
let db_env = Arc::new(store::new_env(".grin".to_string()));
let server = Arc::new(
p2p::Server::new(
db_env,
p2p::Capabilities::UNKNOWN,
p2p_config.clone(),
net_adapter.clone(),
Hash::from_vec(&vec![]),
Arc::new(Mutex::new(StopState::new())),
)
.unwrap(),
);
let p2p_inner = server.clone();
let _ = thread::spawn(move || p2p_inner.listen());
thread::sleep(time::Duration::from_secs(1));
let addr = SocketAddr::new(p2p_config.host, p2p_config.port);
let mut socket = TcpStream::connect_timeout(&addr, time::Duration::from_secs(10)).unwrap();
let my_addr = "127.0.0.1:5000".parse().unwrap();
let mut peer = Peer::connect(
&mut socket,
p2p::Capabilities::UNKNOWN,
Difficulty::min(),
my_addr,
&p2p::handshake::Handshake::new(Hash::from_vec(&vec![]), p2p_config.clone()),
net_adapter,
)
.unwrap();
assert!(peer.info.user_agent.ends_with(env!("CARGO_PKG_VERSION")));
peer.start(socket);
thread::sleep(time::Duration::from_secs(1));
peer.send_ping(Difficulty::min(), 0).unwrap();
thread::sleep(time::Duration::from_secs(1));
let server_peer = server.peers.get_connected_peer(&my_addr).unwrap();
assert_eq!(server_peer.info.total_difficulty(), Difficulty::min());
assert!(server.peers.peer_count() > 0);
}