From 9d3d6d1de7848497cd0589db914723dd352a6532 Mon Sep 17 00:00:00 2001 From: Vitor Avila <96086495+Vitor-Avila@users.noreply.github.com> Date: Wed, 29 May 2024 16:05:48 -0300 Subject: [PATCH] fix(reports): Update the element class to wait for when taking a screenshot (#28745) (cherry picked from commit 020c79970f49a599402befc475c2bd18c275146b) --- superset/utils/webdriver.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/superset/utils/webdriver.py b/superset/utils/webdriver.py index 4552600fc9..661fc6cd4a 100644 --- a/superset/utils/webdriver.py +++ b/superset/utils/webdriver.py @@ -187,7 +187,7 @@ class WebDriverPlaywright(WebDriverProxy): try: # chart containers didn't render logger.debug("Wait for chart containers to draw at url: %s", url) - slice_container_locator = page.locator(".slice_container") + slice_container_locator = page.locator(".chart-container") slice_container_locator.first.wait_for() for slice_container_elem in slice_container_locator.all(): slice_container_elem.wait_for() @@ -377,7 +377,7 @@ class WebDriverSelenium(WebDriverProxy): logger.debug("Wait for chart containers to draw at url: %s", url) WebDriverWait(driver, self._screenshot_locate_wait).until( EC.visibility_of_all_elements_located( - (By.CLASS_NAME, "slice_container") + (By.CLASS_NAME, "chart-container") ) ) except TimeoutException as ex: