Test ANALYZE in BaseIcebergConnectorSmokeTest

Author: WillPelech

plugin/trino-iceberg/src/test/java/io/trino/plugin/iceberg/BaseIcebergConnectorSmokeTest.java

@@ -56,6 +56,7 @@
 import static com.google.common.base.Verify.verify;
 import static com.google.common.collect.ImmutableList.toImmutableList;
 import static io.airlift.concurrent.MoreFutures.tryGetFutureValue;
+import static io.trino.plugin.iceberg.IcebergSessionProperties.COLLECT_EXTENDED_STATISTICS_ON_WRITE;
 import static io.trino.plugin.iceberg.IcebergTestUtils.FILE_IO_FACTORY;
 import static io.trino.plugin.iceberg.IcebergTestUtils.getFileSystemFactory;
 import static io.trino.plugin.iceberg.IcebergTestUtils.getMetadataFileAndUpdatedMillis;
@@ -893,6 +894,38 @@ public void testIcebergTablesSystemTable()
         }
     }
 
+    @Test
+    public void testAnalyze()
+    {
+        Session noStatsOnWrite = Session.builder(getSession())
+                .setCatalogSessionProperty("iceberg", COLLECT_EXTENDED_STATISTICS_ON_WRITE, "false")
+                .build();
+
+        try (TestTable table = new TestTable(
+                sql -> getQueryRunner().execute(noStatsOnWrite, sql),
+                "test_analyze",
+                "(id int)")) {
+            assertUpdate(noStatsOnWrite, "INSERT INTO " + table.getName() + " VALUES 1, 2, 3", 3);
+
+            try {
+                assertUpdate("ANALYZE " + table.getName());
+            }
+            catch (Exception e) {
+                verifyAnalyzeFailurePermissible(e);
+                return;
+            }
+            assertQuery("SHOW STATS FOR " + table.getName(),
+                    "VALUES " +
+                            "('id', null, 3e0, 0e0, null, '1', '3'), " +
+                            "(null, null, null, null, 3e0, null, null)");
+        }
+    }
+
+    protected String verifyAnalyzeFailurePermissible(Exception e)
+    {
+        throw new AssertionError("Unexpected analyze failure", e);
+    }
+
     protected void dropSchema(String schema)
             throws Exception
     {