child function is now returning TreeNode

treedatastructure/build.gradle

@@ -15,7 +15,7 @@ android {
         minSdkVersion 15
         targetSdkVersion 29
         versionCode 1
-        versionName "1.2.1"
+        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
     }
 
     /**

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())
     }
-
-
 }