KTIJ-27879 [AA] Enforce lazy resolve of code fragments in ContextCollector

Without it, the lazy body of code fragment is not properly resolved
and cannot be analyzed

Use the correct function from `BodyResolveContext` to take the scopes
from code fragment into account

^KTIJ-27879 Fixed

analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt

@@ -332,6 +332,14 @@ private class ContextCollectorVisitor(
         }
     }
 
+    override fun visitCodeFragment(codeFragment: FirCodeFragment) {
+        codeFragment.lazyResolveToPhase(FirResolvePhase.BODY_RESOLVE)
+
+        context.withCodeFragment(codeFragment, holder) {
+            super.visitCodeFragment(codeFragment)
+        }
+    }
+
     override fun visitAnnotationCall(annotationCall: FirAnnotationCall) {
         dumpContext(annotationCall, ContextKind.SELF)