Propegate contextual types through nonnull assertion operators (#42526)

Author: weswigham

src/compiler/checker.ts

@@ -24757,6 +24757,8 @@ namespace ts {
                     const tag = isInJSFile(parent) ? getJSDocTypeTag(parent) : undefined;
                     return tag ? getTypeFromTypeNode(tag.typeExpression.type) : getContextualType(<ParenthesizedExpression>parent, contextFlags);
                 }
+                case SyntaxKind.NonNullExpression:
+                    return getContextualType(parent as NonNullExpression, contextFlags);
                 case SyntaxKind.JsxExpression:
                     return getContextualTypeForJsxExpression(<JsxExpression>parent);
                 case SyntaxKind.JsxAttribute:

tests/baselines/reference/nonnullAssertionPropegatesContextualType.js

@@ -0,0 +1,6 @@
+//// [nonnullAssertionPropegatesContextualType.ts]
+let rect2: SVGRectElement = document.querySelector('.svg-rectangle')!; // Error: Element
+
+//// [nonnullAssertionPropegatesContextualType.js]
+"use strict";
+var rect2 = document.querySelector('.svg-rectangle'); // Error: Element

tests/baselines/reference/nonnullAssertionPropegatesContextualType.symbols

@@ -0,0 +1,8 @@
+=== tests/cases/compiler/nonnullAssertionPropegatesContextualType.ts ===
+let rect2: SVGRectElement = document.querySelector('.svg-rectangle')!; // Error: Element
+>rect2 : Symbol(rect2, Decl(nonnullAssertionPropegatesContextualType.ts, 0, 3))
+>SVGRectElement : Symbol(SVGRectElement, Decl(lib.dom.d.ts, --, --), Decl(lib.dom.d.ts, --, --))
+>document.querySelector : Symbol(ParentNode.querySelector, Decl(lib.dom.d.ts, --, --), Decl(lib.dom.d.ts, --, --), Decl(lib.dom.d.ts, --, --))
+>document : Symbol(document, Decl(lib.dom.d.ts, --, --))
+>querySelector : Symbol(ParentNode.querySelector, Decl(lib.dom.d.ts, --, --), Decl(lib.dom.d.ts, --, --), Decl(lib.dom.d.ts, --, --))
+

tests/baselines/reference/nonnullAssertionPropegatesContextualType.types

@@ -0,0 +1,10 @@
+=== tests/cases/compiler/nonnullAssertionPropegatesContextualType.ts ===
+let rect2: SVGRectElement = document.querySelector('.svg-rectangle')!; // Error: Element
+>rect2 : SVGRectElement
+>document.querySelector('.svg-rectangle')! : SVGRectElement
+>document.querySelector('.svg-rectangle') : SVGRectElement | null
+>document.querySelector : { <K extends keyof HTMLElementTagNameMap>(selectors: K): HTMLElementTagNameMap[K] | null; <K extends keyof SVGElementTagNameMap>(selectors: K): SVGElementTagNameMap[K] | null; <E extends Element = Element>(selectors: string): E | null; }
+>document : Document
+>querySelector : { <K extends keyof HTMLElementTagNameMap>(selectors: K): HTMLElementTagNameMap[K] | null; <K extends keyof SVGElementTagNameMap>(selectors: K): SVGElementTagNameMap[K] | null; <E extends Element = Element>(selectors: string): E | null; }
+>'.svg-rectangle' : ".svg-rectangle"
+

tests/cases/compiler/nonnullAssertionPropegatesContextualType.ts

@@ -0,0 +1,2 @@
+// @strict: true
+let rect2: SVGRectElement = document.querySelector('.svg-rectangle')!; // Error: Element