Add more tests for header compilation.

^KT-78422

Author: aviroop-123

compiler/fir/analysis-tests/testData/resolve/headerMode/classDeclaration.fir.txt

@@ -21,6 +21,12 @@ FILE: classDeclaration.kt
 
         public final fun funD(): R|kotlin/Int|
 
+        public final inline fun funE(): R|kotlin/String| {
+            local final fun funF(): R|kotlin/String|
+
+            ^funE R|<local>/funF|()
+        }
+
     }
     public abstract interface B : R|kotlin/Any| {
         public open fun funA(): R|kotlin/String|

compiler/fir/analysis-tests/testData/resolve/headerMode/classDeclaration.kt

@@ -25,6 +25,11 @@ class A {
     }
 
     fun funD() = 1 + 2
+
+    inline fun funE(): String {
+        fun funF() = "funF body"
+        return funF()
+    }
 }
 
 interface B {

compiler/fir/analysis-tests/testData/resolve/headerMode/constructorDeclaration.fir.txt

@@ -0,0 +1,45 @@
+FILE: constructorDeclaration.kt
+    public final class A : R|kotlin/Any| {
+        public constructor(a: R|kotlin/String|, b: R|kotlin/Int|): R|A| {
+            super<R|kotlin/Any|>()
+        }
+
+        public final val a: R|kotlin/String| = R|<local>/a|
+            public get(): R|kotlin/String|
+
+        public final val b: R|kotlin/Int| = R|<local>/b|
+            public get(): R|kotlin/Int|
+
+        public constructor(e: R|kotlin/String|): R|A| {
+            this<R|A|>(R|<local>/e|, Int(0))
+            lval message: R|kotlin/String| = String(Secondary constructor body)
+        }
+
+        public constructor(): R|A| {
+            this<R|A|>(String())
+            lval message: R|kotlin/String| = String(Secondary constructor body with delegated constructor call)
+        }
+
+    }
+    public final class B : R|kotlin/Any| {
+        private constructor(a: R|kotlin/String|, b: R|kotlin/Int|): R|B| {
+            super<R|kotlin/Any|>()
+        }
+
+        public final val a: R|kotlin/String| = R|<local>/a|
+            public get(): R|kotlin/String|
+
+        public final val b: R|kotlin/Int| = R|<local>/b|
+            public get(): R|kotlin/Int|
+
+        public constructor(e: R|kotlin/String|): R|B| {
+            this<R|B|>(R|<local>/e|, Int(0))
+            lval message: R|kotlin/String| = String(Secondary constructor body)
+        }
+
+        public constructor(): R|B| {
+            this<R|B|>(String())
+            lval message: R|kotlin/String| = String(Secondary constructor body with delegated constructor call)
+        }
+
+    }

compiler/fir/analysis-tests/testData/resolve/headerMode/constructorDeclaration.kt

@@ -0,0 +1,22 @@
+// RUN_PIPELINE_TILL: BACKEND
+// FIR_DUMP
+class A(val a: String, val b: Int) {
+    constructor(e: String): this(e, 0) {
+        val message = "Secondary constructor body"
+    }
+    constructor(): this("") {
+        val message = "Secondary constructor body with delegated constructor call"
+    }
+}
+
+class B private constructor(val a: String, val b: Int) {
+    constructor(e: String): this(e, 0) {
+        val message = "Secondary constructor body"
+    }
+    constructor(): this("") {
+        val message = "Secondary constructor body with delegated constructor call"
+    }
+}
+
+/* GENERATED_FIR_TAGS: classDeclaration, functionDeclaration, primaryConstructor, propertyDeclaration,
+secondaryConstructor */

compiler/fir/analysis-tests/testData/resolve/headerMode/enumClassDeclaration.fir.txt

@@ -0,0 +1,49 @@
+FILE: enumClassDeclaration.kt
+    public final enum class A : R|kotlin/Enum<A>| {
+        private constructor(): R|A| {
+            super<R|kotlin/Enum<A>|>()
+        }
+
+        public final static enum entry EAST: R|A|
+        public final static enum entry WEST: R|A|
+        public final static fun values(): R|kotlin/Array<A>|
+
+        public final static fun valueOf(value: R|kotlin/String|): R|A|
+
+        public final static val entries: R|kotlin/enums/EnumEntries<A>|
+            public get(): R|kotlin/enums/EnumEntries<A>|
+
+    }
+    public final enum class B : R|kotlin/Enum<B>| {
+        private constructor(): R|B| {
+            super<R|kotlin/Enum<B>|>()
+        }
+
+        public final static enum entry NORTH: R|B| = object : R|B| {
+            private constructor(): R|<anonymous>| {
+                super<R|B|>()
+            }
+
+            public open override fun getString(): R|kotlin/String|
+
+        }
+
+        public final static enum entry SOUTH: R|B| = object : R|B| {
+            private constructor(): R|<anonymous>| {
+                super<R|B|>()
+            }
+
+            public open override fun getString(): R|kotlin/String|
+
+        }
+
+        public abstract fun getString(): R|kotlin/String|
+
+        public final static fun values(): R|kotlin/Array<B>|
+
+        public final static fun valueOf(value: R|kotlin/String|): R|B|
+
+        public final static val entries: R|kotlin/enums/EnumEntries<B>|
+            public get(): R|kotlin/enums/EnumEntries<B>|
+
+    }

compiler/fir/analysis-tests/testData/resolve/headerMode/enumClassDeclaration.kt

@@ -0,0 +1,21 @@
+// RUN_PIPELINE_TILL: BACKEND
+// FIR_DUMP
+enum class A {
+    EAST,
+    WEST
+}
+
+enum class B {
+    NORTH {
+        override fun getString() = "north"
+    },
+    SOUTH {
+        override fun getString(): String {
+            return "south"
+        }
+    };
+
+    abstract fun getString(): String
+}
+
+/* GENERATED_FIR_TAGS: enumDeclaration, enumEntry */

compiler/fir/analysis-tests/testData/resolve/headerMode/functionDeclaration.fir.txt

@@ -11,3 +11,23 @@ FILE: functionDeclaration.kt
 
     private final fun funC(): R|kotlin/String|
     public final fun funD(): R|kotlin/Int|
+    public final inline fun funE(): R|kotlin/String| {
+        local final fun funF(): R|kotlin/String|
+
+        ^funE R|<local>/funF|()
+    }
+    public final inline fun funG(): R|kotlin/String| {
+        local final class classA : R|kotlin/Any| {
+            public constructor(): R|<local>/classA| {
+                super<R|kotlin/Any|>()
+            }
+
+            public final fun funH(): R|kotlin/String|
+
+        }
+
+        lval a: R|<local>/classA| = R|<local>/classA.classA|()
+        ^funG R|<local>/a|.R|<local>/funH|()
+    }
+    public final fun funI(): R|kotlin/Int|
+    public final fun funJ(): R|kotlin/String|

compiler/fir/analysis-tests/testData/resolve/headerMode/functionDeclaration.kt

@@ -3,14 +3,17 @@
 import kotlin.contracts.ExperimentalContracts
 import kotlin.contracts.contract
 
+// Public function
 fun funA(): String {
     return "funA body"
 }
 
+// Inline function
 inline fun funB(): String {
     return "funB body"
 }
 
+// Function with contract
 @OptIn(ExperimentalContracts::class)
 fun isNotNull(value: Any?): Boolean {
     contract {
@@ -19,11 +22,38 @@ fun isNotNull(value: Any?): Boolean {
     return value != null
 }
 
+// Private function
 private fun funC(): String {
     return "funC body"
 }
 
+// Implicit return type
 fun funD() = 1 + 2
 
+// Function inside a function
+inline fun funE(): String {
+    fun funF(): String {
+        return "funF body"
+    }
+    return funF()
+}
+
+// Class inside a function
+inline fun funG(): String {
+    class classA {
+        fun funH() = "funH body"
+    }
+    val a = classA()
+    return a.funH()
+}
+
+// Implicit type reference from another function.
+fun funI() = funD()
+
+fun funJ(): String {
+    inline fun funK() = "funK body"
+    return funK()
+}
+
 /* GENERATED_FIR_TAGS: classReference, contractConditionalEffect, contracts, functionDeclaration, inline, nullableType,
 stringLiteral */

compiler/fir/analysis-tests/testData/resolve/headerMode/objectDeclaration.fir.txt

@@ -21,4 +21,10 @@ FILE: objectDeclaration.kt
 
         public final fun funD(): R|kotlin/Int|
 
+        public final inline fun funE(): R|kotlin/String| {
+            local final fun funF(): R|kotlin/String|
+
+            ^funE R|<local>/funF|()
+        }
+
     }

compiler/fir/analysis-tests/testData/resolve/headerMode/objectDeclaration.kt

@@ -25,6 +25,11 @@ object A {
     }
 
     fun funD() = 1 + 2
+
+    inline fun funE(): String {
+        fun funF() = "funF body"
+        return funF()
+    }
 }
 
 /* GENERATED_FIR_TAGS: classReference, contractConditionalEffect, contracts, functionDeclaration, inline, nullableType,