Add Deferred tests

dpwatrous · dpwatrous · commit 869e9725575a · 2023-09-22T11:59:05.000-04:00
diff --git a/packages/bonito-core/src/util/__tests__/deferred.spec.ts b/packages/bonito-core/src/util/__tests__/deferred.spec.ts
@@ -0,0 +1,45 @@
+import { Deferred } from "../deferred";
+
+describe("Deferred tests", () => {
+    test("resolve", () => {
+        const deferred = new Deferred<string>();
+
+        expect(deferred.resolve).toBeDefined();
+        expect(deferred.reject).toBeDefined();
+        expect(deferred.done).toBe(false);
+        expect(deferred.resolved).toBe(false);
+        expect(deferred.rejected).toBe(false);
+        expect(deferred.resolvedTo).toBeUndefined();
+        expect(deferred.rejectedTo).toBeUndefined();
+        expect(deferred.promise).toBeDefined();
+
+        deferred.resolve("Did it!");
+
+        expect(deferred.done).toBe(true);
+        expect(deferred.resolved).toBe(true);
+        expect(deferred.rejected).toBe(false);
+        expect(deferred.resolvedTo).toBe("Did it!");
+        expect(deferred.rejectedTo).toBeUndefined();
+    });
+
+    test("reject", async () => {
+        const deferred = new Deferred<string>();
+
+        deferred.reject("Didn't do it :(");
+
+        let caughtRejection = false;
+        try {
+            await deferred.promise;
+        } catch (e) {
+            caughtRejection = true;
+        }
+
+        expect(caughtRejection).toBe(true);
+
+        expect(deferred.done).toBe(true);
+        expect(deferred.resolved).toBe(false);
+        expect(deferred.rejected).toBe(true);
+        expect(deferred.rejectedTo).toBe("Didn't do it :(");
+        expect(deferred.resolvedTo).toBeUndefined();
+    });
+});
diff --git a/packages/bonito-core/src/util/deferred.ts b/packages/bonito-core/src/util/deferred.ts
@@ -3,19 +3,29 @@
  * at a later point in time with a function call.
  */
 export class Deferred<T = void> {
-    public promise: Promise<T>;
-    public done = false;
-    public resolve!: (result: T) => void;
-    public reject!: (error: unknown) => void;
+    promise: Promise<T>;
+
+    done = false;
+    resolved = false;
+    rejected = false;
+    resolvedTo?: T;
+    rejectedTo?: unknown;
+
+    resolve!: (result: T) => void;
+    reject!: (error: unknown) => void;
 
     constructor() {
         this.promise = new Promise<T>((resolve, reject) => {
             this.reject = (e) => {
                 this.done = true;
+                this.rejected = true;
+                this.rejectedTo = e;
                 reject(e);
             };
             this.resolve = (r: T) => {
                 this.done = true;
+                this.resolved = true;
+                this.resolvedTo = r;
                 resolve(r);
             };
         });
