FIR IDE: fix not initialized containingClassAttr for copied declaration

compiler/fir/raw-fir/psi2fir/src/org/jetbrains/kotlin/fir/builder/RawFirBuilder.kt

@@ -64,23 +64,32 @@ class RawFirBuilder(
         return reference.accept(Visitor(), Unit) as FirTypeRef
     }
 
-    fun buildFunctionWithBody(function: KtNamedFunction): FirFunction<*> {
-        return buildDeclaration(function) as FirFunction<*>
+    fun buildFunctionWithBody(function: KtNamedFunction, original: FirFunction<*>?): FirFunction<*> {
+        return buildDeclaration(function, original) as FirFunction<*>
     }
 
-    fun buildSecondaryConstructor(secondaryConstructor: KtSecondaryConstructor): FirConstructor {
-        return buildDeclaration(secondaryConstructor) as FirConstructor
+    fun buildSecondaryConstructor(secondaryConstructor: KtSecondaryConstructor, original: FirConstructor?): FirConstructor {
+        return buildDeclaration(secondaryConstructor, original) as FirConstructor
     }
 
-    fun buildPropertyWithBody(property: KtProperty): FirProperty {
+    fun buildPropertyWithBody(property: KtProperty, original: FirProperty?): FirProperty {
         require(!property.isLocal) { "Should not be used to build local properties (variables)" }
-        return buildDeclaration(property) as FirProperty
+        return buildDeclaration(property, original) as FirProperty
     }
 
-    private fun buildDeclaration(declaration: KtDeclaration): FirDeclaration {
+    private fun buildDeclaration(declaration: KtDeclaration, original: FirDeclaration?): FirDeclaration {
         assert(mode ==  RawFirBuilderMode.NORMAL) { "Building FIR declarations isn't supported in stub or lazy mode mode" }
-        setupContextForPosition(declaration)
-        return declaration.accept(Visitor(), Unit) as FirDeclaration
+        setupContextForPosition(declaration,)
+        val firDeclaration = declaration.accept(Visitor(), Unit) as FirDeclaration
+        original?.let { firDeclaration.copyContainingClassAttrFrom(it) }
+        return firDeclaration
+    }
+
+    // TODO this is a (temporary) hack, instead we should properly initialize [context]
+    private fun FirDeclaration.copyContainingClassAttrFrom(from: FirDeclaration) {
+        (this as? FirCallableMemberDeclaration<*>)?.let {
+            it.containingClassAttr = (from as? FirCallableMemberDeclaration<*>)?.containingClassAttr
+        }
     }
 
     private fun setupContextForPosition(position: KtElement) {

compiler/fir/raw-fir/psi2fir/tests/org/jetbrains/kotlin/fir/builder/AbstractPartialRawFirBuilderTestCase.kt

@@ -5,7 +5,6 @@
 
 package org.jetbrains.kotlin.fir.builder
 
-import junit.framework.TestCase
 import org.jetbrains.kotlin.fir.FirElement
 import org.jetbrains.kotlin.fir.FirRenderer
 import org.jetbrains.kotlin.fir.render
@@ -36,17 +35,15 @@ abstract class AbstractPartialRawFirBuilderTestCase : AbstractRawFirBuilderTestC
     private fun testFunctionPartialBuilding(filePath: String, nameToFind: String) {
         testPartialBuilding(
             filePath,
-            { file -> file.findDescendantOfType<KtNamedFunction> { it.name == nameToFind }!! },
-            RawFirBuilder::buildFunctionWithBody
-        )
+            { file -> file.findDescendantOfType<KtNamedFunction> { it.name == nameToFind }!! }
+        ) { rawFirBuilder, function -> rawFirBuilder.buildFunctionWithBody(function, original = null) }
     }
 
     private fun testPropertyPartialBuilding(filePath: String, nameToFind: String) {
         testPartialBuilding(
             filePath,
-            { file -> file.findDescendantOfType<KtProperty> { it.name == nameToFind }!! },
-            RawFirBuilder::buildPropertyWithBody
-        )
+            { file -> file.findDescendantOfType<KtProperty> { it.name == nameToFind }!! }
+        ) { rawFirBuilder, property -> rawFirBuilder.buildPropertyWithBody(property, original = null) }
     }
 
     private fun <T : KtElement> testPartialBuilding(