This is an automated email from the ASF dual-hosted git repository.
justinchen pushed a commit to branch restart-check-pw
in repository https://gitbox.apache.org/repos/asf/iotdb.git
The following commit(s) were added to refs/heads/restart-check-pw by this push:
new 61df416ee82 partial
61df416ee82 is described below
commit 61df416ee82e409ddd4ae0f0dfa894f4c2767ef7
Author: Caideyipi <[email protected]>
AuthorDate: Tue Feb 10 09:52:24 2026 +0800
partial
---
.../auto/AbstractPipeDualTreeModelAutoIT.java | 60 ++++++++++++++++------
.../treemodel/auto/basic/IoTDBPipeAlterIT.java | 7 ---
.../treemodel/auto/basic/IoTDBPipeAutoSplitIT.java | 22 ++++----
.../treemodel/auto/basic/IoTDBPipeDataSinkIT.java | 7 ---
.../treemodel/auto/basic/IoTDBPipeProcessorIT.java | 24 +++++----
.../treemodel/auto/basic/IoTDBPipeProtocolIT.java | 25 +++++++--
.../auto/enhanced/IoTDBPipeAutoConflictIT.java | 24 +++++----
.../auto/enhanced/IoTDBPipeAutoDropIT.java | 7 ---
.../auto/enhanced/IoTDBPipeClusterIT.java | 28 +++++-----
.../enhanced/IoTDBPipeConditionalOperationsIT.java | 7 ---
.../auto/enhanced/IoTDBPipeIdempotentIT.java | 25 ++++-----
.../auto/enhanced/IoTDBPipeSinkCompressionIT.java | 25 ++++-----
.../auto/enhanced/IoTDBPipeWithLoadIT.java | 26 +++++-----
.../treemodel/auto/enhanced/PipeNowFunctionIT.java | 7 ---
.../treemodel/manual/IoTDBPipeReqAutoSliceIT.java | 12 +++++
.../IoTDBPipeTsFileDecompositionWithModsIT.java | 12 +++++
16 files changed, 179 insertions(+), 139 deletions(-)
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/AbstractPipeDualTreeModelAutoIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/AbstractPipeDualTreeModelAutoIT.java
index a7fae02f6d1..807f11eff9c 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/AbstractPipeDualTreeModelAutoIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/AbstractPipeDualTreeModelAutoIT.java
@@ -28,29 +28,36 @@ import org.apache.iotdb.itbase.env.BaseEnv;
import org.awaitility.Awaitility;
import org.junit.After;
+import org.junit.AfterClass;
import org.junit.Before;
+import org.junit.BeforeClass;
import java.io.File;
+import java.util.Arrays;
import java.util.Objects;
import java.util.concurrent.TimeUnit;
public abstract class AbstractPipeDualTreeModelAutoIT {
+ protected static ThreadLocal<BaseEnv> senderEnvContainer;
+ protected static ThreadLocal<BaseEnv> receiverEnvContainer;
+
protected BaseEnv senderEnv;
protected BaseEnv receiverEnv;
- @Before
- public void setUp() {
+ @BeforeClass
+ public static void setUp() {
MultiEnvFactory.createEnv(2);
- senderEnv = MultiEnvFactory.getEnv(0);
- receiverEnv = MultiEnvFactory.getEnv(1);
+ senderEnvContainer.set(MultiEnvFactory.getEnv(0));
+ receiverEnvContainer.set(MultiEnvFactory.getEnv(1));
setupConfig();
- senderEnv.initClusterEnvironment();
- receiverEnv.initClusterEnvironment();
+ senderEnvContainer.get().initClusterEnvironment();
+ receiverEnvContainer.get().initClusterEnvironment();
}
- protected void setupConfig() {
- senderEnv
+ protected static void setupConfig() {
+ senderEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -60,8 +67,13 @@ public abstract class AbstractPipeDualTreeModelAutoIT {
.setPipeMemoryManagementEnabled(false)
.setIsPipeEnableMemoryCheck(false)
.setPipeAutoSplitFullEnabled(false);
-
senderEnv.getConfig().getDataNodeConfig().setDataNodeMemoryProportion("3:3:1:1:3:1");
- receiverEnv
+ senderEnvContainer
+ .get()
+ .getConfig()
+ .getDataNodeConfig()
+ .setDataNodeMemoryProportion("3:3:1:1:3:1");
+ receiverEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -70,17 +82,33 @@ public abstract class AbstractPipeDualTreeModelAutoIT {
.setPipeMemoryManagementEnabled(false)
.setIsPipeEnableMemoryCheck(false)
.setPipeAutoSplitFullEnabled(false);
-
receiverEnv.getConfig().getDataNodeConfig().setDataNodeMemoryProportion("3:3:1:1:9:1");
+ receiverEnvContainer
+ .get()
+ .getConfig()
+ .getDataNodeConfig()
+ .setDataNodeMemoryProportion("3:3:1:1:9:1");
// 10 min, assert that the operations will not time out
- senderEnv.getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
- receiverEnv.getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+
senderEnvContainer.get().getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+
receiverEnvContainer.get().getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+ }
+
+ @AfterClass
+ public static void tearDown() {
+ senderEnvContainer.get().cleanClusterEnvironment();
+ receiverEnvContainer.get().cleanClusterEnvironment();
+ }
+
+ @Before
+ public void setEnv() {
+ senderEnv = senderEnvContainer.get();
+ receiverEnv = receiverEnvContainer.get();
}
@After
- public final void tearDown() {
- senderEnv.cleanClusterEnvironment();
- receiverEnv.cleanClusterEnvironment();
+ public void cleanEnvironment() {
+ TestUtils.executeNonQueries(senderEnv, Arrays.asList("drop database
root.**"), null);
+ TestUtils.executeNonQueries(receiverEnv, Arrays.asList("drop database
root.**"), null);
}
protected static void awaitUntilFlush(BaseEnv env) {
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeAlterIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeAlterIT.java
index 43fdf4ee5e5..cd53d5a5075 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeAlterIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeAlterIT.java
@@ -30,7 +30,6 @@ import
org.apache.iotdb.itbase.category.MultiClusterIT2DualTreeAutoBasic;
import
org.apache.iotdb.pipe.it.dual.treemodel.auto.AbstractPipeDualTreeModelAutoIT;
import org.junit.Assert;
-import org.junit.Before;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
@@ -50,12 +49,6 @@ import static org.junit.Assert.fail;
@Category({MultiClusterIT2DualTreeAutoBasic.class})
public class IoTDBPipeAlterIT extends AbstractPipeDualTreeModelAutoIT {
- @Override
- @Before
- public void setUp() {
- super.setUp();
- }
-
@Test
public void testBasicAlterPipe() throws Exception {
final DataNodeWrapper receiverDataNode = receiverEnv.getDataNodeWrapper(0);
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeAutoSplitIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeAutoSplitIT.java
index 31308f1c0e1..f90a175755b 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeAutoSplitIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeAutoSplitIT.java
@@ -31,7 +31,7 @@ import
org.apache.iotdb.itbase.category.MultiClusterIT2DualTreeAutoBasic;
import
org.apache.iotdb.pipe.it.dual.treemodel.auto.AbstractPipeDualTreeModelAutoIT;
import org.junit.Assert;
-import org.junit.Before;
+import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
@@ -47,14 +47,13 @@ import static org.junit.Assert.fail;
@RunWith(IoTDBTestRunner.class)
@Category({MultiClusterIT2DualTreeAutoBasic.class})
public class IoTDBPipeAutoSplitIT extends AbstractPipeDualTreeModelAutoIT {
-
- @Override
- @Before
- public void setUp() {
+ @BeforeClass
+ public static void setUp() {
MultiEnvFactory.createEnv(2);
- senderEnv = MultiEnvFactory.getEnv(0);
- receiverEnv = MultiEnvFactory.getEnv(1);
- senderEnv
+ senderEnvContainer.set(MultiEnvFactory.getEnv(0));
+ receiverEnvContainer.set(MultiEnvFactory.getEnv(1));
+ senderEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -64,7 +63,8 @@ public class IoTDBPipeAutoSplitIT extends
AbstractPipeDualTreeModelAutoIT {
.setPipeMemoryManagementEnabled(false)
.setIsPipeEnableMemoryCheck(false)
.setPipeAutoSplitFullEnabled(true); // set auto split to true
- receiverEnv
+ receiverEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -73,8 +73,8 @@ public class IoTDBPipeAutoSplitIT extends
AbstractPipeDualTreeModelAutoIT {
.setPipeMemoryManagementEnabled(false)
.setIsPipeEnableMemoryCheck(false)
.setPipeAutoSplitFullEnabled(true); // set auto split to true
- senderEnv.initClusterEnvironment();
- receiverEnv.initClusterEnvironment();
+ senderEnvContainer.get().initClusterEnvironment();
+ receiverEnvContainer.get().initClusterEnvironment();
}
@Test
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeDataSinkIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeDataSinkIT.java
index 1fc21fd22b0..44a4256e214 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeDataSinkIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeDataSinkIT.java
@@ -30,7 +30,6 @@ import
org.apache.iotdb.pipe.it.dual.treemodel.auto.AbstractPipeDualTreeModelAut
import org.apache.iotdb.rpc.TSStatusCode;
import org.junit.Assert;
-import org.junit.Before;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
@@ -48,12 +47,6 @@ import java.util.function.Consumer;
@Category({MultiClusterIT2DualTreeAutoBasic.class})
public class IoTDBPipeDataSinkIT extends AbstractPipeDualTreeModelAutoIT {
- @Override
- @Before
- public void setUp() {
- super.setUp();
- }
-
@Test
public void testThriftConnectorWithRealtimeFirstDisabled() throws Exception {
final DataNodeWrapper receiverDataNode = receiverEnv.getDataNodeWrapper(0);
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeProcessorIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeProcessorIT.java
index dfa7957a574..27c8c51ed45 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeProcessorIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeProcessorIT.java
@@ -32,7 +32,7 @@ import
org.apache.iotdb.pipe.it.dual.treemodel.auto.AbstractPipeDualTreeModelAut
import org.apache.iotdb.rpc.TSStatusCode;
import org.junit.Assert;
-import org.junit.Before;
+import org.junit.BeforeClass;
import org.junit.Ignore;
import org.junit.Test;
import org.junit.experimental.categories.Category;
@@ -48,13 +48,14 @@ import java.util.Set;
@Category({MultiClusterIT2DualTreeAutoBasic.class})
public class IoTDBPipeProcessorIT extends AbstractPipeDualTreeModelAutoIT {
- @Before
- public void setUp() {
+ @BeforeClass
+ public static void setUp() {
MultiEnvFactory.createEnv(2);
- senderEnv = MultiEnvFactory.getEnv(0);
- receiverEnv = MultiEnvFactory.getEnv(1);
+ senderEnvContainer.set(MultiEnvFactory.getEnv(0));
+ receiverEnvContainer.set(MultiEnvFactory.getEnv(1));
- senderEnv
+ senderEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -63,7 +64,8 @@ public class IoTDBPipeProcessorIT extends
AbstractPipeDualTreeModelAutoIT {
.setSchemaRegionConsensusProtocolClass(ConsensusFactory.RATIS_CONSENSUS)
.setPipeMemoryManagementEnabled(false)
.setIsPipeEnableMemoryCheck(false);
- receiverEnv
+ receiverEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -73,11 +75,11 @@ public class IoTDBPipeProcessorIT extends
AbstractPipeDualTreeModelAutoIT {
.setIsPipeEnableMemoryCheck(false);
// 10 min, assert that the operations will not time out
- senderEnv.getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
- receiverEnv.getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+
senderEnvContainer.get().getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+
receiverEnvContainer.get().getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
- senderEnv.initClusterEnvironment();
- receiverEnv.initClusterEnvironment();
+ senderEnvContainer.get().initClusterEnvironment();
+ receiverEnvContainer.get().initClusterEnvironment();
}
@Ignore
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeProtocolIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeProtocolIT.java
index b5e4127e946..e5124b183fc 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeProtocolIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/basic/IoTDBPipeProtocolIT.java
@@ -32,8 +32,10 @@ import
org.apache.iotdb.itbase.category.MultiClusterIT2DualTreeAutoBasic;
import
org.apache.iotdb.pipe.it.dual.treemodel.auto.AbstractPipeDualTreeModelAutoIT;
import org.apache.iotdb.rpc.TSStatusCode;
+import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
+import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
@@ -48,12 +50,21 @@ import java.util.Map;
@Category({MultiClusterIT2DualTreeAutoBasic.class})
public class IoTDBPipeProtocolIT extends AbstractPipeDualTreeModelAutoIT {
- @Override
+ @BeforeClass
+ public static void setUp() {
+ // Inner setup
+ }
+
@Before
- public void setUp() {
- MultiEnvFactory.createEnv(2);
- senderEnv = MultiEnvFactory.getEnv(0);
- receiverEnv = MultiEnvFactory.getEnv(1);
+ @Override
+ public void setEnv() {
+ // Inner set
+ }
+
+ @After
+ @Override
+ public final void cleanEnvironment() {
+ AbstractPipeDualTreeModelAutoIT.tearDown();
}
private void innerSetUp(
@@ -64,6 +75,10 @@ public class IoTDBPipeProtocolIT extends
AbstractPipeDualTreeModelAutoIT {
final int dataNodesNum,
int schemaRegionReplicationFactor,
int dataRegionReplicationFactor) {
+ MultiEnvFactory.createEnv(2);
+ senderEnv = MultiEnvFactory.getEnv(0);
+ receiverEnv = MultiEnvFactory.getEnv(1);
+
schemaRegionReplicationFactor = Math.min(schemaRegionReplicationFactor,
dataNodesNum);
dataRegionReplicationFactor = Math.min(dataRegionReplicationFactor,
dataNodesNum);
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeAutoConflictIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeAutoConflictIT.java
index f1a623143aa..5d4dfb34def 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeAutoConflictIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeAutoConflictIT.java
@@ -32,7 +32,7 @@ import
org.apache.iotdb.pipe.it.dual.treemodel.auto.AbstractPipeDualTreeModelAut
import org.apache.iotdb.rpc.TSStatusCode;
import org.junit.Assert;
-import org.junit.Before;
+import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
@@ -49,13 +49,14 @@ import java.util.Set;
@Category({MultiClusterIT2DualTreeAutoEnhanced.class})
public class IoTDBPipeAutoConflictIT extends AbstractPipeDualTreeModelAutoIT {
- @Before
- public void setUp() {
+ @BeforeClass
+ public static void setUp() {
MultiEnvFactory.createEnv(2);
- senderEnv = MultiEnvFactory.getEnv(0);
- receiverEnv = MultiEnvFactory.getEnv(1);
+ senderEnvContainer.set(MultiEnvFactory.getEnv(0));
+ receiverEnvContainer.set(MultiEnvFactory.getEnv(1));
- senderEnv
+ senderEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -64,7 +65,8 @@ public class IoTDBPipeAutoConflictIT extends
AbstractPipeDualTreeModelAutoIT {
.setDataRegionConsensusProtocolClass(ConsensusFactory.IOT_CONSENSUS)
.setPipeMemoryManagementEnabled(false)
.setIsPipeEnableMemoryCheck(false);
- receiverEnv
+ receiverEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -75,11 +77,11 @@ public class IoTDBPipeAutoConflictIT extends
AbstractPipeDualTreeModelAutoIT {
.setIsPipeEnableMemoryCheck(false);
// 10 min, assert that the operations will not time out
- senderEnv.getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
- receiverEnv.getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+
senderEnvContainer.get().getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+
receiverEnvContainer.get().getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
- senderEnv.initClusterEnvironment();
- receiverEnv.initClusterEnvironment();
+ senderEnvContainer.get().initClusterEnvironment();
+ receiverEnvContainer.get().initClusterEnvironment();
}
@Test
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeAutoDropIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeAutoDropIT.java
index 405c5f72371..35cfb934f7e 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeAutoDropIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeAutoDropIT.java
@@ -32,7 +32,6 @@ import
org.apache.iotdb.pipe.it.dual.treemodel.auto.AbstractPipeDualTreeModelAut
import org.apache.iotdb.rpc.TSStatusCode;
import org.junit.Assert;
-import org.junit.Before;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
@@ -52,12 +51,6 @@ import static org.awaitility.Awaitility.await;
@Category({MultiClusterIT2DualTreeAutoEnhanced.class})
public class IoTDBPipeAutoDropIT extends AbstractPipeDualTreeModelAutoIT {
- @Override
- @Before
- public void setUp() {
- super.setUp();
- }
-
@Test
public void testAutoDropInHistoricalTransfer() throws Exception {
final DataNodeWrapper receiverDataNode = receiverEnv.getDataNodeWrapper(0);
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeClusterIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeClusterIT.java
index ff06e6531cf..b0750c8e902 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeClusterIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeClusterIT.java
@@ -42,7 +42,7 @@ import org.apache.iotdb.rpc.TSStatusCode;
import org.apache.thrift.TException;
import org.junit.Assert;
-import org.junit.Before;
+import org.junit.BeforeClass;
import org.junit.Ignore;
import org.junit.Test;
import org.junit.experimental.categories.Category;
@@ -65,15 +65,14 @@ import static org.junit.Assert.fail;
@RunWith(IoTDBTestRunner.class)
@Category({MultiClusterIT2DualTreeAutoEnhanced.class})
public class IoTDBPipeClusterIT extends AbstractPipeDualTreeModelAutoIT {
-
- @Override
- @Before
- public void setUp() {
+ @BeforeClass
+ public static void setUp() {
MultiEnvFactory.createEnv(2);
- senderEnv = MultiEnvFactory.getEnv(0);
- receiverEnv = MultiEnvFactory.getEnv(1);
+ senderEnvContainer.set(MultiEnvFactory.getEnv(0));
+ receiverEnvContainer.set(MultiEnvFactory.getEnv(1));
- senderEnv
+ senderEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -82,9 +81,10 @@ public class IoTDBPipeClusterIT extends
AbstractPipeDualTreeModelAutoIT {
.setDataRegionConsensusProtocolClass(ConsensusFactory.IOT_CONSENSUS)
.setPipeMemoryManagementEnabled(false)
.setIsPipeEnableMemoryCheck(false);
-
senderEnv.getConfig().getConfigNodeConfig().setLeaderDistributionPolicy("HASH");
+
senderEnvContainer.get().getConfig().getConfigNodeConfig().setLeaderDistributionPolicy("HASH");
- receiverEnv
+ receiverEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -97,11 +97,11 @@ public class IoTDBPipeClusterIT extends
AbstractPipeDualTreeModelAutoIT {
.setIsPipeEnableMemoryCheck(false);
// 10 min, assert that the operations will not time out
- senderEnv.getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
- receiverEnv.getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+
senderEnvContainer.get().getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+
receiverEnvContainer.get().getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
- senderEnv.initClusterEnvironment(3, 3, 180);
- receiverEnv.initClusterEnvironment(3, 3, 180);
+ senderEnvContainer.get().initClusterEnvironment(3, 3, 180);
+ receiverEnvContainer.get().initClusterEnvironment(3, 3, 180);
}
@Test
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeConditionalOperationsIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeConditionalOperationsIT.java
index 7d0e7f71c0a..ec110ead1a5 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeConditionalOperationsIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeConditionalOperationsIT.java
@@ -29,7 +29,6 @@ import
org.apache.iotdb.itbase.category.MultiClusterIT2DualTreeAutoEnhanced;
import
org.apache.iotdb.pipe.it.dual.treemodel.auto.AbstractPipeDualTreeModelAutoIT;
import org.junit.Assert;
-import org.junit.Before;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
@@ -45,12 +44,6 @@ import static org.junit.Assert.fail;
@Category({MultiClusterIT2DualTreeAutoEnhanced.class})
public class IoTDBPipeConditionalOperationsIT extends
AbstractPipeDualTreeModelAutoIT {
- @Override
- @Before
- public void setUp() {
- super.setUp();
- }
-
@Test
public void testBasicCreatePipeIfNotExists() throws Exception {
final DataNodeWrapper receiverDataNode = receiverEnv.getDataNodeWrapper(0);
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeIdempotentIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeIdempotentIT.java
index fb499ae5e43..c0d026a98a1 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeIdempotentIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeIdempotentIT.java
@@ -32,7 +32,7 @@ import
org.apache.iotdb.pipe.it.dual.treemodel.auto.AbstractPipeDualTreeModelAut
import org.apache.iotdb.rpc.TSStatusCode;
import org.junit.Assert;
-import org.junit.Before;
+import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
@@ -48,17 +48,17 @@ import java.util.Set;
@Category({MultiClusterIT2DualTreeAutoEnhanced.class})
public class IoTDBPipeIdempotentIT extends AbstractPipeDualTreeModelAutoIT {
- @Override
- @Before
- public void setUp() {
+ @BeforeClass
+ public static void setUp() {
MultiEnvFactory.createEnv(2);
- senderEnv = MultiEnvFactory.getEnv(0);
- receiverEnv = MultiEnvFactory.getEnv(1);
+ senderEnvContainer.set(MultiEnvFactory.getEnv(0));
+ receiverEnvContainer.set(MultiEnvFactory.getEnv(1));
// All the schema operations must be under the same database to
// be in the same region, therefore a non-idempotent operation can block
the next one
// and fail the IT
- senderEnv
+ senderEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -71,7 +71,8 @@ public class IoTDBPipeIdempotentIT extends
AbstractPipeDualTreeModelAutoIT {
.setPipeMemoryManagementEnabled(false)
.setIsPipeEnableMemoryCheck(false);
- receiverEnv
+ receiverEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -82,11 +83,11 @@ public class IoTDBPipeIdempotentIT extends
AbstractPipeDualTreeModelAutoIT {
.setIsPipeEnableMemoryCheck(false);
// 10 min, assert that the operations will not time out
- senderEnv.getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
- receiverEnv.getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+
senderEnvContainer.get().getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+
receiverEnvContainer.get().getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
- senderEnv.initClusterEnvironment();
- receiverEnv.initClusterEnvironment();
+ senderEnvContainer.get().initClusterEnvironment();
+ receiverEnvContainer.get().initClusterEnvironment();
}
@Test
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeSinkCompressionIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeSinkCompressionIT.java
index 3900506a658..2bf6b3002f7 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeSinkCompressionIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeSinkCompressionIT.java
@@ -36,7 +36,7 @@ import
org.apache.iotdb.pipe.it.dual.treemodel.auto.AbstractPipeDualTreeModelAut
import org.apache.iotdb.rpc.TSStatusCode;
import org.junit.Assert;
-import org.junit.Before;
+import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
@@ -57,15 +57,15 @@ import static org.junit.Assert.fail;
@Category({MultiClusterIT2DualTreeAutoEnhanced.class})
public class IoTDBPipeSinkCompressionIT extends
AbstractPipeDualTreeModelAutoIT {
- @Override
- @Before
- public void setUp() {
+ @BeforeClass
+ public static void setUp() {
// Override to enable air-gap
MultiEnvFactory.createEnv(2);
- senderEnv = MultiEnvFactory.getEnv(0);
- receiverEnv = MultiEnvFactory.getEnv(1);
+ senderEnvContainer.set(MultiEnvFactory.getEnv(0));
+ receiverEnvContainer.set(MultiEnvFactory.getEnv(1));
- senderEnv
+ senderEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -75,7 +75,8 @@ public class IoTDBPipeSinkCompressionIT extends
AbstractPipeDualTreeModelAutoIT
.setIsPipeEnableMemoryCheck(false)
.setPipeAutoSplitFullEnabled(false);
- receiverEnv
+ receiverEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -87,11 +88,11 @@ public class IoTDBPipeSinkCompressionIT extends
AbstractPipeDualTreeModelAutoIT
.setPipeAutoSplitFullEnabled(false);
// 10 min, assert that the operations will not time out
- senderEnv.getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
- receiverEnv.getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+
senderEnvContainer.get().getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+
receiverEnvContainer.get().getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
- senderEnv.initClusterEnvironment();
- receiverEnv.initClusterEnvironment();
+ senderEnvContainer.get().initClusterEnvironment();
+ receiverEnvContainer.get().initClusterEnvironment();
}
@Test
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeWithLoadIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeWithLoadIT.java
index 26790479f22..3b13f1395ea 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeWithLoadIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/IoTDBPipeWithLoadIT.java
@@ -32,7 +32,7 @@ import
org.apache.iotdb.pipe.it.dual.treemodel.auto.AbstractPipeDualTreeModelAut
import org.apache.iotdb.rpc.TSStatusCode;
import org.junit.Assert;
-import org.junit.Before;
+import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
@@ -46,13 +46,14 @@ import java.util.Map;
@Category({MultiClusterIT2DualTreeAutoEnhanced.class})
public class IoTDBPipeWithLoadIT extends AbstractPipeDualTreeModelAutoIT {
- @Before
- public void setUp() {
+ @BeforeClass
+ public static void setUp() {
MultiEnvFactory.createEnv(2);
- senderEnv = MultiEnvFactory.getEnv(0);
- receiverEnv = MultiEnvFactory.getEnv(1);
+ senderEnvContainer.set(MultiEnvFactory.getEnv(0));
+ receiverEnvContainer.set(MultiEnvFactory.getEnv(1));
- senderEnv
+ senderEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -64,7 +65,8 @@ public class IoTDBPipeWithLoadIT extends
AbstractPipeDualTreeModelAutoIT {
.setEnableCrossSpaceCompaction(false)
.setPipeMemoryManagementEnabled(false)
.setIsPipeEnableMemoryCheck(false);
- receiverEnv
+ receiverEnvContainer
+ .get()
.getConfig()
.getCommonConfig()
.setAutoCreateSchemaEnabled(true)
@@ -74,13 +76,13 @@ public class IoTDBPipeWithLoadIT extends
AbstractPipeDualTreeModelAutoIT {
.setIsPipeEnableMemoryCheck(false);
// 10 min, assert that the operations will not time out
- senderEnv.getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
- receiverEnv.getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+
senderEnvContainer.get().getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
+
receiverEnvContainer.get().getConfig().getCommonConfig().setDnConnectionTimeoutMs(600000);
-
senderEnv.getConfig().getConfigNodeConfig().setLeaderDistributionPolicy("HASH");
+
senderEnvContainer.get().getConfig().getConfigNodeConfig().setLeaderDistributionPolicy("HASH");
- senderEnv.initClusterEnvironment();
- receiverEnv.initClusterEnvironment();
+ senderEnvContainer.get().initClusterEnvironment();
+ receiverEnvContainer.get().initClusterEnvironment();
}
/**
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/PipeNowFunctionIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/PipeNowFunctionIT.java
index f2735938e16..fe808a2966b 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/PipeNowFunctionIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/auto/enhanced/PipeNowFunctionIT.java
@@ -34,7 +34,6 @@ import
org.apache.iotdb.pipe.it.dual.treemodel.auto.AbstractPipeDualTreeModelAut
import org.apache.iotdb.rpc.TSStatusCode;
import org.junit.Assert;
-import org.junit.Before;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
@@ -52,12 +51,6 @@ import static org.junit.Assert.fail;
@Category({MultiClusterIT2DualTreeAutoEnhanced.class})
public class PipeNowFunctionIT extends AbstractPipeDualTreeModelAutoIT {
- @Override
- @Before
- public void setUp() {
- super.setUp();
- }
-
@Test
public void testPipeNowFunction() throws Exception {
final DataNodeWrapper receiverDataNode = receiverEnv.getDataNodeWrapper(0);
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/manual/IoTDBPipeReqAutoSliceIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/manual/IoTDBPipeReqAutoSliceIT.java
index 6f123c0da56..fd632545860 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/manual/IoTDBPipeReqAutoSliceIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/manual/IoTDBPipeReqAutoSliceIT.java
@@ -22,6 +22,7 @@ package org.apache.iotdb.pipe.it.dual.treemodel.manual;
import org.apache.iotdb.commons.utils.function.CheckedTriConsumer;
import org.apache.iotdb.db.it.utils.TestUtils;
import org.apache.iotdb.isession.ISession;
+import org.apache.iotdb.it.env.MultiEnvFactory;
import org.apache.iotdb.it.framework.IoTDBTestRunner;
import org.apache.iotdb.itbase.category.MultiClusterIT2DualTreeManual;
import org.apache.iotdb.rpc.IoTDBConnectionException;
@@ -34,6 +35,7 @@ import org.apache.tsfile.write.record.Tablet;
import org.apache.tsfile.write.schema.IMeasurementSchema;
import org.apache.tsfile.write.schema.MeasurementSchema;
import org.junit.Assert;
+import org.junit.BeforeClass;
import org.junit.Ignore;
import org.junit.Test;
import org.junit.experimental.categories.Category;
@@ -50,6 +52,16 @@ import java.util.Random;
public class IoTDBPipeReqAutoSliceIT extends AbstractPipeDualTreeModelManualIT
{
private static final int generateDataSize = 10;
+ @BeforeClass
+ public static void setUp() {
+ MultiEnvFactory.createEnv(2);
+ senderEnvContainer.set(MultiEnvFactory.getEnv(0));
+ receiverEnvContainer.set(MultiEnvFactory.getEnv(1));
+ setupConfig();
+ senderEnvContainer.get().initClusterEnvironment();
+ receiverEnvContainer.get().initClusterEnvironment();
+ }
+
protected static void setupConfig() {
AbstractPipeDualTreeModelManualIT.setupConfig();
senderEnvContainer
diff --git
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/manual/IoTDBPipeTsFileDecompositionWithModsIT.java
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/manual/IoTDBPipeTsFileDecompositionWithModsIT.java
index 7a448e8a8a7..f5a5ca21ed0 100644
---
a/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/manual/IoTDBPipeTsFileDecompositionWithModsIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/pipe/it/dual/treemodel/manual/IoTDBPipeTsFileDecompositionWithModsIT.java
@@ -20,9 +20,11 @@
package org.apache.iotdb.pipe.it.dual.treemodel.manual;
import org.apache.iotdb.db.it.utils.TestUtils;
+import org.apache.iotdb.it.env.MultiEnvFactory;
import org.apache.iotdb.it.framework.IoTDBTestRunner;
import org.apache.iotdb.itbase.category.MultiClusterIT2DualTreeManual;
+import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
@@ -36,6 +38,16 @@ import static
org.apache.iotdb.db.it.utils.TestUtils.executeNonQueryWithRetry;
@Category({MultiClusterIT2DualTreeManual.class})
public class IoTDBPipeTsFileDecompositionWithModsIT extends
AbstractPipeDualTreeModelManualIT {
+ @BeforeClass
+ public static void setUp() {
+ MultiEnvFactory.createEnv(2);
+ senderEnvContainer.set(MultiEnvFactory.getEnv(0));
+ receiverEnvContainer.set(MultiEnvFactory.getEnv(1));
+ setupConfig();
+ senderEnvContainer.get().initClusterEnvironment();
+ receiverEnvContainer.get().initClusterEnvironment();
+ }
+
protected static void setupConfig() {
AbstractPipeDualTreeModelManualIT.setupConfig();
senderEnvContainer.get().getConfig().getConfigNodeConfig().setLeaderDistributionPolicy("HASH");
