fix: support for dwaiting constructor calls.

rzvxa · Jan 1, 2024 · 3821135 · 3821135
1 parent a8f7638
commit 3821135
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/dwait.ts b/src/dwait.ts
@@ -28,7 +28,7 @@ function noop() {}
  * @internal
  */
 function dwaitInternal<T, Y>(
-  promise: Promise<T>,
+  promise: Promise<T> | T,
   lhs?: Box<Y>
 ): DeferredPromise<T> {
   const task = Promise.resolve(promise);
@@ -84,7 +84,7 @@ function dwaitInternal<T, Y>(
  *
  * @returns A {@link DeferredPromise} pointing to the target `Promise`
  */
-function dwait<T>(promise: Promise<T>): DeferredPromise<T> {
+function dwait<T>(promise: Promise<T> | T): DeferredPromise<T> {
   return dwaitInternal(promise);
 }
 

diff --git a/tests/dwait.spec.ts b/tests/dwait.spec.ts
@@ -104,4 +104,10 @@ describe("dwait Tests", () => {
     const dwaitPromise = dwait(resolveMock()).toPromise();
     await expect(dwaitPromise).resolves.toBe(OK);
   });
+  test("should be able to wrap a constructor successfully", async () => {
+    const dwaitPromise = dwait(new TestClass(OK, NUMBER, OK));
+    await expect(dwaitPromise.await).resolves.toEqual(
+      expect.objectContaining({ foo: OK + OK })
+    );
+  });
 });