Refactor more helpers

iterative · Jul 12, 2022 · 614ff4a · 614ff4a
1 parent 060dac2
commit 614ff4a
Showing 1 changed file with 8 additions and 18 deletions.
diff --git a/webview/src/test/tableDataFixture.ts b/webview/src/test/tableDataFixture.ts
@@ -41,23 +41,13 @@ export const transformRows = (
   } as TableData
 }
 
-export const setExperimentsAsStarred = (
-  fixture: TableData,
-  labelOrIds: string[]
-) => {
-  return transformRows(fixture, labelOrIds, row => ({ ...row, starred: true }))
-}
+export const setRowPropertyAsTrue =
+  (prop: keyof Row) => (fixture: TableData, labelOrIds: string[]) => {
+    return transformRows(fixture, labelOrIds, row => ({ ...row, [prop]: true }))
+  }
 
-export const setExperimentsAsSelected = (
-  fixture: TableData,
-  labelOrIds: string[]
-) => {
-  return transformRows(fixture, labelOrIds, row => ({ ...row, selected: true }))
-}
+export const setExperimentsAsStarred = setRowPropertyAsTrue('starred')
 
-export const setExperimentsAsRunning = (
-  fixture: TableData,
-  labelOrIds: string[]
-) => {
-  return transformRows(fixture, labelOrIds, row => ({ ...row, running: true }))
-}
+export const setExperimentsAsSelected = setRowPropertyAsTrue('selected')
+
+export const setExperimentsAsRunning = setRowPropertyAsTrue('running')