use std::sync::Arc;
use std::time::Duration;
use maplit::btreeset;
#[allow(unused_imports)]
use pretty_assertions::assert_eq;
#[allow(unused_imports)]
use pretty_assertions::assert_ne;
#[allow(unused_imports)]
use pretty_assertions::assert_str_eq;
use crate::Membership;
use crate::MembershipState;
use crate::Vote;
use crate::engine::Engine;
use crate::engine::testing::UTConfig;
use crate::engine::testing::log_id;
use crate::type_config::TypeConfigExt;
use crate::type_config::alias::EffectiveMembershipOf;
use crate::utime::Leased;
fn m01() -> Membership<u64, ()> {
Membership::<u64, ()>::new_with_defaults(vec![btreeset! {0,1}], [])
}
fn m23() -> Membership<u64, ()> {
Membership::<u64, ()>::new_with_defaults(vec![btreeset! {2,3}], btreeset! {1,2,3})
}
fn eng() -> Engine<UTConfig> {
let mut eng = Engine::testing_default(0);
eng.state.enable_validation(false);
eng.config.id = 1;
eng.state.apply_progress_mut().accept(log_id(0, 1, 0));
eng.state.vote = Leased::new(
UTConfig::<()>::now(),
Duration::from_millis(500),
Vote::new_committed(3, 1),
);
eng.state.log_ids.append(log_id(1, 1, 1));
eng.state.log_ids.append(log_id(2, 1, 3));
eng.state.membership_state = MembershipState::new(
Arc::new(EffectiveMembershipOf::<UTConfig>::new(Some(log_id(1, 1, 1)), m01())),
Arc::new(EffectiveMembershipOf::<UTConfig>::new(Some(log_id(2, 1, 3)), m23())),
);
eng.testing_new_leader();
eng.state.server_state = eng.calc_server_state();
eng
}
#[test]
fn test_get_read_log_id() -> anyhow::Result<()> {
let mut eng = eng();
eng.state.apply_progress_mut().accept(log_id(0, 1, 0));
eng.leader.as_mut().unwrap().noop_log_id = log_id(1, 1, 2);
let got = eng.try_leader_handler()?.get_read_log_id();
assert_eq!(log_id(1, 1, 2), got);
eng.state.apply_progress_mut().accept(log_id(2, 1, 3));
let got = eng.try_leader_handler()?.get_read_log_id();
assert_eq!(log_id(2, 1, 3), got);
Ok(())
}