From 19e12cad579396bb50ec4a3328f46439075440bf Mon Sep 17 00:00:00 2001
From: Yavuz Tas <yavuzslmtas@gmail.com>
Date: Sat, 7 Dec 2019 02:08:35 +0100
Subject: [PATCH] change package name and update print method

---
 .../BinaryTreeModel.java                              |  2 +-
 .../BinaryTreePrinter.java                            |  8 +++++---
 .../PrintingBinaryTreeModelUnitTest.java              | 11 +++++++----
 3 files changed, 13 insertions(+), 8 deletions(-)
 rename core-java-modules/core-java/src/main/java/com/baeldung/{binarytree => printbinarytree}/BinaryTreeModel.java (94%)
 rename core-java-modules/core-java/src/main/java/com/baeldung/{binarytree => printbinarytree}/BinaryTreePrinter.java (91%)
 rename core-java-modules/core-java/src/test/java/com/baeldung/{binarytree => printbinarytree}/PrintingBinaryTreeModelUnitTest.java (94%)

diff --git a/core-java-modules/core-java/src/main/java/com/baeldung/binarytree/BinaryTreeModel.java b/core-java-modules/core-java/src/main/java/com/baeldung/printbinarytree/BinaryTreeModel.java
similarity index 94%
rename from core-java-modules/core-java/src/main/java/com/baeldung/binarytree/BinaryTreeModel.java
rename to core-java-modules/core-java/src/main/java/com/baeldung/printbinarytree/BinaryTreeModel.java
index 6c62a60d01..94796bf997 100644
--- a/core-java-modules/core-java/src/main/java/com/baeldung/binarytree/BinaryTreeModel.java
+++ b/core-java-modules/core-java/src/main/java/com/baeldung/printbinarytree/BinaryTreeModel.java
@@ -1,4 +1,4 @@
-package com.baeldung.binarytree;
+package com.baeldung.printbinarytree;
 
 public class BinaryTreeModel {
 
diff --git a/core-java-modules/core-java/src/main/java/com/baeldung/binarytree/BinaryTreePrinter.java b/core-java-modules/core-java/src/main/java/com/baeldung/printbinarytree/BinaryTreePrinter.java
similarity index 91%
rename from core-java-modules/core-java/src/main/java/com/baeldung/binarytree/BinaryTreePrinter.java
rename to core-java-modules/core-java/src/main/java/com/baeldung/printbinarytree/BinaryTreePrinter.java
index 861e42fba1..0ec844d7b6 100644
--- a/core-java-modules/core-java/src/main/java/com/baeldung/binarytree/BinaryTreePrinter.java
+++ b/core-java-modules/core-java/src/main/java/com/baeldung/printbinarytree/BinaryTreePrinter.java
@@ -1,4 +1,6 @@
-package com.baeldung.binarytree;
+package com.baeldung.printbinarytree;
+
+import java.io.PrintStream;
 
 public class BinaryTreePrinter {
 
@@ -54,8 +56,8 @@ public class BinaryTreePrinter {
 
     }
 
-    public void print() {
-        System.out.print(traversePreOrder(tree));
+    public void print(PrintStream os) {
+        os.print(traversePreOrder(tree));
     }
 
 }
\ No newline at end of file
diff --git a/core-java-modules/core-java/src/test/java/com/baeldung/binarytree/PrintingBinaryTreeModelUnitTest.java b/core-java-modules/core-java/src/test/java/com/baeldung/printbinarytree/PrintingBinaryTreeModelUnitTest.java
similarity index 94%
rename from core-java-modules/core-java/src/test/java/com/baeldung/binarytree/PrintingBinaryTreeModelUnitTest.java
rename to core-java-modules/core-java/src/test/java/com/baeldung/printbinarytree/PrintingBinaryTreeModelUnitTest.java
index 8ea0bdd5ba..5c250730f9 100644
--- a/core-java-modules/core-java/src/test/java/com/baeldung/binarytree/PrintingBinaryTreeModelUnitTest.java
+++ b/core-java-modules/core-java/src/test/java/com/baeldung/printbinarytree/PrintingBinaryTreeModelUnitTest.java
@@ -1,4 +1,4 @@
-package com.baeldung.binarytree;
+package com.baeldung.printbinarytree;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 
@@ -10,6 +10,9 @@ import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
 
+import com.baeldung.printbinarytree.BinaryTreeModel;
+import com.baeldung.printbinarytree.BinaryTreePrinter;
+
 public class PrintingBinaryTreeModelUnitTest {
 
     private BinaryTreeModel balanced;
@@ -129,7 +132,7 @@ public class PrintingBinaryTreeModelUnitTest {
         expected.append("   ├──node5").append("\n");
         expected.append("   └──node6");
         
-        new BinaryTreePrinter(balanced).print();
+        new BinaryTreePrinter(balanced).print(System.out);
         
         assertEquals(expected.toString(), output.toString());
     }
@@ -148,7 +151,7 @@ public class PrintingBinaryTreeModelUnitTest {
         expected.append("│                 └──node8").append("\n");
         expected.append("└──node2");
         
-        new BinaryTreePrinter(leftSkewed).print();
+        new BinaryTreePrinter(leftSkewed).print(System.out);
         
         assertEquals(expected.toString(), output.toString());
     }
@@ -167,7 +170,7 @@ public class PrintingBinaryTreeModelUnitTest {
         expected.append("               └──node7").append("\n");
         expected.append("                  └──node8");
         
-        new BinaryTreePrinter(rightSkewed).print();
+        new BinaryTreePrinter(rightSkewed).print(System.out);
         
         assertEquals(expected.toString(), output.toString());
     }