[FE] Add test for KT-54055

compiler/testData/diagnostics/tests/controlFlowAnalysis/uninitializedCompanionOfEnum_after.fir.kt

@@ -1,5 +1,5 @@
 // LANGUAGE: +ProhibitAccessToEnumCompanionMembersInEnumConstructorCall
-// ISSUE: KT-49110
+// ISSUE: KT-49110, KT-54055
 
 enum class SomeEnum(val x: Int) {
     A(<!UNINITIALIZED_ENUM_COMPANION!>companionFun<!>().length),// UNINITIALIZED_ENUM_COMPANION
@@ -37,3 +37,28 @@ fun OtherEnum.Companion.extensionFun(): String = companionFun()
 val OtherEnum.Companion.extensionProp: String
     get() = companionProp
 
+enum class EnumWithLambda(val lambda: () -> Unit) {
+    M({
+      companionFun().length
+      companionProp.length
+
+      EnumWithLambda.companionFun().length
+      EnumWithLambda.companionProp.length
+
+      <!UNINITIALIZED_ENUM_COMPANION!>extensionFun<!>().length
+      <!UNINITIALIZED_ENUM_COMPANION!>extensionProp<!>.length
+
+      <!UNINITIALIZED_ENUM_COMPANION!>EnumWithLambda<!>.extensionFun().length
+      <!UNINITIALIZED_ENUM_COMPANION!>EnumWithLambda<!>.extensionProp.length
+      });
+
+    companion object {
+        val companionProp = "someString"
+        fun companionFun(): String = "someString"
+    }
+}
+
+fun EnumWithLambda.Companion.extensionFun(): String = companionFun()
+val EnumWithLambda.Companion.extensionProp: String
+    get() = companionProp
+