@homebound/beam 2.321.1 → 2.322.0

package/dist/utils/rtl.d.ts

@@ -29,7 +29,7 @@ export declare function rowAnd(r: RenderResult, rowNum: number, testId: string):
       "
     `);
  * */
-export declare function tableSnapshot(r: RenderResult): string;
+export declare function tableSnapshot(r: RenderResult, columnNames?: string[]): string;
 /** RTL wrapper for Beam's SuperDrawer/Modal context. */
 export declare const withBeamRTL: Wrapper;
 /**

package/dist/utils/rtl.js

@@ -76,17 +76,30 @@ exports.rowAnd = rowAnd;
       "
     `);
  * */
-function tableSnapshot(r) {
+function tableSnapshot(r, columnNames = []) {
     const tableEl = r.getByTestId("gridTable");
     const dataRows = Array.from(tableEl.querySelectorAll("[data-gridrow]"));
     const hasExpandableHeader = !!tableEl.querySelector(`[data-testid="expandableColumn"]`);
-    const tableDataAsStrings = dataRows.map((row) => {
+    let tableDataAsStrings = dataRows.map((row) => {
         return Array.from(row.childNodes).map(getTextFromTableCellNode);
     });
-    return toMarkupTableString({ tableRows: tableDataAsStrings, hasExpandableHeader });
+    // If the user wants a subset of columns, look for column names
+    if (columnNames.length > 0) {
+        const headerCells = tableDataAsStrings[0];
+        if (headerCells) {
+            const columnIndices = columnNames.map((name) => {
+                const i = headerCells.indexOf(name);
+                if (i === -1)
+                    throw new Error(`Could not find header '${name}' in ${headerCells.join(", ")}`);
+                return i;
+            });
+            tableDataAsStrings = tableDataAsStrings.map((row) => columnIndices.map((index) => row[index]));
+        }
+    }
+    return toMarkupTableString(tableDataAsStrings, hasExpandableHeader);
 }
 exports.tableSnapshot = tableSnapshot;
-function toMarkupTableString({ tableRows, hasExpandableHeader, }) {
+function toMarkupTableString(tableRows, hasExpandableHeader) {
     // Find the largest width of each column to set a consistent width for each row
     const columnWidths = tableRows.reduce((acc, row) => {
         row.forEach((cell, columnIndex) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@homebound/beam",
-  "version": "2.321.1",
+  "version": "2.322.0",
   "author": "Homebound",
   "license": "MIT",
   "main": "dist/index.js",