[HELIX-72] Allow pluggable rebalancer in controller
[HELIX-7]Fix test failure
[HELIX-7] fix test failures