70 lines
2 KiB
Rust
70 lines
2 KiB
Rust
|
use datastream::frame::{Lifetime, NodeId, StreamId};
|
||
|
|
use datastream::{Mux, Record};
|
||
|
|
use distribution::telemetry::{
|
||
|
|
CacheEntryRec, DIST_STATE, DistributionState, MembershipTransition, RegistryEntryRec,
|
||
|
|
};
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn distribution_state_record_round_trips_from_owner_crate() {
|
||
|
|
let state = DistributionState {
|
||
|
|
cache_size: 2,
|
||
|
|
cache_entries: vec![CacheEntryRec {
|
||
|
|
actor_addr: "actor-a".into(),
|
||
|
|
node_id: "node-a".into(),
|
||
|
|
}],
|
||
|
|
directory_route_count: 7,
|
||
|
|
registry_size: 3,
|
||
|
|
registry_tombstones: 1,
|
||
|
|
registry_entries: vec![RegistryEntryRec {
|
||
|
|
name: "svc".into(),
|
||
|
|
actor_addr: "actor-b".into(),
|
||
|
|
node_id: "node-b".into(),
|
||
|
|
tombstone: false,
|
||
|
|
}],
|
||
|
|
recent_probe_targets: vec!["node-c".into()],
|
||
|
|
peer_auth_mode: "allow-list".into(),
|
||
|
|
authorized_peer_count: 4,
|
||
|
|
};
|
||
|
|
|
||
|
|
assert_eq!(DistributionState::CHANNEL, DIST_STATE);
|
||
|
|
assert_eq!(DistributionState::decode(&state.encode()).unwrap(), state);
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn distribution_emits_owned_channel_through_datastream_mux() {
|
||
|
|
let stream = StreamId::new(NodeId::new("dist-node"), Lifetime(1));
|
||
|
|
let mux = Mux::unbounded(stream);
|
||
|
|
let state = DistributionState {
|
||
|
|
registry_size: 9,
|
||
|
|
..Default::default()
|
||
|
|
};
|
||
|
|
|
||
|
|
let pos = mux.submit(DistributionState::channel(), state.encode());
|
||
|
|
let frames = mux.drain();
|
||
|
|
|
||
|
|
assert_eq!(pos.0, 0);
|
||
|
|
assert_eq!(frames.len(), 1);
|
||
|
|
assert_eq!(frames[0].channel.as_str(), DIST_STATE);
|
||
|
|
assert_eq!(
|
||
|
|
DistributionState::decode(&frames[0].payload)
|
||
|
|
.unwrap()
|
||
|
|
.registry_size,
|
||
|
|
9
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn membership_transition_record_round_trips_from_owner_crate() {
|
||
|
|
let transition = MembershipTransition {
|
||
|
|
peer: "peer-a".into(),
|
||
|
|
from: "alive".into(),
|
||
|
|
to: "suspect".into(),
|
||
|
|
reason: "probe timeout".into(),
|
||
|
|
};
|
||
|
|
|
||
|
|
assert_eq!(
|
||
|
|
MembershipTransition::decode(&transition.encode()).unwrap(),
|
||
|
|
transition
|
||
|
|
);
|
||
|
|
}
|