[TEST] Fix SearchStatsTests to have all shards allocated

If randomization brings up a single shard per index in this test
we might run our searches on only one index which causes the assertions
to fail afterwards that's why we need to wait until everything is alloated.
[TEST] Beef up SearchStatsTests