From 5a729d8acbe362c577c168cc7c30b0e2c608f55a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Orhun=20Parmaks=C4=B1z?= <orhunparmaksiz@gmail.com>
Date: Sat, 18 Dec 2021 14:28:57 +0300
Subject: [PATCH] test(tree): add output test for tree format

---
 systeroid-core/src/tree.rs | 23 +++++++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

diff --git a/systeroid-core/src/tree.rs b/systeroid-core/src/tree.rs
index e5d86fe..f3e79eb 100644
--- a/systeroid-core/src/tree.rs
+++ b/systeroid-core/src/tree.rs
@@ -221,9 +221,28 @@ mod tests {
     }
 
     #[test]
-    fn test_print_line() {
-        let value = String::from("abc\ndef");
+    fn test_tree_output() {
         env::set_var("NO_COLOR", "1");
+        let lines = ["a", "a/b/e", "a/b", "a/b/c/d"];
+
+        let tree = Tree::from_input(&mut lines.iter(), '/');
+        let mut output = Vec::new();
+        tree.print(&mut output, Color::White).unwrap();
+
+        let expected_output = "\
+a
+└── b
+    ├── e
+    └── c
+        └── d\n";
+
+        assert_eq!(expected_output, String::from_utf8_lossy(&output));
+    }
+
+    #[test]
+    fn test_print_line() {
+        env::set_var("NO_COLOR", "1");
+        let value = String::from("abc\ndef");
 
         let mut output = Vec::new();
         TreeNode {