Return the result of the callback in the promise

TheBrainFamily · Dec 16, 2021 · 2e7924e · 2e7924e
1 parent 6be6e2e
commit 2e7924e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/index.d.ts b/src/index.d.ts
@@ -6,4 +6,4 @@
  * @param  interval  Number  Wait-between-retries interval, 50ms by default
  * @return  Promise  Promise to return a callback result
  */
-export default function waitForExpect(expectation: () => void | Promise<void>, timeout?: number, interval?: number): any;
+export default function waitForExpect<T>(expectation: () => T | Promise<T>, timeout?: number, interval?: number): Promise<T>;
diff --git a/src/index.ts b/src/index.ts
@@ -13,8 +13,8 @@ const defaults = {
  * @param  interval  Number  Wait-between-retries interval, 50ms by default
  * @return  Promise  Promise to return a callback result
  */
-const waitForExpect = function waitForExpect(
-  expectation: () => void | Promise<void>,
+const waitForExpect = function waitForExpect<T>(
+  expectation: () => T | Promise<T>,
   timeout = defaults.timeout,
   interval = defaults.interval
 ) {
@@ -37,7 +37,7 @@ const waitForExpect = function waitForExpect(
       tries += 1;
       try {
         Promise.resolve(expectation())
-          .then(() => resolve())
+          .then(result => resolve(result))
           .catch(rejectOrRerun);
       } catch (error) {
         rejectOrRerun(error);