Adjust max_records_in_commit for tests

Author: kim

crates/commitlog/src/commitlog.rs

@@ -920,7 +920,7 @@ fn range_is_empty(range: &impl RangeBounds<u64>) -> bool {
 
 #[cfg(test)]
 mod tests {
-    use std::{cell::Cell, iter::repeat};
+    use std::{cell::Cell, iter::repeat, num::NonZeroU16};
 
     use pretty_assertions::assert_matches;
 
@@ -1231,6 +1231,7 @@ mod tests {
             log.repo.clone(),
             Options {
                 max_segment_size: 1024,
+                max_records_in_commit: NonZeroU16::new(10).unwrap(),
                 ..Options::default()
             },
         )

crates/commitlog/src/segment.rs

@@ -772,7 +772,16 @@ mod tests {
     fn write_read_roundtrip() {
         let repo = repo::Memory::unlimited();
 
-        let mut writer = repo::create_segment_writer(&repo, Options::default(), Commit::DEFAULT_EPOCH, 0).unwrap();
+        let mut writer = repo::create_segment_writer(
+            &repo,
+            Options {
+                max_records_in_commit: NonZeroU16::new(4).unwrap(),
+                ..<_>::default()
+            },
+            Commit::DEFAULT_EPOCH,
+            0,
+        )
+        .unwrap();
         writer.append([0; 32]).unwrap();
         writer.append([1; 32]).unwrap();
         writer.append([2; 32]).unwrap();
@@ -801,7 +810,16 @@ mod tests {
     fn metadata() {
         let repo = repo::Memory::unlimited();
 
-        let mut writer = repo::create_segment_writer(&repo, Options::default(), Commit::DEFAULT_EPOCH, 0).unwrap();
+        let mut writer = repo::create_segment_writer(
+            &repo,
+            Options {
+                max_records_in_commit: NonZeroU16::new(3).unwrap(),
+                ..<_>::default()
+            },
+            Commit::DEFAULT_EPOCH,
+            0,
+        )
+        .unwrap();
         // Commit 0..2
         writer.append([0; 32]).unwrap();
         writer.append([0; 32]).unwrap();
@@ -835,7 +853,17 @@ mod tests {
         let repo = repo::Memory::unlimited();
         let commits = vec![vec![[1; 32], [2; 32]], vec![[3; 32]], vec![[4; 32], [5; 32]]];
 
-        let mut writer = repo::create_segment_writer(&repo, Options::default(), Commit::DEFAULT_EPOCH, 0).unwrap();
+        let mut writer = repo::create_segment_writer(
+            &repo,
+            Options {
+                max_records_in_commit: NonZeroU16::new(3).unwrap(),
+                ..<_>::default()
+            },
+            Commit::DEFAULT_EPOCH,
+            0,
+        )
+        .unwrap();
+
         for commit in &commits {
             for tx in commit {
                 writer.append(*tx).unwrap();
@@ -868,7 +896,16 @@ mod tests {
         let repo = repo::Memory::unlimited();
         let commits = vec![vec![[1; 32], [2; 32]], vec![[3; 32]], vec![[4; 32], [5; 32]]];
 
-        let mut writer = repo::create_segment_writer(&repo, Options::default(), Commit::DEFAULT_EPOCH, 0).unwrap();
+        let mut writer = repo::create_segment_writer(
+            &repo,
+            Options {
+                max_records_in_commit: NonZeroU16::new(3).unwrap(),
+                ..<_>::default()
+            },
+            Commit::DEFAULT_EPOCH,
+            0,
+        )
+        .unwrap();
         for commit in &commits {
             for tx in commit {
                 writer.append(*tx).unwrap();

crates/commitlog/src/tests/helpers.rs

@@ -1,4 +1,4 @@
-use std::fmt::Debug;
+use std::{fmt::Debug, num::NonZeroU16};
 
 use env_logger::Env;
 
@@ -13,6 +13,7 @@ pub fn mem_log<T: Encode>(max_segment_size: u64) -> commitlog::Generic<repo::Mem
         repo::Memory::unlimited(),
         Options {
             max_segment_size,
+            max_records_in_commit: NonZeroU16::new(10).unwrap(),
             ..Options::default()
         },
     )

crates/commitlog/src/tests/partial.rs

@@ -180,6 +180,7 @@ fn open_log<T>(repo: ShortMem) -> commitlog::Generic<ShortMem, T> {
         repo,
         Options {
             max_segment_size: 1024,
+            max_records_in_commit: NonZeroU16::new(10).unwrap(),
             ..Options::default()
         },
     )