diff --git a/fe/fe-core/src/test/java/org/apache/doris/nereids/mv/OptimizeGetAvailableMvsTest.java b/fe/fe-core/src/test/java/org/apache/doris/nereids/mv/OptimizeGetAvailableMvsTest.java index 0bd07bddc32c32..dc0d9f417dbe7f 100644 --- a/fe/fe-core/src/test/java/org/apache/doris/nereids/mv/OptimizeGetAvailableMvsTest.java +++ b/fe/fe-core/src/test/java/org/apache/doris/nereids/mv/OptimizeGetAvailableMvsTest.java @@ -43,6 +43,7 @@ import com.google.common.collect.Sets; import mockit.Mock; import mockit.MockUp; +import org.junit.jupiter.api.AfterEach; import org.junit.jupiter.api.Assertions; import org.junit.jupiter.api.Test; @@ -58,6 +59,21 @@ */ public class OptimizeGetAvailableMvsTest extends SqlTestBase { + @Override + protected void runBeforeEach() throws Exception { + dropTestMvs(); + } + + @AfterEach + void dropTestMvsAfterEach() throws Exception { + dropTestMvs(); + } + + private void dropTestMvs() throws Exception { + dropMvByNereids("drop materialized view if exists mv1"); + dropMvByNereids("drop materialized view if exists mv2"); + } + @Test void testWhenNotPartitionPrune() throws Exception { connectContext.getSessionVariable().setDisableNereidsRules("PRUNE_EMPTY_PARTITION"); @@ -165,8 +181,6 @@ public boolean canBeCandidate() { Assertions.assertEquals(1, mvCanRewritePartitionsMap.size()); Assertions.assertTrue(mvCanRewritePartitionsMap.keySet().iterator().next().getTableName() .equalsIgnoreCase("mv1")); - - dropMvByNereids("drop materialized view mv1"); } @Test @@ -233,6 +247,13 @@ public List getSelectedPartitionIds() { } }; + new MockUp() { + @Mock + public boolean canBeCandidate() { + return true; + } + }; + connectContext.getSessionVariable().enableMaterializedViewRewrite = true; connectContext.getSessionVariable().enableMaterializedViewNestRewrite = true; createMvByNereids("create materialized view mv2 " @@ -278,7 +299,5 @@ public List getSelectedPartitionIds() { Assertions.assertEquals(1, mvCanRewritePartitionsMap.size()); Assertions.assertTrue(mvCanRewritePartitionsMap.keySet().iterator().next().getTableName() .equalsIgnoreCase("mv2")); - - dropMvByNereids("drop materialized view mv2"); } }