[test] fix race condition in issue5980-aws-sdk-embedded-youtobe-video-crash

GnorTech · GnorTech · commit 5c35a0d6a081 · 2018-02-16T20:11:01.000+08:00
diff --git a/test/sanity/issue5980-aws-sdk-embedded-youtobe-video-crash/index.html b/test/sanity/issue5980-aws-sdk-embedded-youtobe-video-crash/index.html
@@ -6,6 +6,6 @@
   <title>issue5980-aws-sdk-embedded-youtobe-vedio-crash</title>
 </head>
 <body>
-  <iframe src="https://www.youtube.com/embed/C0DPdy98e4c"></iframe>
+  <iframe src="https://www.youtube.com/embed/C0DPdy98e4c" id="iframe_a"></iframe>
 </body>
 </html>
diff --git a/test/sanity/issue5980-aws-sdk-embedded-youtobe-video-crash/test.py b/test/sanity/issue5980-aws-sdk-embedded-youtobe-video-crash/test.py
@@ -27,10 +27,22 @@
     switch_to_app(driver)
     print driver.current_url
     print 'waiting for crash'
-    wait_for_element_tag(driver, "iframe")
-    assert(driver.find_element_by_tag_name("iframe") is not None)
-    driver.switch_to_frame(driver.find_element_by_tag_name("iframe"))
-    wait_for_element_class(driver, "ytp-large-play-button")
+    wait_for_element_id(driver, "iframe_a")
+    driver.switch_to_frame("iframe_a")
+    timeout = 10
+    while timeout > 0:
+        try:
+            ret = driver.find_element_by_class_name('ytp-large-play-button')
+            break
+        except selenium.common.exceptions.NoSuchElementException:
+            driver.switch_to_default_content()
+            driver.switch_to_frame("iframe_a")
+        except selenium.common.exceptions.WebDriverException:
+            pass
+        time.sleep(1)
+        timeout = timeout - 1
+        if timeout <= 0:
+             raise Exception('Timeout when waiting for element ytp-large-play-button')
     assert(driver.find_element_by_class_name("ytp-large-play-button") is not None)
     print 'There is no crash'
 finally:
