Skip to content

Commit ac89802

Browse files
committed
enhance UT
1 parent 9edc503 commit ac89802

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

iotdb-core/confignode/src/test/java/org/apache/iotdb/confignode/manager/load/balancer/region/GreedyCopySetRemoveNodeReplicaSelectTest.java

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -81,15 +81,20 @@ public void testSelectDestNode() {
8181
DATA_REGION_PER_DATA_NODE * TEST_DATA_NODE_NUM / DATA_REPLICATION_FACTOR;
8282

8383
List<TRegionReplicaSet> allocateResult = new ArrayList<>();
84+
List<TRegionReplicaSet> databaseAllocateResult = new ArrayList<>();
8485
for (int index = 0; index < dataRegionGroupNum; index++) {
85-
allocateResult.add(
86+
TRegionReplicaSet replicaSet =
8687
GCR_ALLOCATOR.generateOptimalRegionReplicasDistribution(
8788
AVAILABLE_DATA_NODE_MAP,
8889
FREE_SPACE_MAP,
8990
allocateResult,
9091
allocateResult,
9192
DATA_REPLICATION_FACTOR,
92-
new TConsensusGroupId(TConsensusGroupType.DataRegion, index)));
93+
new TConsensusGroupId(TConsensusGroupType.DataRegion, index));
94+
TRegionReplicaSet replicaSetCopy = new TRegionReplicaSet(replicaSet);
95+
96+
allocateResult.add(replicaSet);
97+
databaseAllocateResult.add(replicaSetCopy);
9398
}
9499

95100
List<TRegionReplicaSet> migratedReplicas =
@@ -158,7 +163,7 @@ public void testSelectDestNode() {
158163
}
159164
Map<TConsensusGroupId, TRegionReplicaSet> remainReplicasMap = new HashMap<>();
160165
Map<String, List<TRegionReplicaSet>> databaseAllocatedRegionGroupMap = new HashMap<>();
161-
databaseAllocatedRegionGroupMap.put("database", allocateResult);
166+
databaseAllocatedRegionGroupMap.put("database", databaseAllocateResult);
162167

163168
for (TRegionReplicaSet remainReplicaSet : remainReplicas) {
164169
remainReplicasMap.put(remainReplicaSet.getRegionId(), remainReplicaSet);

0 commit comments

Comments
 (0)