fix(utils): add missing extractionBody

azu · Aug 24, 2019 · 3bca03e · 3bca03e
1 parent 331e160
commit 3bca03e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/packages/comment-to-assert/src/ast-utils.js b/packages/comment-to-assert/src/ast-utils.js
@@ -18,6 +18,9 @@ export function tryGetCodeFromComments(comments) {
         return matchResult[1];
     }
 }
+function extractionBody(ast) {
+    return ast.body[0];
+}
 export function wrapAssert(actualNode, expectedNode) {
     assert(typeof expectedNode !== "undefined");
     var type = expectedNode.type || extractionBody(expectedNode).type;

diff --git a/packages/comment-to-assert/src/comment-to-assert.js b/packages/comment-to-assert/src/comment-to-assert.js
@@ -24,6 +24,7 @@ export function toAssertFromSource(code, filePath) {
     var output = source.transform(toAssertFromAST).output();
     return output.codeWithMap;
 }
+
 /**
  * transform AST to asserted AST.
  * @param {ESTree.Node} ast