[test] fix flaky issue5762-user-data-dir

rogerwang · rogerwang · commit ca8f707eee4c · 2017-12-15T17:00:04.000+08:00
diff --git a/test/sanity/issue5762-user-data-dir/test.py b/test/sanity/issue5762-user-data-dir/test.py
@@ -69,8 +69,27 @@ def copytree(src, dst, symlinks=False, ignore=None):
 p = subprocess.Popen([exe])
 time.sleep(10)
 
+user_data_dir_exists = False
+check_file_exists = False
+
+timeout = 30
+while timeout > 0:
+    if os.path.exists(user_data_dir):
+        user_data_dir_exists = True
+        break
+    time.sleep(1)
+    timeout = timeout - 1
+
+if user_data_dir_exists:
+    timeout = 30
+    while timeout > 0:
+        if os.path.exists(check_file):
+            check_file_exists = True
+            break
+        time.sleep(1)
+        timeout = timeout - 1
 try:
-    assert os.path.exists(user_data_dir)
-    assert os.path.exists(check_file)
+    assert user_data_dir_exists
+    assert check_file_exists
 finally:
     p.terminate()
