With clause enhance

integration-test/src/main/java/org/apache/iotdb/it/env/cluster/config/MppCommonConfig.java

@@ -657,6 +657,18 @@ public CommonConfig setRestrictObjectLimit(boolean restrictObjectLimit) {
     return this;
   }
 
+  @Override
+  public CommonConfig setCteBufferSize(long cteBufferSize) {
+    setProperty("cte_buffer_size_in_bytes", String.valueOf(cteBufferSize));
+    return this;
+  }
+
+  @Override
+  public CommonConfig setMaxRowsInCteBuffer(int maxRows) {
+    setProperty("max_rows_in_cte_buffer", String.valueOf(maxRows));
+    return this;
+  }
+
   // For part of the log directory
   public String getClusterConfigStr() {
     return fromConsensusFullNameToAbbr(properties.getProperty(CONFIG_NODE_CONSENSUS_PROTOCOL_CLASS))

integration-test/src/main/java/org/apache/iotdb/it/env/cluster/config/MppSharedCommonConfig.java

@@ -691,4 +691,18 @@ public CommonConfig setRestrictObjectLimit(boolean restrictObjectLimit) {
     dnConfig.setRestrictObjectLimit(restrictObjectLimit);
     return this;
   }
+
+  @Override
+  public CommonConfig setCteBufferSize(long cteBufferSize) {
+    dnConfig.setCteBufferSize(cteBufferSize);
+    cnConfig.setCteBufferSize(cteBufferSize);
+    return this;
+  }
+
+  @Override
+  public CommonConfig setMaxRowsInCteBuffer(int maxRows) {
+    dnConfig.setMaxRowsInCteBuffer(maxRows);
+    cnConfig.setMaxRowsInCteBuffer(maxRows);
+    return this;
+  }
 }

integration-test/src/main/java/org/apache/iotdb/it/env/remote/config/RemoteCommonConfig.java

@@ -482,4 +482,14 @@ public CommonConfig setAuditableOperationResult(String auditableOperationResult)
   public CommonConfig setRestrictObjectLimit(boolean restrictObjectLimit) {
     return this;
   }
+
+  @Override
+  public CommonConfig setCteBufferSize(long cteBufferSize) {
+    return this;
+  }
+
+  @Override
+  public CommonConfig setMaxRowsInCteBuffer(int maxRows) {
+    return this;
+  }
 }

integration-test/src/main/java/org/apache/iotdb/itbase/env/CommonConfig.java

@@ -213,4 +213,8 @@ default CommonConfig setDefaultDatabaseLevel(int defaultDatabaseLevel) {
   CommonConfig setAuditableOperationResult(String auditableOperationResult);
 
   CommonConfig setRestrictObjectLimit(boolean restrictObjectLimit);
+
+  CommonConfig setCteBufferSize(long cteBufferSize);
+
+  CommonConfig setMaxRowsInCteBuffer(int maxRows);
 }