feat: add Promise.prototype.toTask

tdreyno · Jan 8, 2021 · 4b4fe1b · 4b4fe1b
1 parent 7a1a7ae
commit 4b4fe1b
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/src/Task/Task.ts b/src/Task/Task.ts
@@ -989,3 +989,13 @@ export class ExternalTask<E, S> extends Task<E, S> {
     }
   }
 }
+
+declare global {
+  interface Promise<T> {
+    toTask(): Task<unknown, T>
+  }
+}
+
+Promise.prototype.toTask = function <S>(this: Promise<S>): Task<unknown, S> {
+  return fromPromise(this)
+}
diff --git a/src/Task/__tests__/fromPromise.spec.ts b/src/Task/__tests__/fromPromise.spec.ts
@@ -57,4 +57,18 @@ describe("fromPromise", () => {
     expect(reject).toBeCalledWith(ERROR_RESULT)
     expect(resolve).not.toBeCalled()
   })
+
+  test("should succeed using prototype helper", async () => {
+    const resolve = jest.fn()
+    const reject = jest.fn()
+
+    const promise = Promise.resolve(SUCCESS_RESULT)
+
+    promise.toTask().fork(reject, resolve)
+
+    await promise.catch(() => void 0)
+
+    expect(resolve).toBeCalledWith(SUCCESS_RESULT)
+    expect(reject).not.toBeCalled()
+  })
 })