Skip to content

Commit 2f201ba

Browse files
committed
Update IoTDBRegionMigrateWithDeletionMultiDataDirIT.java
1 parent d370851 commit 2f201ba

1 file changed

Lines changed: 12 additions & 4 deletions

File tree

integration-test/src/test/java/org/apache/iotdb/confignode/it/regionmigration/pass/daily/iotv1/IoTDBRegionMigrateWithDeletionMultiDataDirIT.java

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -85,19 +85,24 @@ public void testRegionMigratePreservesDeletionWithMultiDataDirs() throws Excepti
8585
Map<Integer, Pair<Integer, Set<Integer>>> dataRegionMapWithLeader =
8686
getDataRegionMapWithLeader(statement);
8787
int dataRegionIdForTest =
88-
dataRegionMapWithLeader.keySet().stream().max(Integer::compareTo).orElseThrow();
88+
dataRegionMapWithLeader.keySet().stream()
89+
.max(Integer::compareTo)
90+
.orElseThrow(() -> new AssertionError("No DataRegion found"));
8991
assertDeletionVisibleOnAllReplicas(statement, dataRegionIdForTest, 1);
9092

9193
Pair<Integer, Set<Integer>> leaderAndNodes = dataRegionMapWithLeader.get(dataRegionIdForTest);
9294
Set<Integer> allDataNodes = getAllDataNodes(statement);
9395
int leaderId = leaderAndNodes.getLeft();
9496
int followerId =
95-
leaderAndNodes.getRight().stream().filter(id -> id != leaderId).findFirst().orElseThrow();
97+
leaderAndNodes.getRight().stream()
98+
.filter(id -> id != leaderId)
99+
.findFirst()
100+
.orElseThrow(() -> new AssertionError("No follower DataNode found"));
96101
int destDataNodeId =
97102
allDataNodes.stream()
98103
.filter(id -> id != leaderId && id != followerId)
99104
.findFirst()
100-
.orElseThrow();
105+
.orElseThrow(() -> new AssertionError("No destination DataNode found"));
101106

102107
statement.execute(
103108
String.format(
@@ -132,7 +137,10 @@ private void assertDeletionVisibleOnAllReplicas(
132137
Set<Integer> replicaDataNodeIds = getReplicaDataNodeIds(statement, dataRegionId);
133138
for (int dataNodeId : replicaDataNodeIds) {
134139
DataNodeWrapper dataNodeWrapper =
135-
EnvFactory.getEnv().dataNodeIdToWrapper(dataNodeId).orElseThrow();
140+
EnvFactory.getEnv()
141+
.dataNodeIdToWrapper(dataNodeId)
142+
.orElseThrow(
143+
() -> new AssertionError("No DataNode wrapper found for DataNode " + dataNodeId));
136144
Awaitility.await()
137145
.atMost(2, TimeUnit.MINUTES)
138146
.pollDelay(500, TimeUnit.MILLISECONDS)

0 commit comments

Comments
 (0)