test: children

Author: KSXGitHub

tests/deduplicate_hardlinks.rs

@@ -6,12 +6,14 @@ pub use _utils::*;
 
 use command_extra::CommandExtra;
 use parallel_disk_usage::{
+    data_tree::Reflection,
     json_data::{JsonData, UnitAndTree},
     size::Bytes,
 };
 use pipe_trait::Pipe;
 use pretty_assertions::assert_eq;
 use std::{
+    iter,
     ops::{Add, Mul},
     process::{Command, Stdio},
 };
@@ -39,18 +41,42 @@ fn deduplicate_multiple_hardlinks_to_a_single_file() {
         .pipe_as_ref(serde_json::from_str::<JsonData>)
         .expect("parse stdout as JsonData");
 
-    let actual_size = match &json.unit_and_tree {
-        UnitAndTree::Blocks(_) => panic!("expecting Bytes, but got {:?}", &json.unit_and_tree),
-        UnitAndTree::Bytes(tree) => tree.size,
+    let UnitAndTree::Bytes(tree) = &json.unit_and_tree else {
+        panic!("expecting Bytes but got {:?}", &json.unit_and_tree);
     };
 
-    let expected_size = workspace
+    let file_size = workspace
         .join("file.txt")
         .pipe_as_ref(read_apparent_size)
-        .add(read_apparent_size(&workspace))
         .pipe(Bytes::new);
 
+    let actual_size = tree.size;
+    let expected_size = workspace
+        .pipe_as_ref(read_apparent_size)
+        .pipe(Bytes::new)
+        .add(file_size);
     assert_eq!(actual_size, expected_size);
+
+    let actual_children = {
+        let mut children = tree.children.clone();
+        children.sort_by(|a, b| a.name.cmp(&b.name));
+        children
+    };
+    let expected_children: Vec<_> = {
+        let links = (0..10).map(|num| format!("link.{num}"));
+        let node = |name| Reflection {
+            name,
+            size: file_size,
+            children: Vec::new(),
+        };
+        "file.txt"
+            .to_string()
+            .pipe(iter::once)
+            .chain(links)
+            .map(node)
+            .collect()
+    };
+    assert_eq!(actual_children, expected_children);
 }
 
 #[test]