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

Fix broken class parser
TypeStrong · Sep 25, 2024 · b39a6a0 · b39a6a0
2 parents d815ba9 + d7c642f
commit b39a6a0
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 4 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@typestrong/ts-mockito",
-  "version": "2.7.11",
+  "version": "2.7.12",
   "description": "Mocking library for TypeScript",
   "main": "lib/ts-mockito.js",
   "typings": "lib/ts-mockito",

diff --git a/src/utils/ObjectPropertyCodeRetriever.ts b/src/utils/ObjectPropertyCodeRetriever.ts
@@ -17,7 +17,7 @@ export class ObjectPropertyCodeRetriever {
                         const gx = new RegExp(`^(async)?\\s{0,}\\*?${prop}`);
                         const isMethod = gx.test(fnStr);
                         return `
-                            ${isMethod ? fnStr : `${prop}: ${fnStr}`}
+                            ${isMethod ? fnStr : `"${prop}": ${fnStr}`}
                         `;
                     }
                     return '';

diff --git a/test/verification.spec.ts b/test/verification.spec.ts
@@ -837,6 +837,30 @@ cases.forEach(testData => {
                 expect(mocked).toBeDefined();
             });
         });
+
+        describe("method names with special chars", () => {
+            class TestClass {
+                "special_chars-@?" = () => {
+                    // do nothing
+                }
+            }
+
+            const TestObject = {
+                "special_chars-@?": () => {
+                    // do nothing
+                }
+            }
+
+            it("should mock class", () => {
+                const mocked = mock(TestClass);
+                expect(mocked).toBeDefined();
+            });
+
+            it("should mock object", () => {
+                const mocked = mock(TestObject);
+                expect(mocked).toBeDefined();
+            });
+        });
     });
 });