tests: more fixes

cbouy · cbouy · commit ec2ce452b8ae · 2025-12-18T18:17:10.000+01:00
diff --git a/tests/test_interface.py b/tests/test_interface.py
@@ -60,10 +60,10 @@ def driver():
     options.add_argument("--enable-automation")
     options.add_argument("--no-sandbox")
     options.add_argument("--disable-dev-shm-usage")
-    options.add_argument("--start-maximized")
     driver = CustomDriver(options=options)
     driver.set_page_load_timeout(10)
     driver.set_window_size(1920, 1080)
+    driver.maximize_window()
     yield driver
     driver.quit()
 
@@ -248,7 +248,7 @@ def test_selection_check_uncheck_invert(driver: CustomDriver, html_doc):
 
 @pytest.mark.parametrize("prerender", [True, False])
 def test_image_size(driver: CustomDriver, df, prerender):
-    size = expected_size = (200, 300)
+    size = expected_size = (220, 200)
     grid = get_grid(df, size=size, prerender=prerender)
     doc = get_doc(
         grid,
@@ -483,8 +483,8 @@ def test_style(driver: CustomDriver, grid):
     assert el.value_of_css_property("color") == "rgba(255, 0, 0, 1)"
     el = driver.find_by_css_selector(".m2g-cell .data-_Name")
     assert el.value_of_css_property("color") == "rgba(0, 0, 255, 1)"
-    ActionChains(driver).click(driver.find_by_css_selector(".m2g-info")).perform()
-    el = driver.find_by_css_selector("div.m2g-popover span")
+    tooltip = driver.find_tooltip()
+    el = tooltip.find_element(By.TAG_NAME, "span")
     assert el.value_of_css_property("color") == "rgba(0, 0, 255, 1)"
 
 
@@ -713,7 +713,7 @@ def test_static_template(driver: CustomDriver, sdf_path):
     el = driver.find_by_css_selector("td.col-0")
     assert el.find_element(By.CLASS_NAME, "data-mols2grid-id").text == "8"
     tooltip = driver.get_tooltip_content(
-        selector=".m2g-cell-0", content_selector="div.popover-body"
+        trigger_selector=".m2g-cell-0", tooltip_selector="div.popover-body"
     )[:-1]
     assert tooltip == [
         ["strong", ["_Name"]],
diff --git a/tests/webdriver_utils.py b/tests/webdriver_utils.py
@@ -35,7 +35,7 @@ def __call__(self, driver):
 
 
 class CustomDriver(webdriver.Chrome):
-    def wait_for_img_load(self, max_delay=30, selector=".m2g-cell .data-img svg"):
+    def wait_for_img_load(self, max_delay=30, selector=".m2g-cell .data-img svg rect"):
         return WebDriverWait(self, max_delay).until(
             EC.presence_of_element_located((By.CSS_SELECTOR, selector))
         )
@@ -135,21 +135,34 @@ def _tree_to_list(cls, nodes: NodeList[minidom.Node | minidom.Text], data: list)
                     branch.append(subbranch)
                     cls._tree_to_list(node.childNodes, subbranch)
 
-    def get_tooltip_content(
-        self, selector=".m2g-cell .m2g-info", content_selector="div.m2g-popover"
-    ):
-        ActionChains(self).click(self.find_by_css_selector(selector)).perform()
-        tooltip = self.find_by_css_selector(content_selector)
+    def get_tooltip_content(self, **kwargs):
+        tooltip = self.find_tooltip(**kwargs)
         content = tooltip.get_attribute("innerHTML")
         doc = minidom.parseString(f"<body>{content}</body>")
         data = []
         self._tree_to_list(doc.childNodes[0].childNodes, data)
         return data
 
+    def find_tooltip(
+        self,
+        trigger_selector=".m2g-cell .m2g-info",
+        tooltip_selector="div.m2g-popover",
+        pause=0.2,
+    ):
+        el = self.find_clickable(By.CSS_SELECTOR, trigger_selector)
+        ActionChains(self).move_to_element(el).pause(pause).click().perform()
+        # selenium is a bit flaky with move_to_element,
+        # best to trigger through JS as well
+        self.execute_script(
+            'var hover = new Event("mouseenter"); '
+            f'document.querySelector("{trigger_selector}").dispatchEvent(hover);'
+        )
+        return self.find_by_css_selector(tooltip_selector)
+
     def trigger_callback(self, selector=".m2g-cell .m2g-callback", pause=0.2):
         self.wait_for_img_load()
         el = self.find_clickable(By.CSS_SELECTOR, selector)
-        (ActionChains(self).move_to_element(el).pause(pause).click().perform())
+        ActionChains(self).move_to_element(el).pause(pause).click().perform()
 
     def click_checkbox(self, is_empty=False):
         self.find_clickable(By.CSS_SELECTOR, ".m2g-cb").click()
