Fix broken fbcode//mcrouter:mcrouter_integration_tests - test_migrated_failover (mcrout

Reviewed By: stuclar

Differential Revision: D114996665

fbshipit-source-id: a1ce5c0c066dd9abbb285bf886aafabeeebf8eea

Author: generatedunixname2012458493028417

mcrouter/test/test_migrated_failover.py

@@ -6,6 +6,8 @@
 
 # pyre-unsafe
 
+import time
+
 from carbon.carbon_result.thrift_types import Result
 from mcrouter.test.MCProcess import MockMemcached
 from mcrouter.test.McrouterTestCase import McrouterTestCase
@@ -37,6 +39,16 @@ def test_migrated_failover(self):
 
         mc_a.terminate()
 
-        reply = client.mcGet(b"key")
+        reply = None
+        for _ in range(20):
+            try:
+                reply = client.mcGet(b"key")
+            except Exception:
+                reply = None
+            if reply is not None and reply.result == Result.FOUND:
+                break
+            time.sleep(0.5)
+
+        self.assertIsNotNone(reply)
         self.assertEqual(Result.FOUND, reply.result)
         self.assertEqual(b"b", bytes(reply.value))