Tests: Control reallocation in upgrade test to stop rebalancing causing upgrade requests to be lost.