437 lines
13 KiB
Rust
437 lines
13 KiB
Rust
use std::net::{Ipv4Addr, SocketAddrV4};
|
|
use std::time::Duration;
|
|
|
|
use pallas_network::facades::{PeerClient, PeerServer};
|
|
use pallas_network::miniprotocols::blockfetch::BlockRequest;
|
|
use pallas_network::miniprotocols::chainsync::{ClientRequest, HeaderContent, Tip};
|
|
use pallas_network::miniprotocols::{
|
|
blockfetch,
|
|
chainsync::{self, NextResponse},
|
|
Point,
|
|
};
|
|
use tokio::net::TcpListener;
|
|
|
|
#[tokio::test]
|
|
#[ignore]
|
|
pub async fn chainsync_history_happy_path() {
|
|
let mut peer = PeerClient::connect("preview-node.world.dev.cardano.org:30002", 2)
|
|
.await
|
|
.unwrap();
|
|
|
|
let client = peer.chainsync();
|
|
|
|
let known_point = Point::Specific(
|
|
1654413,
|
|
hex::decode("7de1f036df5a133ce68a82877d14354d0ba6de7625ab918e75f3e2ecb29771c2").unwrap(),
|
|
);
|
|
|
|
let (point, _) = client
|
|
.find_intersect(vec![known_point.clone()])
|
|
.await
|
|
.unwrap();
|
|
|
|
println!("{point:?}");
|
|
|
|
assert!(matches!(client.state(), chainsync::State::Idle));
|
|
|
|
match point {
|
|
Some(point) => assert_eq!(point, known_point),
|
|
None => panic!("expected point"),
|
|
}
|
|
|
|
let next = client.request_next().await.unwrap();
|
|
|
|
match next {
|
|
NextResponse::RollBackward(point, _) => assert_eq!(point, known_point),
|
|
_ => panic!("expected rollback"),
|
|
}
|
|
|
|
assert!(matches!(client.state(), chainsync::State::Idle));
|
|
|
|
for _ in 0..10 {
|
|
let next = client.request_next().await.unwrap();
|
|
|
|
match next {
|
|
NextResponse::RollForward(_, _) => (),
|
|
_ => panic!("expected roll-forward"),
|
|
}
|
|
|
|
assert!(matches!(client.state(), chainsync::State::Idle));
|
|
}
|
|
|
|
client.send_done().await.unwrap();
|
|
|
|
assert!(matches!(client.state(), chainsync::State::Done));
|
|
}
|
|
|
|
#[tokio::test]
|
|
#[ignore]
|
|
pub async fn chainsync_tip_happy_path() {
|
|
let mut peer = PeerClient::connect("preview-node.world.dev.cardano.org:30002", 2)
|
|
.await
|
|
.unwrap();
|
|
|
|
let client = peer.chainsync();
|
|
|
|
client.intersect_tip().await.unwrap();
|
|
|
|
assert!(matches!(client.state(), chainsync::State::Idle));
|
|
|
|
let next = client.request_next().await.unwrap();
|
|
|
|
assert!(matches!(next, NextResponse::RollBackward(..)));
|
|
|
|
let mut await_count = 0;
|
|
|
|
for _ in 0..4 {
|
|
let next = if client.has_agency() {
|
|
client.request_next().await.unwrap()
|
|
} else {
|
|
await_count += 1;
|
|
client.recv_while_must_reply().await.unwrap()
|
|
};
|
|
|
|
match next {
|
|
NextResponse::RollForward(_, _) => (),
|
|
NextResponse::Await => (),
|
|
_ => panic!("expected roll-forward or await"),
|
|
}
|
|
}
|
|
|
|
assert!(await_count > 0, "tip was never reached");
|
|
|
|
client.send_done().await.unwrap();
|
|
|
|
assert!(matches!(client.state(), chainsync::State::Done));
|
|
}
|
|
|
|
#[tokio::test]
|
|
#[ignore]
|
|
pub async fn blockfetch_happy_path() {
|
|
let mut peer = PeerClient::connect("preview-node.world.dev.cardano.org:30002", 2)
|
|
.await
|
|
.unwrap();
|
|
|
|
let client = peer.blockfetch();
|
|
|
|
let known_point = Point::Specific(
|
|
1654413,
|
|
hex::decode("7de1f036df5a133ce68a82877d14354d0ba6de7625ab918e75f3e2ecb29771c2").unwrap(),
|
|
);
|
|
|
|
let range_ok = client
|
|
.request_range((known_point.clone(), known_point))
|
|
.await;
|
|
|
|
assert!(matches!(client.state(), blockfetch::State::Streaming));
|
|
|
|
println!("streaming...");
|
|
|
|
assert!(range_ok.is_ok());
|
|
|
|
for _ in 0..1 {
|
|
let next = client.recv_while_streaming().await.unwrap();
|
|
|
|
match next {
|
|
Some(body) => assert_eq!(body.len(), 3251),
|
|
_ => panic!("expected block body"),
|
|
}
|
|
|
|
assert!(matches!(client.state(), blockfetch::State::Streaming));
|
|
}
|
|
|
|
let next = client.recv_while_streaming().await.unwrap();
|
|
|
|
assert!(next.is_none());
|
|
|
|
client.send_done().await.unwrap();
|
|
|
|
assert!(matches!(client.state(), blockfetch::State::Done));
|
|
}
|
|
|
|
#[tokio::test]
|
|
#[ignore]
|
|
pub async fn blockfetch_server_and_client_happy_path() {
|
|
let block_bodies = vec![
|
|
hex::decode("deadbeefdeadbeef").unwrap(),
|
|
hex::decode("c0ffeec0ffeec0ffee").unwrap(),
|
|
];
|
|
|
|
let point = Point::Specific(
|
|
1337,
|
|
hex::decode("deadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeef").unwrap(),
|
|
);
|
|
|
|
let server = tokio::spawn({
|
|
let bodies = block_bodies.clone();
|
|
let point = point.clone();
|
|
async move {
|
|
// server setup
|
|
|
|
let server_listener = TcpListener::bind(SocketAddrV4::new(Ipv4Addr::LOCALHOST, 30001))
|
|
.await
|
|
.unwrap();
|
|
|
|
let mut peer_server = PeerServer::accept(&server_listener, 0).await.unwrap();
|
|
|
|
let server_bf = peer_server.blockfetch();
|
|
|
|
// server receives range from client, sends blocks
|
|
|
|
let BlockRequest(range_request) = server_bf.recv_while_idle().await.unwrap().unwrap();
|
|
|
|
assert_eq!(range_request, (point.clone(), point.clone()));
|
|
assert_eq!(*server_bf.state(), blockfetch::State::Busy);
|
|
|
|
server_bf.send_block_range(bodies).await.unwrap();
|
|
|
|
assert_eq!(*server_bf.state(), blockfetch::State::Idle);
|
|
|
|
// server receives range from client, sends NoBlocks
|
|
|
|
let BlockRequest(_) = server_bf.recv_while_idle().await.unwrap().unwrap();
|
|
|
|
server_bf.send_block_range(vec![]).await.unwrap();
|
|
|
|
assert_eq!(*server_bf.state(), blockfetch::State::Idle);
|
|
|
|
assert!(server_bf.recv_while_idle().await.unwrap().is_none());
|
|
|
|
assert_eq!(*server_bf.state(), blockfetch::State::Done);
|
|
}
|
|
});
|
|
|
|
let client = tokio::spawn(async move {
|
|
tokio::time::sleep(Duration::from_secs(1)).await;
|
|
|
|
// client setup
|
|
|
|
let mut client_to_server_conn = PeerClient::connect("localhost:30001", 0).await.unwrap();
|
|
|
|
let client_bf = client_to_server_conn.blockfetch();
|
|
|
|
// client sends request range
|
|
|
|
client_bf
|
|
.send_request_range((point.clone(), point.clone()))
|
|
.await
|
|
.unwrap();
|
|
|
|
assert!(client_bf.recv_while_busy().await.unwrap().is_some());
|
|
|
|
// client receives blocks until idle
|
|
|
|
let mut received_bodies = Vec::new();
|
|
|
|
while let Some(received_body) = client_bf.recv_while_streaming().await.unwrap() {
|
|
received_bodies.push(received_body)
|
|
}
|
|
|
|
assert_eq!(received_bodies, block_bodies);
|
|
|
|
// client sends request range
|
|
|
|
client_bf
|
|
.send_request_range((point.clone(), point.clone()))
|
|
.await
|
|
.unwrap();
|
|
|
|
// recv_while_busy returns None for NoBlocks message
|
|
assert!(client_bf.recv_while_busy().await.unwrap().is_none());
|
|
|
|
// client sends done
|
|
|
|
client_bf.send_done().await.unwrap();
|
|
});
|
|
|
|
_ = tokio::join!(client, server);
|
|
}
|
|
|
|
#[tokio::test]
|
|
#[ignore]
|
|
pub async fn chainsync_server_and_client_happy_path_n2n() {
|
|
let point1 = Point::Specific(1, vec![0x01]);
|
|
let point2 = Point::Specific(2, vec![0x02]);
|
|
|
|
let server = tokio::spawn({
|
|
let point1 = point1.clone();
|
|
let point2 = point2.clone();
|
|
async move {
|
|
// server setup
|
|
|
|
let server_listener = TcpListener::bind(SocketAddrV4::new(Ipv4Addr::LOCALHOST, 30001))
|
|
.await
|
|
.unwrap();
|
|
|
|
let mut peer_server = PeerServer::accept(&server_listener, 0).await.unwrap();
|
|
|
|
let server_cs = peer_server.chainsync();
|
|
|
|
// server receives find intersect from client, sends intersect point
|
|
|
|
assert_eq!(*server_cs.state(), chainsync::State::Idle);
|
|
|
|
let intersect_points = match server_cs.recv_while_idle().await.unwrap().unwrap() {
|
|
ClientRequest::Intersect(points) => points,
|
|
ClientRequest::RequestNext => panic!("unexpected message"),
|
|
};
|
|
|
|
assert_eq!(*server_cs.state(), chainsync::State::Intersect);
|
|
assert_eq!(intersect_points, vec![point2.clone(), point1.clone()]);
|
|
|
|
server_cs
|
|
.send_intersect_found(point2.clone(), Tip(point2.clone(), 1337))
|
|
.await
|
|
.unwrap();
|
|
|
|
assert_eq!(*server_cs.state(), chainsync::State::Idle);
|
|
|
|
// server receives request next from client, sends rollbackwards
|
|
|
|
match server_cs.recv_while_idle().await.unwrap().unwrap() {
|
|
ClientRequest::RequestNext => (),
|
|
ClientRequest::Intersect(_) => panic!("unexpected message"),
|
|
};
|
|
|
|
assert_eq!(*server_cs.state(), chainsync::State::CanAwait);
|
|
|
|
server_cs
|
|
.send_roll_backward(point2.clone(), Tip(point2.clone(), 1337))
|
|
.await
|
|
.unwrap();
|
|
|
|
assert_eq!(*server_cs.state(), chainsync::State::Idle);
|
|
|
|
// server receives request next from client, sends rollforwards
|
|
|
|
match server_cs.recv_while_idle().await.unwrap().unwrap() {
|
|
ClientRequest::RequestNext => (),
|
|
ClientRequest::Intersect(_) => panic!("unexpected message"),
|
|
};
|
|
|
|
assert_eq!(*server_cs.state(), chainsync::State::CanAwait);
|
|
|
|
let header2 = HeaderContent {
|
|
variant: 1,
|
|
byron_prefix: None,
|
|
cbor: hex::decode("c0ffeec0ffeec0ffee").unwrap(),
|
|
};
|
|
|
|
server_cs
|
|
.send_roll_forward(header2, Tip(point2.clone(), 1337))
|
|
.await
|
|
.unwrap();
|
|
|
|
assert_eq!(*server_cs.state(), chainsync::State::Idle);
|
|
|
|
// server receives request next from client, sends await reply
|
|
// then rollforwards
|
|
|
|
match server_cs.recv_while_idle().await.unwrap().unwrap() {
|
|
ClientRequest::RequestNext => (),
|
|
ClientRequest::Intersect(_) => panic!("unexpected message"),
|
|
};
|
|
|
|
assert_eq!(*server_cs.state(), chainsync::State::CanAwait);
|
|
|
|
server_cs.send_await_reply().await.unwrap();
|
|
|
|
assert_eq!(*server_cs.state(), chainsync::State::MustReply);
|
|
|
|
let header1 = HeaderContent {
|
|
variant: 1,
|
|
byron_prefix: None,
|
|
cbor: hex::decode("deadbeefdeadbeef").unwrap(),
|
|
};
|
|
|
|
server_cs
|
|
.send_roll_forward(header1, Tip(point1.clone(), 123))
|
|
.await
|
|
.unwrap();
|
|
|
|
assert_eq!(*server_cs.state(), chainsync::State::Idle);
|
|
|
|
// server receives client done
|
|
|
|
assert!(server_cs.recv_while_idle().await.unwrap().is_none());
|
|
assert_eq!(*server_cs.state(), chainsync::State::Done);
|
|
}
|
|
});
|
|
|
|
let client = tokio::spawn(async move {
|
|
tokio::time::sleep(Duration::from_secs(1)).await;
|
|
|
|
// client setup
|
|
|
|
let mut client_to_server_conn = PeerClient::connect("localhost:30001", 0).await.unwrap();
|
|
|
|
let client_cs = client_to_server_conn.chainsync();
|
|
|
|
// client sends find intersect
|
|
|
|
let intersect_response = client_cs
|
|
.find_intersect(vec![point2.clone(), point1.clone()])
|
|
.await
|
|
.unwrap();
|
|
|
|
assert_eq!(intersect_response.0, Some(point2.clone()));
|
|
assert_eq!(intersect_response.1 .0, point2.clone());
|
|
assert_eq!(intersect_response.1 .1, 1337);
|
|
|
|
// client sends msg request next
|
|
|
|
client_cs.send_request_next().await.unwrap();
|
|
|
|
// client receives rollback
|
|
|
|
match client_cs.recv_while_can_await().await.unwrap() {
|
|
NextResponse::RollBackward(point, tip) => {
|
|
assert_eq!(point, point2.clone());
|
|
assert_eq!(tip.0, point2.clone());
|
|
assert_eq!(tip.1, 1337);
|
|
}
|
|
_ => panic!("unexpected response"),
|
|
}
|
|
|
|
client_cs.send_request_next().await.unwrap();
|
|
|
|
// client receives roll forward
|
|
|
|
match client_cs.recv_while_can_await().await.unwrap() {
|
|
NextResponse::RollForward(content, tip) => {
|
|
assert_eq!(content.cbor, hex::decode("c0ffeec0ffeec0ffee").unwrap());
|
|
assert_eq!(tip.0, point2.clone());
|
|
assert_eq!(tip.1, 1337);
|
|
}
|
|
_ => panic!("unexpected response"),
|
|
}
|
|
|
|
// client sends msg request next
|
|
|
|
client_cs.send_request_next().await.unwrap();
|
|
|
|
// client receives await
|
|
|
|
match client_cs.recv_while_can_await().await.unwrap() {
|
|
NextResponse::Await => (),
|
|
_ => panic!("unexpected response"),
|
|
}
|
|
|
|
match client_cs.recv_while_must_reply().await.unwrap() {
|
|
NextResponse::RollForward(content, tip) => {
|
|
assert_eq!(content.cbor, hex::decode("deadbeefdeadbeef").unwrap());
|
|
assert_eq!(tip.0, point1.clone());
|
|
assert_eq!(tip.1, 123);
|
|
}
|
|
_ => panic!("unexpected response"),
|
|
}
|
|
|
|
// client sends done
|
|
|
|
client_cs.send_done().await.unwrap();
|
|
});
|
|
|
|
_ = tokio::join!(client, server);
|
|
}
|
|
|
|
// TODO: redo txsubmission client test
|