Collection<BalancerBlock> srcBlockList =
new TreeSet<BalancerBlock>(new BalancerBlock.BalancerBlockComparator());
int[] replicas = new int[] {1, 2, 3, 1, 3, 1};
int id = 0;
for (int locationsNum : replicas) {
BalancerBlock block = new BalancerBlock(new Block(id, 5, id));
id++;
for (int i = 0; i < locationsNum; i++) {
block.addLocation(new Balancer.Target(new DatanodeInfo(), 30));
}
srcBlockList.add(block);
}
Arrays.sort(replicas);
int i = 0;
for (BalancerBlock block : srcBlockList) {
assertEquals(replicas[replicas.length - i - 1], block.getLocations().size());
i++;
}
assertEquals(i, replicas.length);
}