PY-20682 Fixed: await something() as function argument is not recognized by syntax analyzer

Copy `async` flag in ParsingScope.withSuite

python/testData/psi/AwaitOnTheSameLineAsFor.py

@@ -0,0 +1,2 @@
+async def y(a1, a2):
+    for i in range(10): a1.foo(await a1.x(a2))

python/testData/psi/AwaitOnTheSameLineAsFor.txt

@@ -0,0 +1,63 @@
+PyFile:AwaitOnTheSameLineAsFor.py
+  PyFunction('y')
+    PsiElement(Py:ASYNC_KEYWORD)('async')
+    PsiWhiteSpace(' ')
+    PsiElement(Py:DEF_KEYWORD)('def')
+    PsiWhiteSpace(' ')
+    PsiElement(Py:IDENTIFIER)('y')
+    PyParameterList
+      PsiElement(Py:LPAR)('(')
+      PyNamedParameter('a1')
+        PsiElement(Py:IDENTIFIER)('a1')
+      PsiElement(Py:COMMA)(',')
+      PsiWhiteSpace(' ')
+      PyNamedParameter('a2')
+        PsiElement(Py:IDENTIFIER)('a2')
+      PsiElement(Py:RPAR)(')')
+    PsiElement(Py:COLON)(':')
+    PsiWhiteSpace('\n    ')
+    PyStatementList
+      PyForStatement
+        PyForPart
+          PsiElement(Py:FOR_KEYWORD)('for')
+          PsiWhiteSpace(' ')
+          PyTargetExpression: i
+            PsiElement(Py:IDENTIFIER)('i')
+          PsiWhiteSpace(' ')
+          PsiElement(Py:IN_KEYWORD)('in')
+          PsiWhiteSpace(' ')
+          PyCallExpression: range
+            PyReferenceExpression: range
+              PsiElement(Py:IDENTIFIER)('range')
+            PyArgumentList
+              PsiElement(Py:LPAR)('(')
+              PyNumericLiteralExpression
+                PsiElement(Py:INTEGER_LITERAL)('10')
+              PsiElement(Py:RPAR)(')')
+          PsiElement(Py:COLON)(':')
+          PsiWhiteSpace(' ')
+          PyStatementList
+            PyExpressionStatement
+              PyCallExpression: a1.foo
+                PyReferenceExpression: foo
+                  PyReferenceExpression: a1
+                    PsiElement(Py:IDENTIFIER)('a1')
+                  PsiElement(Py:DOT)('.')
+                  PsiElement(Py:IDENTIFIER)('foo')
+                PyArgumentList
+                  PsiElement(Py:LPAR)('(')
+                  PyPrefixExpression
+                    PsiElement(Py:AWAIT_KEYWORD)('await')
+                    PsiWhiteSpace(' ')
+                    PyCallExpression: a1.x
+                      PyReferenceExpression: x
+                        PyReferenceExpression: a1
+                          PsiElement(Py:IDENTIFIER)('a1')
+                        PsiElement(Py:DOT)('.')
+                        PsiElement(Py:IDENTIFIER)('x')
+                      PyArgumentList
+                        PsiElement(Py:LPAR)('(')
+                        PyReferenceExpression: a2
+                          PsiElement(Py:IDENTIFIER)('a2')
+                        PsiElement(Py:RPAR)(')')
+                  PsiElement(Py:RPAR)(')')