[test] fix race condition in issue4286-inject-start-end

rogerwang · rogerwang · commit ca9ba83099c4 · 2016-11-24T09:54:40.000+08:00
diff --git a/test/sanity/issue4286-inject-start-end/test.py b/test/sanity/issue4286-inject-start-end/test.py
@@ -15,10 +15,10 @@
 try:
     print driver.current_url
     wait_window_handles(driver, 2)
-    result = driver.find_element_by_id('inject_start').get_attribute('innerHTML')
+    result = wait_for_element_id(driver, 'inject_start')
     print 'inject_js_start: %s' % result
     assert('success' in result)
-    result = driver.find_element_by_id('inject_end').get_attribute('innerHTML')
+    result = wait_for_element_id(driver, 'inject_end')
     print 'inject_js_end: %s' % result
     assert('success' in result)
 finally:
