K2: process also arguments while calculating postponed atoms

#KT-61933 Fixed

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/inference/CompletionModeCalculator.kt

@@ -57,7 +57,7 @@ private class CalculatorForNestedCall(
     private val typesToProcess: Queue<KotlinTypeMarker> = ArrayDeque()
 
     private val postponedAtoms: List<PostponedResolvedAtom> by lazy {
-        candidate.postponedAtoms.filterNot { it.analyzed }
+        ConstraintSystemCompleter.getOrderedNotAnalyzedPostponedArguments(candidate)
     }
 
     fun computeCompletionMode(): ConstraintSystemCompletionMode = with(context) {

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/inference/ConstraintSystemCompleter.kt

@@ -407,6 +407,11 @@ class ConstraintSystemCompleter(components: BodyResolveComponents, private val c
     }
 
     companion object {
+        internal fun getOrderedNotAnalyzedPostponedArguments(candidate: Candidate): List<PostponedResolvedAtom> {
+            val callSite = candidate.callInfo.callSite as? FirStatement ?: return emptyList()
+            return getOrderedNotAnalyzedPostponedArguments(listOf(callSite))
+        }
+
         private fun getOrderedNotAnalyzedPostponedArguments(topLevelAtoms: List<FirStatement>): List<PostponedResolvedAtom> {
             val notAnalyzedArguments = arrayListOf<PostponedResolvedAtom>()
             for (primitive in topLevelAtoms) {

compiler/testData/diagnostics/tests/suspendConversion/inGenericArgument.fir.kt

@@ -1,17 +0,0 @@
-// ISSUE: KT-61933
-
-data class Bar(
-    val foo: Foo<suspend () -> Unit>
-)
-
-data class Foo<out TCallback : Any>(
-    val state: TCallback?,
-)
-
-fun usage(b: Boolean) {
-    Bar(
-        foo = <!ARGUMENT_TYPE_MISMATCH!>Foo(
-            state = if (b) { -> } else null,
-        )<!>
-    )
-}

compiler/testData/diagnostics/tests/suspendConversion/inGenericArgument.kt

@@ -1,3 +1,4 @@
+// FIR_IDENTICAL
 // ISSUE: KT-61933
 
 data class Bar(