From f4d033b9cad6ff7b82d4dafff35cfbc41cf2ceb1 Mon Sep 17 00:00:00 2001
From: coderaiser <coderaiser@cloudcmd.io>
Date: Wed, 16 Jan 2019 13:02:20 +0200
Subject: [PATCH] feature(estree-to-babel) add tokens support

---
 lib/estree-to-babel.js         |  2 ++
 test/estree-to-babel.js        | 14 +++++++-------
 test/fixture/class-method.json |  2 +-
 3 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/lib/estree-to-babel.js b/lib/estree-to-babel.js
index 71bce34..3484fe0 100644
--- a/lib/estree-to-babel.js
+++ b/lib/estree-to-babel.js
@@ -16,6 +16,7 @@ const isNull = (a) => a === null;
 module.exports = (node) => {
     const {
         comments = [],
+        tokens,
         ...program
     } = node;
     
@@ -23,6 +24,7 @@ module.exports = (node) => {
         type: 'File',
         program,
         comments,
+        tokens,
     };
     
     traverse(ast, {
diff --git a/test/estree-to-babel.js b/test/estree-to-babel.js
index d41a16b..83fb981 100644
--- a/test/estree-to-babel.js
+++ b/test/estree-to-babel.js
@@ -58,7 +58,7 @@ test('estree-to-babel: property', (t) => {
     
     update('property', result);
     
-    t.deepEqual(result, fixture.ast.property, 'should equal');
+    t.jsonEqual(result, fixture.ast.property, 'should equal');
     t.end();
 });
 
@@ -68,7 +68,7 @@ test('estree-to-babel: object-method', (t) => {
     
     update('object-method', result);
     
-    t.deepEqual(result, fixture.ast.objectMethod, 'should equal');
+    t.jsonEqual(result, fixture.ast.objectMethod, 'should equal');
     t.end();
 });
 
@@ -78,7 +78,7 @@ test('estree-to-babel: string-literal', (t) => {
     
     update('string-literal', result);
     
-    t.deepEqual(result, fixture.ast.stringLiteral, 'should equal');
+    t.jsonEqual(result, fixture.ast.stringLiteral, 'should equal');
     t.end();
 });
 
@@ -88,7 +88,7 @@ test('estree-to-babel: numeric-literal', (t) => {
     
     update('numeric-literal', result);
     
-    t.deepEqual(result, fixture.ast.numericLiteral, 'should equal');
+    t.jsonEqual(result, fixture.ast.numericLiteral, 'should equal');
     t.end();
 });
 
@@ -98,7 +98,7 @@ test('estree-to-babel: null literal', (t) => {
     
     update('null-literal', result);
     
-    t.deepEqual(result, fixture.ast.nullLiteral, 'should equal');
+    t.jsonEqual(result, fixture.ast.nullLiteral, 'should equal');
     t.end();
 });
 
@@ -108,7 +108,7 @@ test('estree-to-babel: comments', (t) => {
     
     update('comments', result);
     
-    t.deepEqual(result, fixture.ast.comments, 'should equal');
+    t.jsonEqual(result, fixture.ast.comments, 'should equal');
     t.end();
 });
 
@@ -118,7 +118,7 @@ test('estree-to-babel: class method', (t) => {
     
     update('class-method', result);
     
-    t.deepEqual(result, fixture.ast.classMethod, 'should equal');
+    t.jsonEqual(result, fixture.ast.classMethod, 'should equal');
     t.end();
 });
 
diff --git a/test/fixture/class-method.json b/test/fixture/class-method.json
index 0e7bd4b..dc4f921 100644
--- a/test/fixture/class-method.json
+++ b/test/fixture/class-method.json
@@ -220,4 +220,4 @@
         "sourceType": "script"
     },
     "comments": []
-}
+}
\ No newline at end of file