[test] fix race condition in cookie-lost-devtools-close

rogerwang · rogerwang · commit 7439b6133114 · 2017-01-08T10:22:28.000+08:00
diff --git a/test/sanity/cookie-lost-devtools-close/test.py b/test/sanity/cookie-lost-devtools-close/test.py
@@ -30,7 +30,7 @@
 def click_expect(expected):
     global idx
     driver.find_element_by_id('get-cookie').click()
-    result = driver.find_element_by_id('result-%s' % idx).get_attribute('innerHTML')
+    result = wait_for_element_id(driver, 'result-%s' % idx)
     idx += 1
     print result
     assert(expected in result)
