sk0x50 commented on code in PR #5581: URL: https://github.com/apache/ignite-3/pull/5581#discussion_r2032933136
########## modules/distribution-zones/src/integrationTest/java/org/apache/ignite/internal/rebalance/ItRebalanceDistributedTest.java: ########## @@ -1062,45 +1086,96 @@ private Node getLeaseholderNodeForPartition(Node node, int partId) { .get(); } - private static Set<Assignment> getPartitionClusterNodes(Node node, int partNum) { - return getPartitionClusterNodes(node, TABLE_NAME, partNum); + private static Set<Assignment> getPartitionStableAssignments(Node node, int partNum) { + return getPartitionStableAssignments(node, TABLE_NAME, partNum); } - private static Set<Assignment> getPartitionClusterNodes(Node node, String tableName, int partNum) { - return Optional.ofNullable(getTableId(node, tableName)) - .map(tableId -> stablePartitionAssignments(node.metaStorageManager, tableId, partNum).join()) + private static Set<Assignment> getPartitionStableAssignments(Node node, String tableName, int partNum) { + int zoneId = getZoneIdByTableNameStrict(node.catalogManager, tableName, node.hybridClock.nowLong()); Review Comment: Extracted to `TestRebalanceUtil` class. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: notifications-unsubscr...@ignite.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org