[UT] Fix some unstable FE UTs (backport #64762) (#64765)

Signed-off-by: satanson <ranpanf@gmail.com>
Co-authored-by: satanson <ranpanf@gmail.com>

Author: mergify[bot]

fe/fe-core/src/test/java/com/starrocks/planner/PushDownSubfieldHashJoinTest.java

@@ -33,11 +33,12 @@ public class PushDownSubfieldHashJoinTest {
 
     @BeforeAll
     public static void setUp() throws Exception {
+        FeConstants.runningUnitTest = true;
         UtFrameUtils.createMinStarRocksCluster();
         ctx = UtFrameUtils.createDefaultCtx();
         ctx.getSessionVariable().setEnablePipelineEngine(true);
         ctx.getSessionVariable().setCboPushDownAggregateMode(-1);
-        FeConstants.runningUnitTest = true;
+        ctx.getSessionVariable().setOptimizerExecuteTimeout(10000);
         starRocksAssert = new StarRocksAssert(ctx);
         starRocksAssert.withDatabase(StatsConstants.STATISTICS_DB_NAME)
                 .useDatabase(StatsConstants.STATISTICS_DB_NAME)

fe/fe-core/src/test/java/com/starrocks/planner/TablePruningTest.java

@@ -38,11 +38,12 @@
 public class TablePruningTest extends TablePruningTestBase {
     @BeforeAll
     public static void setUp() throws Exception {
+        FeConstants.runningUnitTest = true;
         UtFrameUtils.createMinStarRocksCluster();
         ctx = UtFrameUtils.createDefaultCtx();
         ctx.getSessionVariable().setEnablePipelineEngine(true);
         ctx.getSessionVariable().setCboPushDownAggregateMode(-1);
-        FeConstants.runningUnitTest = true;
+        ctx.getSessionVariable().setOptimizerExecuteTimeout(10000);
         starRocksAssert = new StarRocksAssert(ctx);
         starRocksAssert.withDatabase(StatsConstants.STATISTICS_DB_NAME)
                 .useDatabase(StatsConstants.STATISTICS_DB_NAME)
@@ -443,41 +444,48 @@ public void testPermuteFromClausesRandomlyAndMoveOnJoinPredicatesToWhereClauses(
                 "E",
                 "FI",
                 "FII");
-        List<String> predicates = Lists.newArrayList(
-                "A0.a_pk = A1.a_pk",
-                "A0.a_pk = A2.a_pk",
-                "A2.a_pk = D0.d_pk",
-                "D0.d_pk = D1.d_pk",
-                "D0.d_pk = D2.d_pk",
-                "A0.a_b_fk = B.b_pk",
-                "B.b_ci_fk = CI.ci_pk",
-                "B.b_cii_fk0 = CII.cii_pk0",
-                "B.b_cii_fk1 = CII.cii_pk1",
-                "D0.d_e_fk = E.e_pk",
-                "E.e_fi_fk = FI.fi_pk",
-                "E.e_fii_fk0 = FII.fii_pk0",
-                "E.e_fii_fk1 = FII.fii_pk1",
-                "D2.d_c3>100"
-        );
+
         Object[][] testCases = new Object[][] {
                 {"avg(A0.a_c0+A1.a_c1), sum(A2.a_c2), max(B.b_pk), min(D0.d_c0), min(D1.d_c1)", "", 1, 1},
                 {"A0.a_c0+A1.a_c1, A2.a_c2, B.b_pk, D0.d_c0, D1.d_c1", "limit 1", 1, 1},
                 {"sum(FII.fii_c0), sum(FII.fii_c0+1)", "limit 1", 1, 1},
                 {"sum(FII.fii_c0), sum(CII.cii_c0)", "limit 1", 1, 1, 1},
         };
+        String[] whereClauses = new String[] {
+                "A0.a_pk = A2.a_pk AND A0.a_pk = A1.a_pk AND E.e_fii_fk1 = FII.fii_pk1 AND D0.d_e_fk = E.e_pk " +
+                        "AND D0.d_pk = D2.d_pk AND A2.a_pk = D0.d_pk AND D0.d_pk = D1.d_pk AND A0.a_b_fk = B.b_pk " +
+                        "AND B.b_cii_fk1 = CII.cii_pk1 AND B.b_ci_fk = CI.ci_pk AND B.b_cii_fk0 = CII.cii_pk0 " +
+                        "AND D2.d_c3>100 AND E.e_fi_fk = FI.fi_pk AND E.e_fii_fk0 = FII.fii_pk0",
+                "A0.a_pk = A1.a_pk AND D0.d_pk = D1.d_pk AND A0.a_pk = A2.a_pk AND D0.d_pk = D2.d_pk " +
+                        "AND D0.d_e_fk = E.e_pk AND A2.a_pk = D0.d_pk AND D2.d_c3>100 AND A0.a_b_fk = B.b_pk " +
+                        "AND B.b_ci_fk = CI.ci_pk AND B.b_cii_fk1 = CII.cii_pk1 AND B.b_cii_fk0 = CII.cii_pk0 " +
+                        "AND E.e_fii_fk0 = FII.fii_pk0 AND E.e_fi_fk = FI.fi_pk AND E.e_fii_fk1 = FII.fii_pk1",
+                "A0.a_pk = A1.a_pk AND E.e_fii_fk0 = FII.fii_pk0 AND A0.a_b_fk = B.b_pk AND A0.a_pk = A2.a_pk " +
+                        "AND B.b_cii_fk0 = CII.cii_pk0 AND B.b_cii_fk1 = CII.cii_pk1 AND D2.d_c3>100 " +
+                        "AND B.b_ci_fk = CI.ci_pk AND D0.d_pk = D1.d_pk AND A2.a_pk = D0.d_pk " +
+                        "AND E.e_fi_fk = FI.fi_pk AND D0.d_pk = D2.d_pk AND D0.d_e_fk = E.e_pk " +
+                        "AND E.e_fii_fk1 = FII.fii_pk1",
+                "A0.a_pk = A1.a_pk AND A0.a_pk = A2.a_pk AND D2.d_c3>100 AND A2.a_pk = D0.d_pk " +
+                        "AND D0.d_pk = D1.d_pk AND D0.d_e_fk = E.e_pk AND E.e_fi_fk = FI.fi_pk " +
+                        "AND D0.d_pk = D2.d_pk AND E.e_fii_fk0 = FII.fii_pk0 AND A0.a_b_fk = B.b_pk " +
+                        "AND B.b_ci_fk = CI.ci_pk AND B.b_cii_fk1 = CII.cii_pk1 AND B.b_cii_fk0 = CII.cii_pk0 " +
+                        "AND E.e_fii_fk1 = FII.fii_pk1",
+                "A0.a_pk = A1.a_pk AND A0.a_pk = A2.a_pk AND D0.d_pk = D1.d_pk AND D2.d_c3>100 " +
+                        "AND A2.a_pk = D0.d_pk AND A0.a_b_fk = B.b_pk AND D0.d_pk = D2.d_pk " +
+                        "AND B.b_ci_fk = CI.ci_pk AND B.b_cii_fk0 = CII.cii_pk0 AND D0.d_e_fk = E.e_pk " +
+                        "AND B.b_cii_fk1 = CII.cii_pk1 AND E.e_fi_fk = FI.fi_pk AND E.e_fii_fk0 = FII.fii_pk0 " +
+                        "AND E.e_fii_fk1 = FII.fii_pk1",
+
+        };
         String sqlFmt = "select %s from %s where %s %s";
         for (Object[] tc : testCases) {
             String selectItems = (String) tc[0];
             String extraLimit = (String) tc[1];
             int rboNumHashJoins = (Integer) tc[2];
             int finalNmHashJoins = (Integer) tc[3];
 
-            for (int i = 0; i < 10; ++i) {
-                String fromClause =
-                        tables.stream().sorted((a, b) -> new Random().nextInt(3) - 1)
-                                .collect(Collectors.joining(",\n"));
-                String whereClause = predicates.stream().sorted((a, b) -> new Random().nextInt(3) - 1)
-                        .collect(Collectors.joining("\nAND "));
+            for (String whereClause : whereClauses) {
+                String fromClause = tables.stream().collect(Collectors.joining(",\n"));
                 String sql = String.format(sqlFmt, selectItems, fromClause, whereClause, extraLimit);
                 checkHashJoinCountWithBothRBOAndCBOLessThan(sql, 12);
             }

fe/fe-core/src/test/java/com/starrocks/planner/TimeDriftConstraintTest.java

@@ -35,6 +35,7 @@ private static StarRocksAssert createStarRocksAssert() {
         UtFrameUtils.createMinStarRocksCluster();
         ConnectContext ctx = UtFrameUtils.createDefaultCtx();
         ctx.getSessionVariable().setEnablePipelineEngine(true);
+        ctx.getSessionVariable().setOptimizerExecuteTimeout(10000);
         FeConstants.runningUnitTest = true;
         StarRocksAssert starRocksAssert = new StarRocksAssert(ctx);
         String hits = "" +