void setPartitionState(String partition, PartitionState partitionState) {
    for (IdealState entry : resourcesToIdealStates.values()) {
      for (String partitionInIdealState : entry.getPartitionSet()) {
        if (partitionInIdealState.equals(partition)) {
          partitionToPartitionStates.put(partition, partitionState);
        }
      }
    }
  }
Sets 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/MockHelixAdmin.java