multi table tests cypress -> selenium

Author: Marc-André Rivet

tests/cypress/tests/standalone/select_row_column_test.ts

@@ -139,12 +139,28 @@ def get(self, row=0):
     def hide(self, row=0):
         self.get(row).find_element_by_css_selector(".column-header--hide").click()
 
+    @preconditions(_validate_row)
+    def is_selected(self, row=0):
+        return self.mixin.driver.execute_script(
+            """
+            return document.querySelector('#{} {} tbody tr:nth-of-type({}) th.dash-header[data-dash-column="{}"]:not(.phantom-cell) .column-header--select input').checked;
+        """.format(
+                self.id, self.state, row + 1, self.col_id
+            )
+        )
+
     @preconditions(_validate_row)
     def move_to(self, row=0):
         ac = ActionChains(self.mixin.driver)
         ac.move_to_element(self.get(row))
         return ac.perform()
 
+    @preconditions(_validate_row)
+    def select(self, row=0):
+        self.get(row).find_element_by_css_selector(
+            ".column-header--select input"
+        ).click()
+
     @preconditions(_validate_row)
     def sort(self, row=0):
         self.get(row).find_element_by_css_selector(".column-header--sort").click()
@@ -325,6 +341,17 @@ def _wait_for_table(self, id, state=_ANY):
     def table(self, id):
         return DataTableFacade(id, self)
 
+    def get_table_ids(self):
+        return self.driver.execute_script(
+            """
+            return Array.from(
+                document.querySelectorAll('.dash-spreadsheet-container')
+            ).map(
+                e => e.parentElement.getAttribute('id')
+            )
+        """
+        )
+
     def copy(self):
         with self.hold(Keys.CONTROL):
             self.send_keys("c")

tests/selenium/conftest.py

@@ -0,0 +1,68 @@
+import dash
+from dash.testing import wait
+
+from dash_table import DataTable
+from dash_html_components import Div
+
+import pandas as pd
+
+url = "https://github.com/plotly/datasets/raw/master/" "26k-consumer-complaints.csv"
+rawDf = pd.read_csv(url, nrows=100)
+df = rawDf.to_dict("records")
+
+
+def get_app(props=dict()):
+    app = dash.Dash(__name__)
+
+    baseProps = dict(
+        columns=[dict(name=i, id=i, selectable=True) for i in rawDf.columns],
+        data=df,
+        editable=True,
+        filter_action="native",
+        fixed_columns={"headers": True},
+        fixed_rows={"headers": True},
+        page_action="native",
+        row_deletable=True,
+        row_selectable=True,
+        sort_action="native",
+    )
+
+    baseProps.update(props)
+
+    app.layout = Div([DataTable(**baseProps), DataTable(**baseProps)])
+
+    return app
+
+
+def test_tbmu001_select_row(test):
+    test.start_server(get_app())
+
+    wait.until(lambda: len(test.get_table_ids()) == 2, 3)
+    ids = test.get_table_ids()
+
+    table1 = test.table(ids[0])
+    table2 = test.table(ids[1])
+
+    table1.row(0).select()
+    table2.row(1).select()
+
+    assert table1.row(0).is_selected()
+    assert table2.row(1).is_selected()
+
+    assert test.get_log_errors() == []
+
+
+def test_tbmu002_select_column(test):
+    test.start_server(get_app(dict(column_selectable="single")))
+
+    wait.until(lambda: len(test.get_table_ids()) == 2, 3)
+    ids = test.get_table_ids()
+
+    table1 = test.table(ids[0])
+    table2 = test.table(ids[1])
+
+    table1.column("Complaint ID").select()
+    table2.column("Product").select()
+
+    assert table1.column("Complaint ID").is_selected()
+    assert table2.column("Product").is_selected()