child function is now returning TreeNode

README.md

@@ -22,7 +22,7 @@ val poland = TreeNode("Poland")
 val france = TreeNode("France")
 europe.addChild(poland)
 europe.addChild(france)
-println(root)
+println(root.prettyString())
 ```
 
 **Pretty Kotlin**
@@ -55,7 +55,7 @@ TreeNode<String> poland = new TreeNode<>("Poland");
 TreeNode<String> france = new TreeNode<>("France");
 europe.addChild(poland);
 europe.addChild(france);
-System.out.println(root);
+System.out.println(root.prettyString());
 ```
 
 *Output:*

treedatastructure/build.gradle

@@ -15,7 +15,7 @@ android {
         minSdkVersion 15
         targetSdkVersion 29
         versionCode 1
-        versionName "1.2.0"
+        versionName "1.2.3"
 
         testInstrumentationRunner "androidx.test.runner.AndroidJUnitRunner"
         consumerProguardFiles 'consumer-rules.pro'

treedatastructure/src/main/java/com/github/adriankuta/datastructure/tree/ChildDeclarationInterface.kt

@@ -15,7 +15,8 @@ interface ChildDeclarationInterface<T> {
      *     }
      * }
      * ```
+     * @return New created TreeNode.
      */
     @JvmSynthetic
-    fun child(value: T, childDeclaration: ChildDeclaration<T>? = null)
+    fun child(value: T, childDeclaration: ChildDeclaration<T>? = null): TreeNode<T>
 }

treedatastructure/src/main/java/com/github/adriankuta/datastructure/tree/TreeNode.kt

@@ -27,11 +27,12 @@ open class TreeNode<T>(val value: T) : Iterable<TreeNode<T>>, ChildDeclarationIn
     }
 
     @JvmSynthetic
-    override fun child(child: T, childDeclaration: ChildDeclaration<T>?) {
+    override fun child(value: T, childDeclaration: ChildDeclaration<T>?): TreeNode<T> {
         val newChild = TreeNode(value)
         if(childDeclaration != null)
             newChild.childDeclaration()
         _children.add(newChild)
+        return newChild
     }
 
     /**
@@ -91,6 +92,10 @@ open class TreeNode<T>(val value: T) : Iterable<TreeNode<T>>, ChildDeclarationIn
     }
 
     override fun toString(): String {
+        return value.toString()
+    }
+
+    fun prettyString(): String {
         val stringBuilder = StringBuilder()
         print(stringBuilder, "", "")
         return stringBuilder.toString()

treedatastructure/src/test/java/com/github/adriankuta/datastructure/tree/TreeNodeTest.kt

@@ -128,18 +128,15 @@ class TreeNodeTest {
         europe.addChild(poland)
         europe.addChild(france)
 
-        val rootExt = treeNode("World") {
+        val rootExt = tree("World") {
-            treeNode("North America") {
+            child("North America") {
-                treeNode("USA")
+                child("USA")
             }
-            treeNode("Europe") {
+            child("Europe") {
-                treeNode("Poland")
+                child("Poland")
-                treeNode("France")
+                child("France")
             }
         }
-        println(root)
+        assertEquals(root.prettyString(), rootExt.prettyString())
-        assertEquals(root.toString(), rootExt.toString())
     }
-
-
 }