Sulkta-OSS/pallas
3
0
Fork 0
Code Issues 1 Pull requests 4 Projects Releases Packages Wiki Activity Actions

feat(network): implement stake distribution local state query (#340)

Browse source
This commit is contained in:
Alexsander Falcucci 2023-11-24 10:43:25 +01:00 • committed by GitHub
parent cce85a7482
commit b1ac003716
5 changed files with 125 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

81
pallas-network/tests/protocols.rs
View file

@ -2,7 +2,7 @@ use std::fs;
use std::net::{Ipv4Addr, SocketAddrV4};
use std::time::Duration;
use pallas_codec::utils::AnyCbor;
use pallas_codec::utils::{AnyCbor, KeyValuePairs};
use pallas_network::facades::{NodeClient, PeerClient, PeerServer};
use pallas_network::miniprotocols::blockfetch::BlockRequest;
use pallas_network::miniprotocols::chainsync::{ClientRequest, HeaderContent, Tip};
@ -454,7 +454,6 @@ pub async fn chainsync_server_and_client_happy_path_n2n() {
}
#[tokio::test]
#[ignore]
pub async fn local_state_query_server_and_client_happy_path() {
let server = tokio::spawn({
async move {
@ -503,6 +502,48 @@ pub async fn local_state_query_server_and_client_happy_path() {
assert_eq!(*server.statequery().state(), localstate::State::Acquired);
// server receives query from client
let query: localstate::queries_v16::Request =
match server.statequery().recv_while_acquired().await.unwrap() {
ClientQueryRequest::Query(q) => q.into_decode().unwrap(),
x => panic!("unexpected message from client: {x:?}"),
};
assert_eq!(
query,
localstate::queries_v16::Request::LedgerQuery(
localstate::queries_v16::LedgerQuery::BlockQuery(
5,
localstate::queries_v16::BlockQuery::GetStakeDistribution,
),
)
);
assert_eq!(*server.statequery().state(), localstate::State::Querying);
let fraction = localstate::queries_v16::Fraction { num: 10, dem: 20 };
let pool = localstate::queries_v16::Pool {
stakes: fraction.clone(),
hashes: b"pool1qv4qgv62s3ha74p0643nexee9zvcdydcyahqqnavhj90zheuykz"
.to_vec()
.into(),
};
let pools = vec![(
b"pool1qvfw4r3auysa5mhpr90n7mmdhs55js8gdywh0y2e3sy6568j2wp"
.to_vec()
.into(),
pool,
)];
let pools = KeyValuePairs::from(pools);
let result = AnyCbor::from_encode(localstate::queries_v16::StakeDistribution { pools });
server.statequery().send_result(result).await.unwrap();
assert_eq!(*server.statequery().state(), localstate::State::Acquired);
// server receives re-acquire from the client
let maybe_point = match server.statequery().recv_while_acquired().await.unwrap() {
@ -573,6 +614,42 @@ pub async fn local_state_query_server_and_client_happy_path() {
}
);
let request = AnyCbor::from_encode(localstate::queries_v16::Request::LedgerQuery(
localstate::queries_v16::LedgerQuery::BlockQuery(
5,
localstate::queries_v16::BlockQuery::GetStakeDistribution,
),
));
client.statequery().send_query(request).await.unwrap();
let result: localstate::queries_v16::StakeDistribution = client
.statequery()
.recv_while_querying()
.await
.unwrap()
.into_decode()
.unwrap();
let fraction = localstate::queries_v16::Fraction { num: 10, dem: 20 };
let pool = localstate::queries_v16::Pool {
stakes: fraction.clone(),
hashes: b"pool1qv4qgv62s3ha74p0643nexee9zvcdydcyahqqnavhj90zheuykz"
.to_vec()
.into(),
};
let pools = vec![(
b"pool1qvfw4r3auysa5mhpr90n7mmdhs55js8gdywh0y2e3sy6568j2wp"
.to_vec()
.into(),
pool,
)];
let pools = KeyValuePairs::from(pools);
assert_eq!(result, localstate::queries_v16::StakeDistribution { pools });
// client sends a ReAquire
client