Add getTableData method to html Table component

LabKey · Feb 4, 2025 · e797b06 · e797b06
1 parent 2bd83d5
commit e797b06
Showing 1 changed file with 35 additions and 3 deletions.
diff --git a/src/org/labkey/test/components/html/Table.java b/src/org/labkey/test/components/html/Table.java
@@ -26,7 +26,9 @@
 import org.openqa.selenium.WebElement;
 
 import java.util.ArrayList;
+import java.util.LinkedHashMap;
 import java.util.List;
+import java.util.Map;
 
 public class Table extends WebDriverComponent<Table.Elements>
 {
@@ -62,7 +64,7 @@ protected Elements newElementCache()
         return new Elements();
     }
 
-    protected class Elements extends Component.ElementCache
+    protected class Elements extends Component<?>.ElementCache
     {
         List<WebElement> rows;
 
@@ -86,15 +88,15 @@ public int getRowCount()
      */
     public List<String> getTableHeaderTexts()
     {
-        List<WebElement> headerEls = Locator.xpath("//thead//th").findElements(this);
+        List<WebElement> headerEls = Locator.xpath("./thead/tr[1]/th").findElements(this);
         List<String> columnHeaders = new ArrayList<>();
         for(WebElement headerEl : headerEls){columnHeaders.add(headerEl.getText());}
         return columnHeaders;
     }
 
     public int getTableHeaderIndex(String headerText)
     {
-        List<WebElement> headerEls = Locator.xpath("//thead//th").findElements(this);
+        List<WebElement> headerEls = Locator.xpath("./thead/tr[1]/th").findElements(this);
         int counter = 1;
         for(WebElement headerEl : headerEls)
         {
@@ -105,6 +107,36 @@ public int getTableHeaderIndex(String headerText)
         throw new RuntimeException( headerText + " column not found");
     }
 
+    /**
+     * Get table data as a list of maps. Each map represents a row.<br>
+     * Assumes a simple table with a single header row with no colspans and unique header labels
+     * @return table data
+     */
+    public List<Map<String, String>> getTableData()
+    {
+        List<Map<String, String>> data = new ArrayList<>();
+
+        List<String> headerTexts = getTableHeaderTexts();
+        List<WebElement> rows = elementCache().getRows();
+
+        for (WebElement row : rows)
+        {
+            List<String> dataTexts = getWrapper().getTexts(Locator.tag("td").findElements(row));
+            if (headerTexts.size() != dataTexts.size())
+            {
+                throw new IllegalStateException("Size of row %s doesn't match table header %s".formatted(dataTexts, headerTexts));
+            }
+            Map<String, String> rowMap = new LinkedHashMap<>();
+            for (int i = 0; i < headerTexts.size(); i++)
+            {
+                rowMap.put(headerTexts.get(i), dataTexts.get(i));
+            }
+            data.add(rowMap);
+        }
+
+        return data;
+    }
+
     public List<String> getTableHeaderColumnData(String headerText)
     {
         List<String> columnData = new ArrayList<>();