protected void updateDataNodeJsonArray(JSONArray dataNodeJsonArray, int basePort, int sslPort, int http2Port,
        JSONArray disks) throws JSONException {
      if (rackAware) {
        updateJsonArrayDataNodesRackAware(dataNodeJsonArray, dataNodeCount, getLocalHost(), basePort, sslPort,
            http2Port, numRacks, disks, HardwareState.AVAILABLE);
      } else {
        updateJsonArrayDataNodes(dataNodeJsonArray, dataNodeCount, getLocalHost(), basePort, sslPort, http2Port,
            HardwareState.AVAILABLE, disks);
      }
    }
@param rangeStart start of the range of partitions.
@param count the number of partitions in the range.
@param expectedClass the expected class of all of the partitions
@param expectedState the expected state of all of the partitions
ambry-clustermap/src/test/java/com/github/ambry/clustermap/TestUtils.java