@Override
  public int compareTo(PartitionId o) {
    MockPartitionId mockPartition = (MockPartitionId) o;
    return Long.compare(partition, mockPartition.partition);
  }
Mock partition id for unit tests
ambry-test-utils/src/main/java/com/github/ambry/clustermap/MockPartitionId.java