[test] fix document-start-end with wait_for_element_id

rogerwang · rogerwang · commit a6c9c2e5b48e · 2016-11-24T09:54:40.000+08:00
diff --git a/test/sanity/document-start-end/test.py b/test/sanity/document-start-end/test.py
@@ -1,17 +1,21 @@
 import time
 import os
+import sys
 
 from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+from nw_util import *
+
 chrome_options = Options()
 chrome_options.add_argument("nwapp=" + os.path.dirname(os.path.abspath(__file__)))
 
 driver = webdriver.Chrome(executable_path=os.environ['CHROMEDRIVER'], chrome_options=chrome_options)
 time.sleep(1)
 try:
     print driver.current_url
-    result = driver.find_element_by_id('result').get_attribute('innerHTML')
-    result2 = driver.find_element_by_id('result2').get_attribute('innerHTML')
+    result = wait_for_element_id(driver, 'result')
+    result2 = wait_for_element_id(driver, 'result2')
     print result
     print result2
     assert(result == 'success from popup' and result2 == 'startiframe')
