v2: convenience for JNode literal creation

Signed-off-by: Alexander Bezzubov <[email protected]>

Author: bzz

src/main/scala/org/bblfsh/client/v2/JNode.scala

@@ -1,5 +1,7 @@
 package org.bblfsh.client.v2
 
+import java.io.Serializable
+
 import scala.collection.mutable
 
 /** UAST nodes representation on the JVM side.
@@ -46,15 +48,31 @@ case class JFloat(num: Double) extends JNode
 case class JUint(num: Long) extends JNode
 case class JInt(num: Long) extends JNode
 case class JBool(value: Boolean) extends JNode
+
 case class JObject(obj: mutable.Buffer[JField]) extends JNode {
   def this() = this(mutable.Buffer[JField]())
   def add(k: String, v: JNode) = {
     obj += ((k, v))
   }
 }
+case object JObject {
+  def apply[T <: (Product with Serializable with JNode)](ns: (String, T)*) = {
+    val jo = new JObject()
+    jo.obj ++= ns
+    jo
+  }
+}
+
 case class JArray(arr: mutable.Buffer[JNode]) extends JNode {
   def this(size: Int) = this(new mutable.ArrayBuffer[JNode](size))
-  def add(v: JNode) = {
-    arr += v
+  def add(n: JNode) = {
+    arr += n
   }
 }
+case object JArray {
+  def apply[T <:  (Product with Serializable with JNode)](ns: T *)   = {
+    val ja = new JArray(ns.length)
+    ja.arr ++= ns
+    ja
+  }
+}

src/test/scala/org/bblfsh/client/v2/JNodeTest.scala

@@ -2,20 +2,18 @@ package org.bblfsh.client.v2
 
 import org.scalatest.{BeforeAndAfter, FlatSpec, Matchers}
 
-import scala.collection.mutable.Buffer
-
 
 class JNodeTest extends FlatSpec
   with BeforeAndAfter
   with Matchers {
 
-  val rootTree = JArray(Buffer(
-    JObject(Buffer(
+  val rootTree = JArray(
+    JObject(
       "k1" -> JString("v1"),
       "k2" -> JBool(false)
-    )),
+    ),
     JString("test")
-  ))
+  )
 
   "JNode" should "expose children" in {
     rootTree.children.size shouldEqual 2