[test] fix flaky issue4199-cookie

rogerwang · rogerwang · commit a00bd01262cb · 2017-12-14T13:05:38.000+08:00
diff --git a/test/sanity/issue4199-cookie/test.py b/test/sanity/issue4199-cookie/test.py
@@ -12,11 +12,9 @@
 
 driver = webdriver.Chrome(executable_path=os.environ['CHROMEDRIVER'], chrome_options=chrome_options)
 driver.implicitly_wait(2)
-time.sleep(1)
 try:
     print driver.current_url
-    cookie = driver.find_element_by_id('cookie').get_attribute('innerHTML')
+    cookie = wait_for_element_id_content(driver, 'cookie', 'Hi there')
     print cookie
-    assert ('Hi there' in cookie)
 finally:
     driver.quit()
