From d59b0129c612a7f84ef8e46f26ce342891838cea Mon Sep 17 00:00:00 2001
From: Max Schmitt <max@schmitt.mx>
Date: Thu, 26 May 2022 01:30:30 +0200
Subject: [PATCH] chore: fix coverage on WebKit

---
 src/PlaywrightEnvironment.ts | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/PlaywrightEnvironment.ts b/src/PlaywrightEnvironment.ts
index 3b81932..c6894e3 100644
--- a/src/PlaywrightEnvironment.ts
+++ b/src/PlaywrightEnvironment.ts
@@ -204,7 +204,12 @@ export const getPlaywrightEnv = (basicEnv = 'node'): unknown => {
     }
 
     async _setCollectCoverage(context: BrowserContext) {
-      await context.exposeFunction('reportCodeCoverage', saveCoverageToFile)
+      await context.exposeFunction(
+        'reportCodeCoverage',
+        (coverage: unknown) => {
+          if (coverage) saveCoverageToFile(coverage)
+        },
+      )
       await context.addInitScript(() =>
         window.addEventListener('beforeunload', () => {
           // @ts-ignore