Mark the direct callee as being a callee for all ApplyExpr not just CallExpr

vukrado · xedin · commit 5b98f8b967c1 · 2020-07-07T12:12:15.000-07:00
(cherry picked from commit 7c5b6d7)
diff --git a/lib/Sema/TypeCheckConstraints.cpp b/lib/Sema/TypeCheckConstraints.cpp
@@ -1161,7 +1161,7 @@ namespace {
       ExprStack.pop_back();
 
       // Mark the direct callee as being a callee.
-      if (auto *call = dyn_cast<CallExpr>(expr))
+      if (auto *call = dyn_cast<ApplyExpr>(expr))
         markDirectCallee(call->getFn());
 
       // Fold sequence expressions.
diff --git a/test/Constraints/operator.swift b/test/Constraints/operator.swift
@@ -290,3 +290,9 @@ func rdar_62054241() {
     return arr.sorted(by: <) // expected-error {{no exact matches in reference to operator function '<'}}
   }
 }
+
+// SR-11399 - Operator returning IUO doesn't implicitly unwrap
+postfix operator ^^^
+postfix func ^^^ (lhs: Int) -> Int! { 0 }
+
+let x: Int = 1^^^

Original file line number	Diff line number	Diff line change
`@@ -290,3 +290,9 @@ func rdar_62054241() {`
`290`	`290`	`return arr.sorted(by: <) // expected-error {{no exact matches in reference to operator function '<'}}`
`291`	`291`	`}`
`292`	`292`	`}`
	`293`	`+`
	`294`	`+// SR-11399 - Operator returning IUO doesn't implicitly unwrap`
	`295`	`+postfix operator ^^^`
	`296`	`+postfix func ^^^ (lhs: Int) -> Int! { 0 }`
	`297`	`+`
	`298`	`+let x: Int = 1^^^`