Added support for ChainExpression node (#113)

Co-authored-by: sanex3339 <yarabotayuvyandex3339>
estools · Aug 5, 2020 · ac91373 · ac91373
1 parent e22afb3
commit ac91373
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 0 deletions.
diff --git a/estraverse.js b/estraverse.js
@@ -84,6 +84,7 @@
         BreakStatement: 'BreakStatement',
         CallExpression: 'CallExpression',
         CatchClause: 'CatchClause',
+        ChainExpression: 'ChainExpression',
         ClassBody: 'ClassBody',
         ClassDeclaration: 'ClassDeclaration',
         ClassExpression: 'ClassExpression',
@@ -159,6 +160,7 @@
         BreakStatement: ['label'],
         CallExpression: ['callee', 'arguments'],
         CatchClause: ['param', 'body'],
+        ChainExpression: ['expression'],
         ClassBody: ['body'],
         ClassDeclaration: ['id', 'superClass', 'body'],
         ClassExpression: ['id', 'superClass', 'body'],

diff --git a/test/traverse.js b/test/traverse.js
@@ -402,6 +402,43 @@ describe('function declaration', function() {
     });
 });
 
+describe('chain expression', function() {
+    it('traverse', function() {
+        const tree = {
+            type: 'ExpressionStatement',
+            expression: {
+                type: 'ChainExpression',
+                expression: {
+                    type: 'MemberExpression',
+                    object: {
+                        type: 'Identifier',
+                        name: 'a'
+                    },
+                    property: {
+                        type: 'Identifier',
+                        name: 'b'
+                    },
+                    computed: false,
+                    optional: true
+                }
+            }
+        };
+
+        checkDump(Dumper.dump(tree), `
+            enter - ExpressionStatement
+            enter - ChainExpression
+            enter - MemberExpression
+            enter - Identifier
+            leave - Identifier
+            enter - Identifier
+            leave - Identifier
+            leave - MemberExpression
+            leave - ChainExpression
+            leave - ExpressionStatement
+        `);
+    });
+});
+
 describe('extending keys', function() {
     it('traverse', function() {
         const tree = {