feature(estree-to-babel) add tokens support

coderaiser · Jan 16, 2019 · f4d033b · f4d033b
1 parent b7c1e64
commit f4d033b
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 8 deletions.
diff --git a/lib/estree-to-babel.js b/lib/estree-to-babel.js
@@ -16,13 +16,15 @@ const isNull = (a) => a === null;
 module.exports = (node) => {
     const {
         comments = [],
+        tokens,
         ...program
     } = node;
 
     const ast = {
         type: 'File',
         program,
         comments,
+        tokens,
     };
 
     traverse(ast, {

diff --git 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
@@ -220,4 +220,4 @@
         "sourceType": "script"
     },
     "comments": []
-}
+}