Fix await code action crashes when await is not in a function (microsoft#22623)

mjbvz · Andy · commit 9774cd5eb573 · 2018-03-16T11:53:17.000-07:00
Fixes microsoft#22614
diff --git a/src/services/codefixes/fixAwaitInSyncFunction.ts b/src/services/codefixes/fixAwaitInSyncFunction.ts
@@ -36,6 +36,10 @@ namespace ts.codefix {
     function getNodes(sourceFile: SourceFile, start: number): { insertBefore: Node, returnType: TypeNode | undefined } | undefined {
         const token = getTokenAtPosition(sourceFile, start, /*includeJsDocComment*/ false);
         const containingFunction = getContainingFunction(token);
+        if (!containingFunction) {
+            return;
+        }
+
         let insertBefore: Node | undefined;
         switch (containingFunction.kind) {
             case SyntaxKind.MethodDeclaration:
diff --git a/tests/cases/fourslash/codeFixAwaitShouldNotCrashIfNotInFunction.ts b/tests/cases/fourslash/codeFixAwaitShouldNotCrashIfNotInFunction.ts
@@ -0,0 +1,5 @@
+/// <reference path='fourslash.ts' />
+
+////await a
+
+verify.not.codeFixAvailable();
