From bc4c3d0c6b0d4bd33d86a11589bbd2e2294ad1b6 Mon Sep 17 00:00:00 2001 From: Pascal Seitz Date: Tue, 17 May 2022 17:42:33 +0800 Subject: [PATCH] add peek_lru test --- src/store/reader.rs | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/src/store/reader.rs b/src/store/reader.rs index b6a50629b..9da46ec38 100644 --- a/src/store/reader.rs +++ b/src/store/reader.rs @@ -66,6 +66,15 @@ impl BlockCache { fn len(&self) -> usize { self.cache.lock().unwrap().len() } + + #[cfg(test)] + fn peek_lru(&self) -> Option { + self.cache + .lock() + .unwrap() + .peek_lru() + .map(|(&k, _)| k as usize) + } } impl StoreReader { @@ -373,6 +382,8 @@ mod tests { assert_eq!(store.cache_stats().cache_hits, 0); assert_eq!(store.cache_stats().cache_misses, 1); + assert_eq!(store.cache.peek_lru(), Some(0)); + let doc = store.get(499)?; assert_eq!(get_text_field(&doc, &title), Some("Doc 499")); @@ -380,6 +391,8 @@ mod tests { assert_eq!(store.cache_stats().cache_hits, 0); assert_eq!(store.cache_stats().cache_misses, 2); + assert_eq!(store.cache.peek_lru(), Some(0)); + let doc = store.get(0)?; assert_eq!(get_text_field(&doc, &title), Some("Doc 0")); @@ -387,6 +400,8 @@ mod tests { assert_eq!(store.cache_stats().cache_hits, 1); assert_eq!(store.cache_stats().cache_misses, 2); + assert_eq!(store.cache.peek_lru(), Some(9210)); + Ok(()) } }