Merge branch 'master' into fix-version-cli

sokra · web-flow · commit aed548b14b71 · 2016-10-31T09:22:38.000+01:00
diff --git a/lib/Parser.js b/lib/Parser.js
@@ -650,6 +650,12 @@ Parser.prototype.walkExpression = function walkExpression(expression) {
 		return this["walk" + expression.type](expression);
 };
 
+Parser.prototype.walkAwaitExpression = function walkAwaitExpression(expression) {
+	var argument = expression.argument
+	if(this["walk" + argument.type])
+		return this["walk" + argument.type](argument);
+}
+
 Parser.prototype.walkArrayExpression = function walkArrayExpression(expression) {
 	if(expression.elements)
 		this.walkExpressions(expression.elements);
@@ -1035,12 +1041,12 @@ Parser.prototype.parseCalculatedString = function parseCalculatedString(expressi
 var POSSIBLE_AST_OPTIONS = [{
 	ranges: true,
 	locations: true,
-	ecmaVersion: 6,
+	ecmaVersion: 2017,
 	sourceType: "module"
 }, {
 	ranges: true,
 	locations: true,
-	ecmaVersion: 6,
+	ecmaVersion: 2017,
 	sourceType: "script"
 }]
 
@@ -1062,7 +1068,7 @@ Parser.prototype.parse = function parse(source, initialState) {
 		ast = acorn.parse(source, {
 			ranges: true,
 			locations: true,
-			ecmaVersion: 6,
+			ecmaVersion: 2017,
 			sourceType: "module",
 			onComment: comments
 		});
@@ -1088,7 +1094,7 @@ Parser.prototype.evaluate = function evaluate(source) {
 	var ast = acorn.parse("(" + source + ")", {
 		ranges: true,
 		locations: true,
-		ecmaVersion: 6,
+		ecmaVersion: 2017,
 		sourceType: "module"
 	});
 	if(!ast || typeof ast !== "object" || ast.type !== "Program")
diff --git a/test/Parser.test.js b/test/Parser.test.js
@@ -307,4 +307,53 @@ describe("Parser", function() {
 			});
 		});
 	});
+
+	describe("async/await support", function() {
+		describe("should accept", function() {
+			var cases = {
+				"async function": "async function x() {}",
+				"async arrow function": "async () => {}",
+				"await expression": "async function x(y) { await y }"
+			};
+			var parser = new Parser();
+			Object.keys(cases).forEach(function(name) {
+				var expr = cases[name];
+				it(name, function() {
+					var actual = parser.parse(expr);
+					should.strictEqual(typeof actual, "object");
+				});
+			});
+		});
+		describe("should parse await", function() {
+			var cases = {
+				"require": [
+					"async function x() { await require('y'); }", {
+						param: "y"
+					}
+				],
+				"System.import": [
+					"async function x() { var y = await System.import('z'); }", {
+						param: "z"
+					}
+				]
+			};
+
+			var parser = new Parser();
+			parser.plugin("call require", function(expr) {
+				var param = this.evaluateExpression(expr.arguments[0]);
+				this.state.param = param.string;
+			});
+			parser.plugin("call System.import", function(expr) {
+				var param = this.evaluateExpression(expr.arguments[0]);
+				this.state.param = param.string;
+			});
+
+			Object.keys(cases).forEach(function(name) {
+				it(name, function() {
+					var actual = parser.parse(cases[name][0]);
+					actual.should.be.eql(cases[name][1]);
+				});
+			});
+		});
+	})
 });
