Ensure no unhandledrejection if canceled (#8)

snewcomer · Oct 28, 2021 · 20f47f4 · 20f47f4
1 parent 764f544
commit 20f47f4
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/addon/decorators/stateful-function.js b/addon/decorators/stateful-function.js
@@ -63,6 +63,13 @@ export function statefulFunction(target, _property, descriptor) {
 
     handler.isRunning = true;
 
+    sp.catch((e) => {
+      // ensure no unhandledrejection if canceled
+      if (!(e instanceof CanceledPromise)) {
+        throw e;
+      }
+    });
+
     return sp;
   };