InMemoryStore getStore(PartitionId partitionId) {
    return storesByPartition.get(partitionId);
  }
Gets the in-memory store for the partition
@param partitionId partition id
@return in-memory store
ambry-replication/src/test/java/com/github/ambry/replication/MockHost.java