78 lines
2.2 KiB
Rust
78 lines
2.2 KiB
Rust
use std::{
|
|
collections::{HashMap, HashSet},
|
|
sync::Arc,
|
|
};
|
|
|
|
use rand::seq::SliceRandom;
|
|
use ruma_events::EventType;
|
|
use ruma_state_res::{
|
|
self as state_res, is_power_event, room_version::RoomVersion, test_utils::INITIAL_EVENTS,
|
|
StateMap,
|
|
};
|
|
|
|
fn test_event_sort() {
|
|
let _ = tracing::subscriber::set_default(tracing_subscriber::fmt().with_test_writer().finish());
|
|
let events = INITIAL_EVENTS();
|
|
|
|
let event_map = events
|
|
.values()
|
|
.map(|ev| ((ev.event_type(), ev.state_key()), ev.clone()))
|
|
.collect::<StateMap<_>>();
|
|
|
|
let auth_chain = HashSet::new();
|
|
|
|
let power_events = event_map
|
|
.values()
|
|
.filter(|pdu| is_power_event(pdu))
|
|
.map(|pdu| pdu.event_id().clone())
|
|
.collect::<Vec<_>>();
|
|
|
|
let sorted_power_events =
|
|
state_res::reverse_topological_power_sort(power_events, &auth_chain, |id| {
|
|
events.get(id).map(Arc::clone)
|
|
})
|
|
.unwrap();
|
|
|
|
let resolved_power = state_res::iterative_auth_check(
|
|
&RoomVersion::version_6(),
|
|
&sorted_power_events,
|
|
&HashMap::new(), // unconflicted events
|
|
|id| events.get(id).map(Arc::clone),
|
|
)
|
|
.expect("iterative auth check failed on resolved events");
|
|
|
|
// don't remove any events so we know it sorts them all correctly
|
|
let mut events_to_sort = events.keys().cloned().collect::<Vec<_>>();
|
|
|
|
events_to_sort.shuffle(&mut rand::thread_rng());
|
|
|
|
let power_level = resolved_power.get(&(EventType::RoomPowerLevels, "".to_owned()));
|
|
|
|
let sorted_event_ids =
|
|
state_res::mainline_sort(&events_to_sort, power_level, |id| events.get(id).map(Arc::clone))
|
|
.unwrap();
|
|
|
|
assert_eq!(
|
|
vec![
|
|
"$CREATE:foo",
|
|
"$IMA:foo",
|
|
"$IPOWER:foo",
|
|
"$IJR:foo",
|
|
"$IMB:foo",
|
|
"$IMC:foo",
|
|
"$START:foo",
|
|
"$END:foo"
|
|
],
|
|
sorted_event_ids.iter().map(|id| id.to_string()).collect::<Vec<_>>()
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn test_sort() {
|
|
for _ in 0..20 {
|
|
// since we shuffle the eventIds before we sort them introducing randomness
|
|
// seems like we should test this a few times
|
|
test_event_sort()
|
|
}
|
|
}
|