Run testIncorrectIcebergFileSizes for ORC too

trinodb · Nov 24, 2021 · 302d942 · 302d942
1 parent ea633f9
commit 302d942
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/plugin/trino-iceberg/src/test/java/io/trino/plugin/iceberg/BaseIcebergConnectorTest.java b/plugin/trino-iceberg/src/test/java/io/trino/plugin/iceberg/BaseIcebergConnectorTest.java
@@ -2371,7 +2371,7 @@ public void testIncorrectIcebergFileSizes()
             throws Exception
     {
         // Create a table with a single insert
-        assertUpdate("CREATE TABLE test_iceberg_file_size (x BIGINT) WITH (format='PARQUET')");
+        assertUpdate("CREATE TABLE test_iceberg_file_size (x BIGINT)");
         assertUpdate("INSERT INTO test_iceberg_file_size VALUES (123), (456), (758)", 3);
 
         // Get manifest file
@@ -2417,7 +2417,10 @@ public void testIncorrectIcebergFileSizes()
         assertQuery(session, "SELECT * FROM test_iceberg_file_size", "VALUES (123), (456), (758)");
 
         // Using Iceberg provided file size fails the query
-        assertQueryFails("SELECT * FROM test_iceberg_file_size", format("Error reading tail from .* with length %d", alteredValue));
+        assertQueryFails("SELECT * FROM test_iceberg_file_size",
+                format == ORC
+                        ? format(".*Error opening Iceberg split.*\\QIncorrect file size (%s) for file (end of stream not reached)\\E.*", alteredValue)
+                        : format("Error reading tail from .* with length %d", alteredValue));
 
         dropTable("test_iceberg_file_size");
     }