add another wait for chart element (#15108)

This commit is contained in:
Elizabeth Thompson
2021-06-15 10:00:36 -07:00
committed by GitHub
parent 7b87eb15bd
commit 7dc0cee5be
2 changed files with 15 additions and 7 deletions

View File

@@ -114,6 +114,12 @@ class WebDriverProxy:
WebDriverWait(driver, self._screenshot_load_wait).until_not(
EC.presence_of_all_elements_located((By.CLASS_NAME, "loading"))
)
logger.debug("Wait for chart to have content")
WebDriverWait(driver, self._screenshot_locate_wait).until(
EC.visibility_of_all_elements_located(
(By.CLASS_NAME, "slice_container")
)
)
logger.info("Taking a PNG screenshot or url %s", url)
img = element.screenshot_as_png
except TimeoutException: