@NotNull
  private Function getRebalanceFunction(InternalDistributedMember dsMember) {
    Function rebalanceFunction;
    if (dsMember.getVersion()
        .isOlderThan(KnownVersion.GEODE_1_12_0)) {
      rebalanceFunction =
          new org.apache.geode.management.internal.cli.functions.RebalanceFunction();
    } else {
      rebalanceFunction = new RebalanceFunction();
    }
    return rebalanceFunction;
  }
This class was introduced so that it can be mocked
to all executeRebalanceOnDS to be unit tested
geode-core/src/main/java/org/apache/geode/management/internal/operation/RebalanceOperationPerformer.java