fix unit test

Author: shizy818

iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/plan/relational/analyzer/PushAggregationThroughUnionTest.java

@@ -19,6 +19,7 @@
 
 package org.apache.iotdb.db.queryengine.plan.relational.analyzer;
 
+import org.apache.iotdb.db.conf.IoTDBDescriptor;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.DistributedQueryPlan;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.LogicalQueryPlan;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.node.PlanNode;
@@ -31,6 +32,7 @@
 
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableMap;
+import org.junit.Before;
 import org.junit.Test;
 
 import java.util.Optional;
@@ -53,6 +55,10 @@
 import static org.junit.Assert.assertEquals;
 
 public class PushAggregationThroughUnionTest {
+  @Before
+  public void setUp() {
+    IoTDBDescriptor.getInstance().getConfig().setDataNodeId(1);
+  }
 
   @Test
   public void UnionAggregationTest() {

iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/plan/relational/planner/CteMaterializerTest.java

@@ -93,20 +93,16 @@ private Type convertType(String columnName) {
 
   private void mockException() {
     CteMaterializer cteMaterializer = Mockito.spy(new CteMaterializer());
-    Mockito.when(cteMaterializer.fetchCteQueryResult(Mockito.any(), Mockito.any(), Mockito.any()))
-        .thenReturn(null);
+    Mockito.doReturn(null)
+        .when(cteMaterializer)
+        .fetchCteQueryResult(Mockito.any(), Mockito.any(), Mockito.any());
     CteMaterializer.setInstance(cteMaterializer);
   }
 
   private void mockSubquery() {
     CteMaterializer cteMaterializer = Mockito.spy(new CteMaterializer());
 
-    Mockito.when(
-            cteMaterializer.fetchCteQueryResult(
-                Mockito.any(Table.class),
-                Mockito.any(Query.class),
-                Mockito.any(MPPQueryContext.class)))
-        .thenAnswer(
+    Mockito.doAnswer(
             (InvocationOnMock invocation) -> {
               Table table = invocation.getArgument(0);
               Query query = invocation.getArgument(1);
@@ -131,7 +127,10 @@ private void mockSubquery() {
 
               TableSchema tableSchema = new TableSchema(table.getName().toString(), columnsSchemas);
               return new CteDataStore(query, tableSchema, ImmutableList.of(0, 1));
-            });
+            })
+        .when(cteMaterializer)
+        .fetchCteQueryResult(
+            Mockito.any(Table.class), Mockito.any(Query.class), Mockito.any(MPPQueryContext.class));
     CteMaterializer.setInstance(cteMaterializer);
   }