void setPartitionState(String partition, PartitionState partitionState) {
    for (MockHelixAdmin helixAdmin : helixAdmins.values()) {
      helixAdmin.setPartitionState(partition, partitionState);
    }
  }
Set the state of a partition
@param partition partition for which state needs to be updated
@param partitionState {@link PartitionState} that needs to be set
ambry-clustermap/src/test/java/com/github/ambry/clustermap/MockHelixCluster.java