refactor: allow customization

Author: KSXGitHub

tests/_utils.rs

@@ -102,14 +102,14 @@ impl SampleWorkspace {
     /// Set up a temporary directory for tests.
     ///
     /// This directory would have a single file being hard-linked multiple times.
-    pub fn multiple_hardlinks_to_a_single_file() -> Self {
+    pub fn multiple_hardlinks_to_a_single_file(bytes: usize, links: u64) -> Self {
         use std::fs::{hard_link, write as write_file};
         let temp = Temp::new_dir().expect("create working directory for sample workspace");
 
         let file_path = temp.join("file.txt");
-        write_file(&file_path, "a".repeat(100_000)).expect("create file.txt");
+        write_file(&file_path, "a".repeat(bytes)).expect("create file.txt");
 
-        for num in 0..10 {
+        for num in 0..links {
             hard_link(&file_path, temp.join(format!("link.{num}")))
                 .unwrap_or_else(|error| panic!("Failed to create 'link.{num}': {error}"));
         }

tests/deduplicate_hardlinks.rs

@@ -27,7 +27,8 @@ fn stdio(command: Command) -> Command {
 
 #[test]
 fn deduplicate_multiple_hardlinks_to_a_single_file() {
-    let workspace = SampleWorkspace::multiple_hardlinks_to_a_single_file();
+    let links = 10;
+    let workspace = SampleWorkspace::multiple_hardlinks_to_a_single_file(100_000, links);
 
     let json = Command::new(PDU)
         .with_current_dir(&workspace)
@@ -63,7 +64,7 @@ fn deduplicate_multiple_hardlinks_to_a_single_file() {
         children
     };
     let expected_children: Vec<_> = {
-        let links = (0..10).map(|num| format!("link.{num}"));
+        let links = (0..links).map(|num| format!("link.{num}"));
         let node = |name| Reflection {
             name,
             size: file_size,
@@ -81,7 +82,8 @@ fn deduplicate_multiple_hardlinks_to_a_single_file() {
 
 #[test]
 fn do_not_deduplicate_multiple_hardlinks_to_a_single_file() {
-    let workspace = SampleWorkspace::multiple_hardlinks_to_a_single_file();
+    let links = 10;
+    let workspace = SampleWorkspace::multiple_hardlinks_to_a_single_file(100_000, links);
 
     let json = Command::new(PDU)
         .with_current_dir(&workspace)
@@ -102,7 +104,7 @@ fn do_not_deduplicate_multiple_hardlinks_to_a_single_file() {
     let expected_size = workspace
         .join("file.txt")
         .pipe_as_ref(read_apparent_size)
-        .mul(11)
+        .mul(links + 1)
         .add(read_apparent_size(&workspace))
         .pipe(Bytes::new);