fixed async functions

TypeStrong · Jul 24, 2024 · ed1c6b0 · ed1c6b0
1 parent f2abe9d
commit ed1c6b0
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 10 deletions.
diff --git a/src/utils/ObjectPropertyCodeRetriever.ts b/src/utils/ObjectPropertyCodeRetriever.ts
@@ -13,7 +13,7 @@ export class ObjectPropertyCodeRetriever {
                         ];
                     } else if (typeof object[prop] === 'function') {
                         const fnStr = String(object[prop]);
-                        const isMethod = fnStr.startsWith(prop);
+                        const isMethod = fnStr.startsWith(prop) || fnStr.startsWith(`async ${prop}`);
                         return `
                             ${isMethod ? fnStr : `${prop}: ${fnStr}`}
                         `;

diff --git a/test/mocking.types.spec.ts b/test/mocking.types.spec.ts
@@ -1,8 +1,8 @@
-import { MethodToStub } from "../src/MethodToStub";
-import { instance, mock, when } from "../src/ts-mockito";
-import { Bar } from "./utils/Bar";
-import { ThenableClass } from "./utils/Thenable";
-import { EventEmitter } from "events";
+import {MethodToStub} from "../src/MethodToStub";
+import {instance, mock, when} from "../src/ts-mockito";
+import {Bar} from "./utils/Bar";
+import {ThenableClass} from "./utils/Thenable";
+import {EventEmitter} from "events";
 
 describe("mocking", () => {
     describe("mocking abstract class", () => {
@@ -228,19 +228,26 @@ describe("mocking", () => {
         });
     });
 
-    describe("mocking native class", () =>{
+    describe("mocking native class", () => {
         it("should mock", () => {
             const mocked = mock(TestEmitter);
             expect(mocked).toBeDefined();
         });
     });
 
-    describe("mocking anon class", () =>{
+    describe("mocking anon class", () => {
         it("should mock", () => {
             const mocked = mock(TestAnonClass);
             expect(mocked).toBeDefined();
         });
     });
+
+    describe("mocking async class", () => {
+        it("should mock", () => {
+            const mocked = mock(AsyncClass);
+            expect(mocked).toBeDefined();
+        });
+    });
 });
 
 abstract class SampleAbstractClass {
@@ -307,8 +314,21 @@ class SampleGeneric<T> {
     }
 }
 
-class TestEmitter extends EventEmitter {}
+class TestEmitter extends EventEmitter {
+}
 
 const TestAnonClass = class {
     private readonly foo = 'abc';
-};
+};
+
+export class AsyncClass {
+    public asyncValueArrowFn = async () => 'value';
+
+    public asyncValueFn = async function hello() {
+        return 'value';
+    };
+
+    public async returnAsyncValue(): Promise<number> {
+        return 0;
+    }
+}