feat: allow the runner to run beforeAll/afterAll hooks (#360)

It's useful if you want to combine these hooks with custom runStep invocations.
puppeteer · Nov 2, 2022 · 3c4ece0 · 3c4ece0
1 parent 2dba511
commit 3c4ece0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/Runner.ts b/src/Runner.ts
@@ -48,6 +48,14 @@ export class Runner {
     this.#flow = flow;
   }
 
+  async runBeforeAllSteps(flow?: UserFlow): Promise<void> {
+    await this.#extension.beforeAllSteps?.(flow);
+  }
+
+  async runAfterAllSteps(flow?: UserFlow): Promise<void> {
+    await this.#extension.afterAllSteps?.(flow);
+  }
+
   /**
    * Runs the provided `step` with `beforeEachStep` and `afterEachStep` hooks.
    * Parameters from the `flow` apply if the `flow` is set.

diff --git a/test/runner.test.ts b/test/runner.test.ts
@@ -817,6 +817,12 @@ describe('Runner', () => {
       extension.getLog(),
       'beforeAll,beforeStep0,runStep0,afterStep0,afterAll'
     );
+    await runner.runBeforeAllSteps();
+    await runner.runAfterAllSteps();
+    assert.strictEqual(
+      extension.getLog(),
+      'beforeAll,beforeStep0,runStep0,afterStep0,afterAll,beforeAll,afterAll'
+    );
   });
 
   it('should record interactions with OOPIFs', async () => {