use crate::engine::testing::UTConfig;
use crate::engine::Engine;
use crate::engine::LogIdList;
use crate::CommittedLeaderId;
use crate::LogId;
fn log_id(term: u64, index: u64) -> LogId<u64> {
LogId::<u64> {
leader_id: CommittedLeaderId::new(term, 0),
index,
}
}
fn eng() -> Engine<UTConfig> {
let mut eng = Engine::default();
eng.state.enable_validation(false);
eng.state.log_ids = LogIdList::new(vec![
log_id(0, 0),
log_id(1, 1),
log_id(3, 3),
log_id(5, 5),
]);
eng
}
#[test]
fn test_calc_purge_upto() -> anyhow::Result<()> {
let cases = vec![
(None, None, 0, None),
(None, None, 1, None),
(None, Some(log_id(1, 1)), 0, Some(log_id(1, 1))),
(None, Some(log_id(1, 1)), 1, Some(log_id(0, 0))),
(None, Some(log_id(1, 1)), 2, None),
(Some(log_id(0, 0)), Some(log_id(1, 1)), 0, Some(log_id(1, 1))),
(Some(log_id(0, 0)), Some(log_id(1, 1)), 1, None),
(Some(log_id(0, 0)), Some(log_id(1, 1)), 2, None),
(None, Some(log_id(3, 4)), 0, Some(log_id(3, 4))),
(None, Some(log_id(3, 4)), 1, Some(log_id(3, 3))),
(None, Some(log_id(3, 4)), 2, Some(log_id(1, 2))),
(None, Some(log_id(3, 4)), 3, Some(log_id(1, 1))),
(None, Some(log_id(3, 4)), 4, Some(log_id(0, 0))),
(None, Some(log_id(3, 4)), 5, None),
(Some(log_id(1, 2)), Some(log_id(3, 4)), 0, Some(log_id(3, 4))),
(Some(log_id(1, 2)), Some(log_id(3, 4)), 1, Some(log_id(3, 3))),
(Some(log_id(1, 2)), Some(log_id(3, 4)), 2, None),
(Some(log_id(1, 2)), Some(log_id(3, 4)), 3, None),
(Some(log_id(1, 2)), Some(log_id(3, 4)), 4, None),
(Some(log_id(1, 2)), Some(log_id(3, 4)), 5, None),
];
for (last_purged, snapshot_last_log_id, max_keep, want) in cases {
let mut eng = eng();
eng.config.max_in_snapshot_log_to_keep = max_keep;
eng.config.purge_batch_size = 1;
if let Some(last_purged) = last_purged {
eng.state.log_ids.purge(&last_purged);
eng.state.purged_next = last_purged.index + 1;
}
eng.state.snapshot_meta.last_log_id = snapshot_last_log_id;
let got = eng.log_handler().calc_purge_upto();
assert_eq!(
want, got,
"case: last_purged: {:?}, snapshot_last_log_id: {:?}, max_keep: {}",
last_purged, snapshot_last_log_id, max_keep
);
}
Ok(())
}