remove override of testCountPushDown

Author: loserwang1024

fluss-flink/fluss-flink-2.2/src/test/java/org/apache/fluss/flink/source/Flink22TableSourceBatchITCase.java

@@ -17,62 +17,5 @@
 
 package org.apache.fluss.flink.source;
 
-import org.apache.flink.types.Row;
-import org.apache.flink.util.CloseableIterator;
-import org.junit.jupiter.params.ParameterizedTest;
-import org.junit.jupiter.params.provider.ValueSource;
-
-import java.util.Collections;
-import java.util.List;
-
-import static org.apache.fluss.flink.source.testutils.FlinkRowAssertionsUtils.collectRowsWithTimeout;
-import static org.assertj.core.api.Assertions.assertThat;
-import static org.assertj.core.api.Assertions.assertThatThrownBy;
-
 /** IT case for batch source in Flink 2.2. */
-public class Flink22TableSourceBatchITCase extends FlinkTableSourceBatchITCase {
-
-    @ParameterizedTest
-    @ValueSource(booleans = {true, false})
-    @Override
-    void testCountPushDown(boolean partitionTable) throws Exception {
-        String tableName = partitionTable ? preparePartitionedLogTable() : prepareLogTable();
-        int expectedRows = partitionTable ? 10 : 5;
-        // normal scan
-        String query = String.format("SELECT COUNT(*) FROM %s", tableName);
-        assertThat(tEnv.explainSql(query))
-                .contains(
-                        String.format(
-                                "TableSourceScan(table=[[testcatalog, defaultdb, %s, "
-                                        + "aggregates=[grouping=[], aggFunctions=[Count1AggFunction()]]]], "
-                                        + "fields=[count1$0])",
-                                tableName));
-        CloseableIterator<Row> iterRows = tEnv.executeSql(query).collect();
-        List<String> collected = collectRowsWithTimeout(iterRows, 1);
-        List<String> expected = Collections.singletonList(String.format("+I[%s]", expectedRows));
-        assertThat(collected).isEqualTo(expected);
-
-        // test not push down grouping count.
-        assertThatThrownBy(
-                        () ->
-                                tEnv.explainSql(
-                                                String.format(
-                                                        "SELECT COUNT(*) FROM %s group by id",
-                                                        tableName))
-                                        .wait())
-                .hasMessageContaining(
-                        "Currently, Fluss only support queries on table with datalake enabled or point queries on primary key when it's in batch execution mode.");
-
-        // test not support primary key now
-        String primaryTableName = prepareSourceTable(new String[] {"id"}, null);
-        assertThatThrownBy(
-                        () ->
-                                tEnv.explainSql(
-                                                String.format(
-                                                        "SELECT COUNT(*) FROM %s ",
-                                                        primaryTableName))
-                                        .wait())
-                .hasMessageContaining(
-                        "Currently, Fluss only support queries on table with datalake enabled or point queries on primary key when it's in batch execution mode.");
-    }
-}
+public class Flink22TableSourceBatchITCase extends FlinkTableSourceBatchITCase {}

fluss-flink/fluss-flink-common/src/test/java/org/apache/fluss/flink/source/FlinkTableSourceBatchITCase.java

@@ -281,11 +281,7 @@ void testCountPushDown(boolean partitionTable) throws Exception {
         String query = String.format("SELECT COUNT(*) FROM %s", tableName);
         assertThat(tEnv.explainSql(query))
                 .contains(
-                        String.format(
-                                "TableSourceScan(table=[[testcatalog, defaultdb, %s, project=[id], "
-                                        + "aggregates=[grouping=[], aggFunctions=[Count1AggFunction()]]]], "
-                                        + "fields=[count1$0])",
-                                tableName));
+                        "aggregates=[grouping=[], aggFunctions=[Count1AggFunction()]]]], fields=[count1$0]");
         CloseableIterator<Row> iterRows = tEnv.executeSql(query).collect();
         List<String> collected = collectRowsWithTimeout(iterRows, 1);
         List<String> expected = Collections.singletonList(String.format("+I[%s]", expectedRows));