improve done function

storybookjs · Oct 3, 2020 · 380726e · 380726e
1 parent e5c217b
commit 380726e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/addons/storyshots/storyshots-core/src/api/snapshotsTestsTemplate.ts b/addons/storyshots/storyshots-core/src/api/snapshotsTestsTemplate.ts
@@ -11,10 +11,9 @@ function snapshotTest({ item, asyncJest, framework, testMethod, testMethodParams
     it(
       name,
       () =>
-        new Promise((done, reject) =>
+        new Promise((resolve, reject) =>
           testMethod({
-            done,
-            reject,
+            done: (error: any) => (error ? reject(error) : resolve()),
             story: item,
             context,
             ...testMethodParams,