Fix incorrect neverFails=true on array_union and array_intersect

Author: jiwen624

core/trino-main/src/main/java/io/trino/operator/scalar/ArrayIntersectFunction.java

@@ -32,7 +32,7 @@
 import static io.trino.spi.function.OperatorType.HASH_CODE;
 import static io.trino.spi.function.OperatorType.IDENTICAL;
 
-@ScalarFunction(value = "array_intersect", neverFails = true)
+@ScalarFunction(value = "array_intersect")
 @Description("Intersects elements of the two given arrays")
 public final class ArrayIntersectFunction
 {

core/trino-main/src/main/java/io/trino/operator/scalar/ArrayUnionFunction.java

@@ -36,7 +36,7 @@
 import static io.trino.spi.function.OperatorType.IDENTICAL;
 import static io.trino.spi.type.BigintType.BIGINT;
 
-@ScalarFunction(value = "array_union", neverFails = true)
+@ScalarFunction(value = "array_union")
 @Description("Union elements of the two given arrays")
 public final class ArrayUnionFunction
 {

core/trino-main/src/test/java/io/trino/type/TestArrayOperators.java

@@ -299,6 +299,24 @@ public void testArraySize()
                 .hasErrorCode(EXCEEDED_FUNCTION_MEMORY_LIMIT);
     }
 
+    @Test
+    public void testArrayUnionSize()
+    {
+        int size = toIntExact(MAX_FUNCTION_MEMORY.toBytes() + 1);
+        assertTrinoExceptionThrownBy(
+                () -> assertions.expression("array_union(ARRAY[lpad('', %1$s , 'x')], ARRAY[lpad('', %1$s , 'y'), lpad('', %1$s , 'z')])".formatted(size)).evaluate())
+                .hasErrorCode(EXCEEDED_FUNCTION_MEMORY_LIMIT);
+    }
+
+    @Test
+    public void testArrayIntersectSize()
+    {
+        int size = toIntExact(MAX_FUNCTION_MEMORY.toBytes() + 1);
+        assertTrinoExceptionThrownBy(
+                () -> assertions.expression("array_intersect(ARRAY[lpad('', %1$s , 'x'), lpad('', %1$s , 'y')], ARRAY[lpad('', %1$s , 'x'), lpad('', %1$s , 'y')])".formatted(size)).evaluate())
+                .hasErrorCode(EXCEEDED_FUNCTION_MEMORY_LIMIT);
+    }
+
     @Test
     public void testArrayToJsonSmoke()
     {