Add promise utils

src/utils/promise.ts

@@ -0,0 +1,34 @@
+export interface ControlledPromise<T = any> extends Promise<T> {
+  resolve: (value: T) => void;
+  reject: (reason?: unknown) => void;
+}
+
+export function controlledPromise<T = any>(): ControlledPromise<T> {
+  let resolve;
+  let reject;
+
+  const promise = new Promise((resolvePromise, rejectPromise) => {
+    resolve = resolvePromise;
+    reject = rejectPromise;
+  }) as ControlledPromise<T>;
+
+  promise.resolve = resolve as any as ControlledPromise["resolve"];
+  promise.reject = reject as any as ControlledPromise["reject"];
+
+  return promise;
+}
+
+export function signalledPromise<T>(signal: AbortSignal, promise: Promise<T>) {
+  return Promise.race([
+    new Promise<never>((_resolve, reject) => {
+      signal.addEventListener("abort", () => {
+        reject(signal.reason);
+      });
+    }),
+    promise,
+  ]);
+}
+
+export function timeoutPromise(ms: number) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}

tests/src/task.test.ts

@@ -4,7 +4,7 @@ import {
   TaskDroppedError,
   TaskStatus,
 } from "src/task";
-import { createControlledPromise } from "src/utils/promise";
+import { controlledPromise } from "src/utils/promise";
 import { describe, test, expect } from "vitest";
 
 describe("Task", () => {
@@ -53,7 +53,7 @@ describe("Task", () => {
 
   describe("#cancel", async () => {
     test("cancelling pending task", async () => {
-      const promise = createControlledPromise();
+      const promise = controlledPromise();
       const task = createTask(() => promise);
 
       expect(task.status).toBe(TaskStatus.Idle);
@@ -78,7 +78,7 @@ describe("Task", () => {
     });
 
     test("cancelling idle task", async () => {
-      const promise = createControlledPromise();
+      const promise = controlledPromise();
       const task = createTask(() => promise);
 
       expect(task.status).toBe(TaskStatus.Idle);