From b74cb17b4d3893611eb853775ff6d2528f8a1adf Mon Sep 17 00:00:00 2001 From: akedin Date: Thu, 30 May 2019 10:22:20 -0700 Subject: [PATCH] [SQL] Add array constructor test --- .../sdk/extensions/sql/BeamComplexTypeTest.java | 13 +++++++++++++ 1 file changed, 13 insertions(+) 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);