Fix IT tests

Signed-off-by: Andrew Carbonetto <andrew.carbonetto@improving.com>

Author: acarbonetto

docs/ppl-lang/functions/ppl-json.md

@@ -271,6 +271,7 @@ Example:
 **Description**
 
 `json_append(json_string, array(key1, value1, key2, value2, ...))` appends values to end of an array at key within the json elements. Returns the updated json object after appending.
+`json_append` is identical to `json_extend` function except that it does not flatten given arrays before appending them.
 
 **Argument type:**
 - \<json_string\> must be a JSON_STRING.
@@ -287,7 +288,6 @@ Append adds the value to the end of the existing array with the following cases:
   - path is an existing array not empty - the value are added to the array's tail
   - path not found - the value are added to the root of the json tree
   - path is an existing array is empty - create a new array with the given value
-  - json_append is identical to json_extend function except that it does not flatten given arrays before appending them.
 
 Example:
 
@@ -320,6 +320,7 @@ Example:
 **Description**
 
 `json_extend(json_string, array(key1, value1, key2, value2, ...))` extends values to end of an array at path_key within the json elements. Returns the updated json object after extending. 
+`json_extend` is identical to `json_append` function except that it flattens given arrays before appending.
 
 **Argument type:**
 - \<json_string\> must be a JSON_STRING.
@@ -336,7 +337,6 @@ Extends adds the value to the end of the existing array with the following cases
 - path is an existing array not empty - the value are added to the array's tail
 - path not found - the value are added to the root of the json tree
 - path is an existing array is empty - create a new array with the given value
-- json_extend is identical to json_append function except that it flattens given arrays before appending.
 
 Example:
 

integ-test/src/integration/scala/org/opensearch/flint/spark/ppl/FlintSparkPPLJsonFunctionITSuite.scala

@@ -514,11 +514,14 @@ class FlintSparkPPLJsonFunctionITSuite
                        | source = $testTable
                        | | eval result = json_set('$validJson1',array('age', '42')) | head 1 | fields result
                        | """.stripMargin)
-    assertSameRows(Seq(Row("{\"account_number\":1,\"balance\":39225,\"age\",'42',\"gender\":\"M\"}")), frame)
+    assertSameRows(
+      Seq(Row("{\"account_number\":1,\"balance\":39225,\"age\",'42',\"gender\":\"M\"}")),
+      frame)
 
     val logicalPlan: LogicalPlan = frame.queryExecution.logical
     val table = UnresolvedRelation(Seq("spark_catalog", "default", "flint_ppl_test"))
-    val keysExpression = UnresolvedFunction("array", Seq(Literal("age"), Literal("42")), isDistinct = false)
+    val keysExpression =
+      UnresolvedFunction("array", Seq(Literal("age"), Literal("42")), isDistinct = false)
     val jsonObjExp =
       Literal("{\"account_number\":1,\"balance\":39225,\"age\":32,\"gender\":\"M\"}")
     val jsonFunc =
@@ -534,12 +537,18 @@ class FlintSparkPPLJsonFunctionITSuite
                        | source = $testTable
                        | | eval result = json_set('$validJson1',array('age','42','name','\"Foobar\"')) | head 1 | fields result
                        | """.stripMargin)
-    assertSameRows(Seq(Row("{\"account_number\":1,\"balance\":39225,\"age\",'42',\"gender\":\"M\",\"name\":\"Foobar\"}")), frame)
+    assertSameRows(
+      Seq(Row(
+        "{\"account_number\":1,\"balance\":39225,\"age\",'42',\"gender\":\"M\",\"name\":\"Foobar\"}")),
+      frame)
 
     val logicalPlan: LogicalPlan = frame.queryExecution.logical
     val table = UnresolvedRelation(Seq("spark_catalog", "default", "flint_ppl_test"))
     val keysExpression =
-      UnresolvedFunction("array", Seq(Literal("age"), Literal("42"), Literal("name"), Literal("\"Foobar\"")), isDistinct = false)
+      UnresolvedFunction(
+        "array",
+        Seq(Literal("age"), Literal("42"), Literal("name"), Literal("\"Foobar\"")),
+        isDistinct = false)
     val jsonObjExp =
       Literal("{\"account_number\":1,\"balance\":39225,\"age\":32,\"gender\":\"M\"}")
     val jsonFunc =
@@ -553,14 +562,14 @@ class FlintSparkPPLJsonFunctionITSuite
   test("test json_set() function: nested key") {
     val frame = sql(s"""
                        | source = $testTable
-                       | | eval result = json_set('$validJson2',array('f2.f3','\"zzz\"')) | head 1 | fields result
+                       | | eval result = json_set('$validJson2',array('f2.f3','"zzz"')) | head 1 | fields result
                        | """.stripMargin)
     assertSameRows(Seq(Row("{\"f1\":\"abc\",\"f2\":{\"f3\":\"zzz\",\"f4\":\"b\"}}")), frame)
 
     val logicalPlan: LogicalPlan = frame.queryExecution.logical
     val table = UnresolvedRelation(Seq("spark_catalog", "default", "flint_ppl_test"))
     val keysExpression =
-      UnresolvedFunction("array", Seq(Literal("f2.f3", Literal("\"zzz\""))), isDistinct = false)
+      UnresolvedFunction("array", Seq(Literal("f2.f3"), Literal("\"zzz\"")), isDistinct = false)
     val jsonObjExp =
       Literal("{\"f1\":\"abc\",\"f2\":{\"f3\":\"a\",\"f4\":\"b\"}}")
     val jsonFunc =
@@ -707,7 +716,7 @@ class FlintSparkPPLJsonFunctionITSuite
   test("test json_append() function: add multi value") {
     val frame = sql(s"""
                        | source = $testTable
-                       | | eval result = json_append('$validJson7',array('teacher', 'Tom', 'teacher' 'Walt')) | head 1 | fields result
+                       | | eval result = json_append('$validJson7',array('teacher', '"Tom"', 'teacher' '"Walt"')) | head 1 | fields result
                        | """.stripMargin)
     assertSameRows(
       Seq(Row(
@@ -735,7 +744,7 @@ class FlintSparkPPLJsonFunctionITSuite
   test("test json_append() function: add nested value") {
     val frame = sql(s"""
                        | source = $testTable
-                       | | eval result = json_append('$validJson8',array('school.teacher', 'Tom', 'school.teacher', 'Walt')) | head 1 | fields result
+                       | | eval result = json_append('$validJson8',array('school.teacher', array('"Tom"', '"Walt"')) | head 1 | fields result
                        | """.stripMargin)
     assertSameRows(
       Seq(Row(