balancer_collection_status.js expects collection to balance in to few rounds

balancer_collection_status.js is waiting for a fixed number of balancer rounds before checking that the collection is balanced, however, if a stepdown happens in the middle of the balancing round, the round might finish without being successful, causing a failure.

We should run balancer rounds until the collection is balanced, instead of running a fixed number of rounds.