[TESTS] Make sure to wait for all shards to be allocated before running the test.