Unit test for the rebalance scheduler