diff --git a/packages/comment-to-assert/src/ast-utils.js b/packages/comment-to-assert/src/ast-utils.js index f5afa05..18265bc 100644 --- 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 index 19caa72..679055a 100644 --- 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