Check lambda inlining in package part files in test framework

compiler/testData/codegen/boxInline/anonymousObject/enumWhen/callSite.kt

@@ -1,3 +1,4 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
 
 package test

compiler/testData/codegen/boxInline/anonymousObject/kt8133.kt

@@ -1,3 +1,4 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
 
 package test

compiler/testData/codegen/boxInline/anonymousObject/kt9591.kt

@@ -1,5 +1,6 @@
-// FILE: 1.kt
+// NO_CHECK_LAMBDA_INLINING
 // WITH_RUNTIME
+// FILE: 1.kt
 package test
 
 

compiler/testData/codegen/boxInline/anonymousObject/kt9877.kt

@@ -1,5 +1,6 @@
-// FILE: 1.kt
+// NO_CHECK_LAMBDA_INLINING
 // WITH_RUNTIME
+// FILE: 1.kt
 package test
 
 fun <T> T.noInline(p: (T) -> Unit) {

compiler/testData/codegen/boxInline/anonymousObject/kt9877_2.kt

@@ -1,3 +1,4 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
 
 package test

compiler/testData/codegen/boxInline/anonymousObject/properRecapturing/lambdaChain.kt

@@ -1,3 +1,4 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
 
 package test

compiler/testData/codegen/boxInline/argumentOrder/boundFunctionReference.kt

@@ -1,3 +1,4 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
 
 package test

compiler/testData/codegen/boxInline/argumentOrder/boundFunctionReference2.kt

@@ -1,3 +1,4 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
 
 package test

compiler/testData/codegen/boxInline/argumentOrder/extension.kt

@@ -1,3 +1,4 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
 
 package test

compiler/testData/codegen/boxInline/argumentOrder/extensionInClass.kt

@@ -1,3 +1,4 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
 // WITH_RUNTIME
 package test

compiler/testData/codegen/boxInline/argumentOrder/lambdaMigration.kt

@@ -1,3 +1,4 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
 
 package test

compiler/testData/codegen/boxInline/argumentOrder/lambdaMigrationInClass.kt

@@ -1,3 +1,4 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
 
 package test

compiler/testData/codegen/boxInline/argumentOrder/simple.kt

@@ -1,3 +1,4 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
 
 package test

compiler/testData/codegen/boxInline/argumentOrder/simpleInClass.kt

@@ -1,3 +1,4 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
 
 package test

compiler/testData/codegen/boxInline/callableReference/intrinsic.kt

@@ -2,8 +2,8 @@
 
 package test
 
-inline fun call(p: String, s: String.() -> Int): Int {
-    return p.s()
+inline fun call(a: String, b: String, s: String.(String) -> String): Int {
+    return a.s(b)
 }
 
 // FILE: 2.kt
@@ -11,5 +11,5 @@ inline fun call(p: String, s: String.() -> Int): Int {
 import test.*
 
 fun box() : String {
-    return if (call("123", String::length) == 3) "OK" else "fail"
+    return return call("O", "K", String::plus)
 }

compiler/testData/codegen/boxInline/complex/swapAndWith.kt

@@ -1,3 +1,4 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
 package test
 

compiler/testData/codegen/boxInline/enclosingInfo/anonymousInLambda.kt

@@ -1,5 +1,6 @@
-// FILE: 1.kt
+// NO_CHECK_LAMBDA_INLINING
 // WITH_REFLECT
+// FILE: 1.kt
 package test
 
 inline fun <R> call(s: () -> R) = s()

compiler/testData/codegen/boxInline/innerClasses/kt10259.kt

@@ -1,5 +1,6 @@
-// FILE: 1.kt
+// NO_CHECK_LAMBDA_INLINING
 // WITH_RUNTIME
+// FILE: 1.kt
 package test
 
 inline fun test(s: () -> Unit) {

compiler/testData/codegen/boxInline/smap/newsmap/mappingInSubInlineLambda.kt

@@ -1,5 +1,5 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
-
 package test
 
 inline fun test(s: () -> Unit) {

compiler/testData/codegen/boxInline/smap/newsmap/mappingInSubInlineLambdaSameFileInline.kt

@@ -1,5 +1,5 @@
+// NO_CHECK_LAMBDA_INLINING
 // FILE: 1.kt
-
 package test
 
 inline fun test(s: () -> Unit) {

compiler/testData/codegen/boxInline/trait/trait.kt

@@ -4,10 +4,18 @@ package test
 
 interface InlineTrait {
 
-    fun finalInline(s: () -> String): String {
+    private inline fun privateInline(s: () -> String): String {
         return s()
     }
 
+    fun testPrivateInline(): String {
+        return privateInline { "private" }
+    }
+
+    fun testPrivateInline2(): String {
+        return privateInline { "private2" }
+    }
+
     companion object {
         inline final fun finalInline(s: () -> String): String {
             return s()
@@ -15,7 +23,7 @@ interface InlineTrait {
     }
 }
 
-class Z: InlineTrait {
+class Z : InlineTrait {
 
 }
 
@@ -23,21 +31,13 @@ class Z: InlineTrait {
 
 import test.*
 
-fun testFinalInline(): String {
-    return Z().finalInline({"final"})
-}
-
-fun testFinalInline2(instance: InlineTrait): String {
-    return instance.finalInline({"final2"})
-}
-
 fun testClassObject(): String {
-    return InlineTrait.finalInline({"classobject"})
+    return InlineTrait.finalInline({ "classobject" })
 }
 
 fun box(): String {
-    if (testFinalInline() != "final") return "test1: ${testFinalInline()}"
-    if (testFinalInline2(Z()) != "final2") return "test2: ${testFinalInline2(Z())}"
+    if (Z().testPrivateInline() != "private") return "test1: ${Z().testPrivateInline()}"
+    if (Z().testPrivateInline2() != "private2") return "test2: ${Z().testPrivateInline2()}"
     if (testClassObject() != "classobject") return "test3: ${testClassObject()}"
 
     return "OK"

compiler/tests-common/org/jetbrains/kotlin/codegen/InlineTestUtil.kt

@@ -40,7 +40,7 @@ object InlineTestUtil {
 
         val skipParameterChecking = sourceFiles.any {
             InTextDirectivesUtils.isDirectiveDefined(it.text, "NO_CHECK_LAMBDA_INLINING")
-        }
+        } || !doLambdaInliningCheck(files, inlineInfo)
 
         if (!skipParameterChecking) {
             val notInlinedParameters = checkParametersInlined(files, inlineInfo)
@@ -54,20 +54,16 @@ object InlineTestUtil {
     private fun obtainInlineInfo(files: Iterable<OutputFile>): InlineInfo {
         val inlineMethods = HashSet<MethodInfo>()
         val binaryClasses = hashMapOf<String, KotlinJvmBinaryClass>()
-
         for (file in files) {
             val binaryClass = loadBinaryClass(file)
             val inlineFunctions = inlineFunctionsJvmNames(binaryClass.classHeader)
 
             val classVisitor = object : ClassVisitorWithName() {
-                override fun visitMethod(access: Int, name: String, desc: String, signature: String?, exceptions: Array<String>?): MethodVisitor {
-                    return object : MethodNode(Opcodes.ASM5, access, name, desc, signature, exceptions) {
-                        override fun visitEnd() {
-                            if (name + desc in inlineFunctions) {
-                                inlineMethods.add(MethodInfo(className, name, this.desc))
-                            }
-                        }
+                override fun visitMethod(access: Int, name: String, desc: String, signature: String?, exceptions: Array<String>?): MethodVisitor? {
+                    if (name + desc in inlineFunctions) {
+                        inlineMethods.add(MethodInfo(className, name, desc))
                     }
+                    return null
                 }
             }
 
@@ -78,6 +74,34 @@ object InlineTestUtil {
         return InlineInfo(inlineMethods, binaryClasses)
     }
 
+    private fun doLambdaInliningCheck(files: Iterable<OutputFile>, inlineInfo: InlineInfo): Boolean {
+        var doLambdaInliningCheck = true
+        for (file in files) {
+            val binaryClass = loadBinaryClass(file)
+            val inlineFunctions = inlineFunctionsJvmNames(binaryClass.classHeader)
+
+            val classVisitor = object : ClassVisitorWithName() {
+                override fun visitMethod(access: Int, name: String, desc: String, signature: String?, exceptions: Array<String>?): MethodVisitor? {
+                    if (name + desc in inlineFunctions) {
+                        return object: MethodNodeWithAnonymousObjectCheck(inlineInfo, access, name, desc, signature, exceptions) {
+                            override fun onAnonymousConstructorCallOrSingletonAccess(owner: String) {
+                                doLambdaInliningCheck = false
+                            }
+                        }
+                    }
+                    return null
+                }
+            }
+
+            ClassReader(file.asByteArray()).accept(classVisitor, 0)
+
+            if (!doLambdaInliningCheck) break
+        }
+
+        return doLambdaInliningCheck
+    }
+
+
     private fun checkInlineMethodNotInvoked(files: Iterable<OutputFile>, inlinedMethods: Set<MethodInfo>): List<NotInlinedCall> {
         val notInlined = ArrayList<NotInlinedCall>()
 
@@ -139,23 +163,10 @@ object InlineTestUtil {
                         return null
                     }
 
-                    return object : MethodNode(Opcodes.ASM5, access, name, desc, signature, exceptions) {
-                        private fun isInlineParameterLikeOwner(owner: String) =
-                                "$" in owner && !isTopLevelOrInnerOrPackageClass(owner, inlineInfo)
-
-                        override fun visitMethodInsn(opcode: Int, owner: String, name: String, desc: String, itf: Boolean) {
-                            if ("<init>".equals(name) && isInlineParameterLikeOwner(owner)) {
-                                val fromCall = MethodInfo(className, this.name, this.desc)
-                                notInlinedParameters.add(NotInlinedParameter(owner, fromCall))
-                            }
-                        }
-
-                        override fun visitFieldInsn(opcode: Int, owner: String, name: String, desc: String) {
-                            if (opcode == Opcodes.GETSTATIC && isInlineParameterLikeOwner(owner)) {
-                                val fromCall = MethodInfo(className, this.name, this.desc)
-                                notInlinedParameters.add(NotInlinedParameter(owner, fromCall))
-                            }
-                            super.visitFieldInsn(opcode, owner, name, desc)
+                    return object: MethodNodeWithAnonymousObjectCheck(inlineInfo, access, name, desc, signature, exceptions) {
+                        override fun onAnonymousConstructorCallOrSingletonAccess(owner: String) {
+                            val fromCall = MethodInfo(className, this.name, this.desc)
+                            notInlinedParameters.add(NotInlinedParameter(owner, fromCall))
                         }
                     }
                 }
@@ -174,6 +185,8 @@ object InlineTestUtil {
 
     private fun isClassOrPackagePartKind(klass: KotlinJvmBinaryClass): Boolean {
         return klass.classHeader.kind == KotlinClassHeader.Kind.CLASS && !klass.classId.isLocal
+               || klass.classHeader.kind == KotlinClassHeader.Kind.FILE_FACADE /*single file facade equals to package part*/
+               || klass.classHeader.kind == KotlinClassHeader.Kind.MULTIFILE_CLASS_PART
     }
 
     private fun loadBinaryClass(file: OutputFile): KotlinJvmBinaryClass {
@@ -207,4 +220,24 @@ object InlineTestUtil {
             super.visit(version, access, name, signature, superName, interfaces)
         }
     }
+
+    private abstract class MethodNodeWithAnonymousObjectCheck(val inlineInfo: InlineInfo, access: Int, name: String, desc: String, signature: String?, exceptions: Array<String>?) : MethodNode(Opcodes.ASM5, access, name, desc, signature, exceptions) {
+        private fun isInlineParameterLikeOwner(owner: String) =
+                "$" in owner && !isTopLevelOrInnerOrPackageClass(owner, inlineInfo)
+
+        override fun visitMethodInsn(opcode: Int, owner: String, name: String, desc: String, itf: Boolean) {
+            if ("<init>" == name && isInlineParameterLikeOwner(owner)) {
+                onAnonymousConstructorCallOrSingletonAccess(owner)
+            }
+        }
+
+        abstract fun onAnonymousConstructorCallOrSingletonAccess(owner: String)
+
+        override fun visitFieldInsn(opcode: Int, owner: String, name: String, desc: String) {
+            if (opcode == Opcodes.GETSTATIC && isInlineParameterLikeOwner(owner)) {
+                onAnonymousConstructorCallOrSingletonAccess(owner)
+            }
+            super.visitFieldInsn(opcode, owner, name, desc)
+        }
+    }
 }