[test] fix flaky worker-nwbin

rogerwang · rogerwang · commit e0951764ae7e · 2017-12-15T17:02:18.000+08:00
diff --git a/test/sanity/worker-nwbin/test.py b/test/sanity/worker-nwbin/test.py
@@ -4,6 +4,9 @@
 import subprocess
 import platform
 from subprocess import Popen, PIPE
+import sys
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+from nw_util import *
 
 from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
@@ -24,8 +27,8 @@
 driver = webdriver.Chrome(executable_path=os.environ['CHROMEDRIVER'], chrome_options=chrome_options)
 try:
     print driver.current_url
-    result = driver.find_element_by_id('result')
-    print result.get_attribute('innerHTML')
-    assert("840" == result.get_attribute('innerHTML'))
+    result = wait_for_element_id_content(driver, 'result', '840')
+    print result
+    assert("840" == result)
 finally:
     driver.quit()
