Fix unintended early return in await inclusion handling. (#2899)

* Fix unintended early return in await inclusion handling.

* Move to form test

src/ast/nodes/AwaitExpression.ts

@@ -15,10 +15,11 @@ export default class AwaitExpression extends NodeBase {
 	}
 
 	include(includeChildrenRecursively: IncludeChildren) {
-		if (!this.included && !this.context.usesTopLevelAwait) {
+		checkTopLevelAwait: if (!this.included && !this.context.usesTopLevelAwait) {
 			let parent = this.parent;
 			do {
-				if (parent instanceof FunctionNode || parent instanceof ArrowFunctionExpression) return;
+				if (parent instanceof FunctionNode || parent instanceof ArrowFunctionExpression)
+					break checkTopLevelAwait;
 			} while ((parent = (parent as Node).parent as Node));
 			this.context.usesTopLevelAwait = true;
 		}

test/form/samples/handles-async-await/_config.js

@@ -0,0 +1,3 @@
+module.exports = {
+	description: 'properly handles exporting async functions'
+};

test/form/samples/handles-async-await/_expected.js

@@ -0,0 +1,6 @@
+async function callback() {
+	const response = await Promise.resolve(43);
+	return response - 1;
+}
+
+export { callback };

test/form/samples/handles-async-await/main.js

@@ -0,0 +1,4 @@
+export async function callback() {
+	const response = await Promise.resolve(43);
+	return response - 1;
+}