diff --git a/sdks/java/extensions/sql/src/test/java/org/apache/beam/sdk/extensions/sql/BeamComplexTypeTest.java b/sdks/java/extensions/sql/src/test/java/org/apache/beam/sdk/extensions/sql/BeamComplexTypeTest.java index fbdf625f81050..1e6eaf3e1d3c4 100644 --- a/sdks/java/extensions/sql/src/test/java/org/apache/beam/sdk/extensions/sql/BeamComplexTypeTest.java +++ b/sdks/java/extensions/sql/src/test/java/org/apache/beam/sdk/extensions/sql/BeamComplexTypeTest.java @@ -164,6 +164,19 @@ public void testBasicRow() { pipeline.run().waitUntilFinish(Duration.standardMinutes(2)); } + @Test + public void testArrayConstructor() { + BeamSqlEnv sqlEnv = BeamSqlEnv.inMemory(readOnlyTableProvider); + PCollection stream = + BeamSqlRelUtils.toPCollection(pipeline, sqlEnv.parseQuery("SELECT ARRAY[1, 2, 3] f_arr")); + PAssert.that(stream) + .containsInAnyOrder( + Row.withSchema(Schema.builder().addArrayField("f_arr", FieldType.INT32).build()) + .addValue(Arrays.asList(1, 2, 3)) + .build()); + pipeline.run().waitUntilFinish(Duration.standardMinutes(2)); + } + @Test public void testRowWithArray() { BeamSqlEnv sqlEnv = BeamSqlEnv.inMemory(readOnlyTableProvider);