From 3bca03e6fbe815fbf32b24e7b92815866fe16338 Mon Sep 17 00:00:00 2001
From: azu <azuciao@gmail.com>
Date: Sun, 9 Aug 2015 16:11:25 +0900
Subject: [PATCH] fix(utils): add missing extractionBody

---
 packages/comment-to-assert/src/ast-utils.js         | 3 +++
 packages/comment-to-assert/src/comment-to-assert.js | 1 +
 2 files changed, 4 insertions(+)

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