test: SURVEY-17302 Update test utilities so that they support pinned …

…columns
linz · Jun 28, 2023 · cb2a8db · cb2a8db
1 parent 7198bfb
commit cb2a8db
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/utils/testUtil.ts b/src/utils/testUtil.ts
@@ -12,7 +12,17 @@ export const findRow = async (rowId: number | string, within?: HTMLElement): Pro
   //if this is not wrapped in an act console errors are logged during testing
   let row!: HTMLDivElement;
   await act(async () => {
-    row = await findQuick<HTMLDivElement>({ tagName: `div[row-id='${rowId}']:not(:empty)` }, within);
+    row = await findQuick<HTMLDivElement>({ tagName: `.ag-center-cols-container div[row-id='${rowId}']:not(:empty)` }, within);
+    const leftCols = await findQuick<HTMLDivElement>(
+      { tagName: `.ag-pinned-left-cols-container div[row-id='${rowId}']` },
+      within,
+    );
+    const rightCols = await findQuick<HTMLDivElement>(
+      { tagName: `.ag-pinned-right-cols-container div[row-id='${rowId}']` },
+      within,
+    );
+    const combineChildren = [...leftCols.children, ...row.children, ...rightCols.children];
+    row.replaceChildren(...combineChildren);
   });
   return row;
 };