fix(core): error on overloaded functions

fix #85
mockdeep · Dec 3, 2018 · b5339a6 · b5339a6
1 parent 45a85e5
commit b5339a6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/packages/typewiz-core/src/instrument.spec.ts b/packages/typewiz-core/src/instrument.spec.ts
@@ -45,6 +45,12 @@ describe('instrument', () => {
         );
     });
 
+    it('should not instrument a function declaration that has no a body', () => {
+        // See issue #85 for details
+        const input = `function foo(n: number): void;`;
+        expect(instrument(input, 'test.ts')).toContain(astPrettyPrint('function foo(n: number): void;'));
+    });
+
     describe('instrumentCallExpressions', () => {
         it('should instrument function calls', () => {
             const input = `foo(bar)`;

diff --git a/packages/typewiz-core/src/transformer.ts b/packages/typewiz-core/src/transformer.ts
@@ -166,7 +166,7 @@ function visitorFactory(
         }
 
         const isArrow = ts.isArrowFunction(node);
-        if (ts.isFunctionDeclaration(node) || ts.isMethodDeclaration(node) || ts.isArrowFunction(node)) {
+        if ((ts.isFunctionDeclaration(node) || ts.isMethodDeclaration(node) || ts.isArrowFunction(node)) && node.body) {
             const instrumentStatements: ts.ExpressionStatement[] = [];
             if (options.instrumentImplicitThis && needsThisInstrumentation(node, semanticDiagnostics)) {
                 const opts: IExtraOptions = { thisType: true };