Merge pull request #54 from roypeled/fix-broken-class-parser

Fix broken class parser

Author: roypeled

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "@typestrong/ts-mockito",
-  "version": "2.7.5",
+  "version": "2.7.6",
   "description": "Mocking library for TypeScript",
   "main": "lib/ts-mockito.js",
   "typings": "lib/ts-mockito",

package.json

@@ -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}`}
                         `;

src/utils/ObjectPropertyCodeRetriever.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;
+    }
+}