SONARPY-2961 Fix FP when the task is within a return statement (#287)

GitOrigin-RevId: f64bfddfa6414c46a1d54dab2d440f39a5f2e189

Author: guillaume-dequenne

python-checks/src/main/java/org/sonar/python/checks/AsyncioTaskNotStoredCheck.java

@@ -56,7 +56,7 @@ private void checkCallExpression(SubscriptionContext ctx) {
     if (asyncioTaskTypeChecks.stream().noneMatch(t -> t.check(calleeType) == TriBool.TRUE)) {
       return;
     }
-    if (TreeUtils.firstAncestorOfKind(callExpression, Tree.Kind.ASSIGNMENT_STMT, Tree.Kind.ASSIGNMENT_EXPRESSION, Tree.Kind.CALL_EXPR) == null) {
+    if (TreeUtils.firstAncestorOfKind(callExpression, Tree.Kind.ASSIGNMENT_STMT, Tree.Kind.ASSIGNMENT_EXPRESSION, Tree.Kind.CALL_EXPR, Tree.Kind.RETURN_STMT) == null) {
       ctx.addIssue(callee, MESSAGE);
     }
   }

python-checks/src/test/resources/checks/asyncioTaskNotStored.py

@@ -28,6 +28,9 @@ async def compliant_examples():
         asyncio.create_task(another_coroutine())
     )
 
+async def create_and_return_task():
+    return asyncio.create_task(some_coroutine())  # Compliant
+
 async def some_coroutine():
     await asyncio.sleep(1)
     return {"result": "value"}