diff --git a/sdks/java/extensions/sql/zetasql/src/test/java/org/apache/beam/sdk/extensions/sql/zetasql/ZetaSqlDialectSpecTest.java b/sdks/java/extensions/sql/zetasql/src/test/java/org/apache/beam/sdk/extensions/sql/zetasql/ZetaSqlDialectSpecTest.java index ec5ea89f97a9..3f711dfef3e2 100644 --- a/sdks/java/extensions/sql/zetasql/src/test/java/org/apache/beam/sdk/extensions/sql/zetasql/ZetaSqlDialectSpecTest.java +++ b/sdks/java/extensions/sql/zetasql/src/test/java/org/apache/beam/sdk/extensions/sql/zetasql/ZetaSqlDialectSpecTest.java @@ -475,6 +475,20 @@ public void testCoalesceBasic() { pipeline.run().waitUntilFinish(Duration.standardMinutes(PIPELINE_EXECUTION_WAITTIME_MINUTES)); } + @Test + public void testCoalesceNotNull() { + String sql = "SELECT COALESCE(@p0, \"yay\") AS ColA"; + ImmutableMap params = + ImmutableMap.of("p0", Value.createSimpleNullValue(TypeKind.TYPE_STRING)); + + PCollection stream = execute(sql, params); + + final Schema schema = Schema.builder().addStringField("field1").build(); + + PAssert.that(stream).containsInAnyOrder(Row.withSchema(schema).addValues("yay").build()); + pipeline.run().waitUntilFinish(Duration.standardMinutes(PIPELINE_EXECUTION_WAITTIME_MINUTES)); + } + @Test public void testCoalesceSingleArgument() { String sql = "SELECT COALESCE(@p0) AS ColA";