wait for shard to be allocated before executing a resume follow api

Author: martijnvg

x-pack/qa/rolling-upgrade-multi-cluster/src/test/java/org/elasticsearch/upgrades/CcrRollingUpgradeIT.java

@@ -272,7 +272,9 @@ public void testBiDirectionalIndexFollowing() throws Exception {
                     index(leaderClient(), "leader_index5", 128);
                     break;
                 case ALL:
+                    ensureGreen(followerClient(), "follower_index5");
                     resumeIndexFollowing(followerClient(), "follower_index5");
+                    ensureGreen(leaderClient(), "follower_index6");
                     resumeIndexFollowing(leaderClient(), "follower_index6");
 
                     assertTotalHitCount("follower_index5", 896, followerClient());
@@ -374,4 +376,13 @@ private static void resumeIndexFollowing(RestClient client, String followerIndex
         assertOK(client.performRequest(new Request("POST", "/" + followerIndex + "/_ccr/resume_follow")));
     }
 
+    private static void ensureGreen(RestClient client, String index) throws IOException {
+        Request request = new Request("GET", "/_cluster/health/" + index);
+        request.addParameter("wait_for_status", "green");
+        request.addParameter("wait_for_no_relocating_shards", "true");
+        request.addParameter("timeout", "70s");
+        request.addParameter("level", "shards");
+        client.performRequest(request);
+    }
+
 }