Merge pull request #1299 from partiql/eval-struct

Adds literal STRUCT test
partiql · Dec 20, 2023 · 35377df · 35377df
2 parents 5617ef6 + 0a16ab5
commit 35377df
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/partiql-eval/src/test/kotlin/org/partiql/eval/internal/PartiQLEngineDefaultTest.kt b/partiql-eval/src/test/kotlin/org/partiql/eval/internal/PartiQLEngineDefaultTest.kt
@@ -404,4 +404,35 @@ class PartiQLEngineDefaultTest {
         )
         assertEquals(expected, output)
     }
+
+    @OptIn(PartiQLValueExperimental::class)
+    @Test
+    fun testStruct() {
+        val source = """
+            SELECT VALUE {
+                'a': 1,
+                'b': NULL,
+                c : d
+            }
+            FROM <<
+                { 'c': 'hello', 'd': 'world' }
+            >>
+        """.trimIndent()
+        val statement = parser.parse(source).root
+        val session = PartiQLPlanner.Session("q", "u")
+        val plan = planner.plan(statement, session)
+
+        val prepared = engine.prepare(plan.plan)
+        val result = engine.execute(prepared) as PartiQLResult.Value
+        val output = result.value
+
+        val expected: PartiQLValue = bagValue(
+            structValue(
+                "a" to int32Value(1),
+                "b" to nullValue(),
+                "hello" to stringValue("world")
+            )
+        )
+        assertEquals(expected, output)
+    }
 }