protected JSONArray getDataNodes(int basePort, int sslPort, int http2Port, JSONArray disks) throws JSONException {
      if (rackAware) {
        return getJsonArrayDataNodesRackAware(dataNodeCount, getLocalHost(), basePort, sslPort, http2Port, numRacks,
            HardwareState.AVAILABLE, disks);
      }
      return getJsonArrayDataNodes(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