Add ability to record custom actions by WaitMiddleware

packages/redux/src/middleware/WaitMiddleware.test.ts

@@ -110,7 +110,7 @@ describe('WaitMiddleware', () => {
 
   describe('record', () => {
 
-    it('records pending actions', async () => {
+    it('records pending actions by default', async () => {
       const wm = new WaitMiddleware()
       const store = getStore(wm)
       const recorder = wm.record()
@@ -127,6 +127,16 @@ describe('WaitMiddleware', () => {
       })
       await promise
     })
+
+    it('records custom actions', async () => {
+      const wm = new WaitMiddleware()
+      const store = getStore(wm)
+      const recorder = wm.record(action => action.type.startsWith('test'))
+      store.dispatch({type: 'test1'} as any)
+      store.dispatch({type: 'tes'} as any)
+      store.dispatch({type: 'test3'} as any)
+      expect(recorder.getActionTypes()).toEqual(['test1', 'test3'])
+    })
   })
 
 })

packages/redux/src/middleware/WaitMiddleware.ts

@@ -14,10 +14,10 @@ export class WaitMiddleware {
   }
 
   /**
-   * Starts recording pending actions and returns the recorder.
+   * Starts recording actions and returns the recorder.
    */
-  record(): Recorder {
+  record(shouldRecord: ShouldRecord = defaultShouldRecord): Recorder {
-    const recorder = new Recorder()
+    const recorder = new Recorder(shouldRecord)
     this.recorders.push(recorder)
     return recorder
   }
@@ -91,15 +91,25 @@ export class WaitMiddleware {
   }
 }
 
+/**
+ * Returns true when an action should be recorded, false otherwise
+ */
+export type ShouldRecord = (action: AnyAction) => boolean
+
+export const defaultShouldRecord: ShouldRecord =
+  (action: AnyAction) => 'status' in action && action.status === 'pending'
+
 class Recorder {
   protected actionTypes: string[] = []
 
+  constructor(protected readonly shouldRecord: ShouldRecord) {}
+
   getActionTypes(): string[] {
     return this.actionTypes.slice()
   }
 
   record(action: AnyAction) {
-    if ('status' in action && action.status === 'pending') {
+    if (this.shouldRecord(action)) {
       this.actionTypes.push(action.type)
     }
   }