Fix NonNullAssert convert in member chain

Author: fisker

src/transform-node.ts

@@ -22,6 +22,10 @@ function getOuterEnd(node: NGNode): number {
 }
 
 function isOptionalObjectOrCallee(node: NGNode): boolean {
+  if (node.type === 'TSNonNullExpression') {
+    return isOptionalObjectOrCallee(node.expression);
+  }
+
   return (
     (node.type === 'OptionalCallExpression' ||
       node.type === 'OptionalMemberExpression') &&

tests/transform.test.ts

@@ -88,6 +88,8 @@ describe.each`
   ${'PropertyRead'}          | ${'MemberExpression'}         | ${' this . a '}                                 | ${true}     | ${true}      | ${true}            | ${true}
   ${'PropertyRead'}          | ${'OptionalMemberExpression'} | ${' a ?. b . c '}                               | ${true}     | ${true}      | ${true}            | ${true}
   ${'PropertyRead'}          | ${'OptionalMemberExpression'} | ${' a ?. b ( ) . c '}                           | ${true}     | ${true}      | ${true}            | ${true}
+  ${'PropertyRead'}          | ${'OptionalMemberExpression'} | ${' foo?.bar!.bam '}                            | ${true}     | ${true}      | ${true}            | ${true}
+  ${'PropertyRead'}          | ${'MemberExpression'}         | ${' (foo?.bar)!.bam '}                          | ${true}     | ${true}      | ${true}            | ${true}
   ${'Binary'}                | ${'AssignmentExpression'}     | ${' a . b = 1 '}                                | ${true}     | ${false}     | ${false}           | ${false}
   ${'Binary'}                | ${'AssignmentExpression'}     | ${' a = 1 '}                                    | ${true}     | ${false}     | ${false}           | ${false}
   ${'Call'}                  | ${'OptionalCallExpression'}   | ${' a ?. b ( ) '}                               | ${true}     | ${true}      | ${true}            | ${true}