[Test] Inline processAllIrModuleFragments utility

After previous changes IR output artifact contains only one IrModuleFragment

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/backend/handlers/HandlerUtils.kt

@@ -14,9 +14,7 @@ import org.jetbrains.kotlin.diagnostics.Severity
 import org.jetbrains.kotlin.diagnostics.impl.BaseDiagnosticsCollector
 import org.jetbrains.kotlin.diagnostics.rendering.DefaultErrorMessages
 import org.jetbrains.kotlin.diagnostics.rendering.RootDiagnosticRendererFactory
-import org.jetbrains.kotlin.ir.declarations.IrModuleFragment
 import org.jetbrains.kotlin.test.FirParser
-import org.jetbrains.kotlin.test.backend.ir.IrBackendInput
 import org.jetbrains.kotlin.test.directives.DiagnosticsDirectives
 import org.jetbrains.kotlin.test.directives.FirDiagnosticsDirectives
 import org.jetbrains.kotlin.test.directives.model.singleOrZeroValue
@@ -28,14 +26,6 @@ import org.jetbrains.kotlin.test.services.*
 import org.jetbrains.kotlin.util.capitalizeDecapitalize.toLowerCaseAsciiOnly
 import java.io.File
 
-// TODO: inline this function
-inline fun IrBackendInput.processAllIrModuleFragments(
-    module: TestModule,
-    processor: (irModuleFragment: IrModuleFragment, moduleName: String) -> Unit
-) {
-    processor(irModuleFragment, module.name)
-}
-
 fun BinaryArtifactHandler<*>.reportKtDiagnostics(module: TestModule, ktDiagnosticReporter: BaseDiagnosticsCollector) {
     val globalMetadataInfoHandler = testServices.globalMetadataInfoHandler
     val firParser = module.directives.singleOrZeroValue(FirDiagnosticsDirectives.FIR_PARSER)

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/backend/handlers/IrInlineBodiesHandler.kt

@@ -26,10 +26,8 @@ class IrInlineBodiesHandler(testServices: TestServices) : AbstractIrHandler(test
 
     @OptIn(ObsoleteDescriptorBasedAPI::class)
     override fun processModule(module: TestModule, info: IrBackendInput) {
-        info.processAllIrModuleFragments(module) { irModule, _ ->
+        info.irModuleFragment.acceptChildrenVoid(InlineFunctionsCollector())
-            irModule.acceptChildrenVoid(InlineFunctionsCollector())
+        info.irModuleFragment.acceptChildrenVoid(InlineCallBodiesCheck(firEnabled = module.frontendKind == FrontendKinds.FIR))
-            irModule.acceptChildrenVoid(InlineCallBodiesCheck(firEnabled = module.frontendKind == FrontendKinds.FIR))
-        }
 
         assertions.assertTrue((info as IrBackendInput.JvmIrBackendInput).backendInput.symbolTable.descriptorExtension.allUnboundSymbols.isEmpty())
     }

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/backend/handlers/IrInterpreterBackendHandler.kt

@@ -37,14 +37,12 @@ open class IrInterpreterBackendHandler(testServices: TestServices) : AbstractIrH
     override fun processAfterAllModules(someAssertionWasFailed: Boolean) {}
 
     override fun processModule(module: TestModule, info: IrBackendInput) {
-        info.processAllIrModuleFragments(module) { moduleFragment, _ ->
+        val evaluator = Evaluator(IrInterpreter(info.irModuleFragment.irBuiltins), globalMetadataInfoHandler)
-            val evaluator = Evaluator(IrInterpreter(moduleFragment.irBuiltins), globalMetadataInfoHandler)
+        for ((irFile, testFile) in matchIrFileWithTestFile(info.irModuleFragment, module)) {
-            for ((irFile, testFile) in matchIrFileWithTestFile(moduleFragment, module)) {
             evaluator.evaluate(irFile, testFile)
         }
     }
 }
-}
 
 private class Evaluator(private val interpreter: IrInterpreter, private val globalMetadataInfoHandler: GlobalMetadataInfoHandler) {
     fun evaluate(irFile: IrFile, testFile: TestFile) {

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/backend/handlers/IrPrettyKotlinDumpHandler.kt

@@ -72,9 +72,8 @@ internal fun dumpModuleKotlinLike(
     multiModuleInfoDumper: MultiModuleInfoDumper,
     options: KotlinLikeDumpOptions,
 ) {
-    info.processAllIrModuleFragments(module) { irModuleFragment, moduleName ->
+    val irFiles = info.irModuleFragment.files
-        val irFiles = irModuleFragment.files
+    val builder = multiModuleInfoDumper.builderForModule(module.name)
-        val builder = multiModuleInfoDumper.builderForModule(moduleName)
     val filteredIrFiles = irFiles.groupWithTestFiles(module).filterNot { (testFile, _) ->
         testFile?.let { EXTERNAL_FILE in it.directives || it.isAdditional } ?: false
     }.map { it.second }
@@ -84,4 +83,3 @@ internal fun dumpModuleKotlinLike(
         builder.append(irFile.dumpKotlinLike(modifiedOptions))
     }
 }
-}

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/backend/handlers/IrSourceRangesDumpHandler.kt

@@ -42,14 +42,11 @@ class IrSourceRangesDumpHandler(
 
     override fun processModule(module: TestModule, info: IrBackendInput) {
         if (CodegenTestDirectives.DUMP_SOURCE_RANGES_IR !in module.directives) return
-        info.processAllIrModuleFragments(module) { irModuleFragment, moduleName ->
+        val builder = baseDumper.builderForModule(module.name)
-            val builder = baseDumper.builderForModule(moduleName)
+        for (irFile in info.irModuleFragment.files) {
-
-            for (irFile in irModuleFragment.files) {
             builder.append(irFile.dumpWithSourceLocations(irFile.fileEntry))
         }
     }
-    }
 
     private fun IrElement.dumpWithSourceLocations(fileEntry: IrFileEntry): String =
         StringBuilder().also {

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/backend/handlers/IrTextDumpHandler.kt

@@ -76,10 +76,8 @@ class IrTextDumpHandler(
             printTypeAbbreviations = false,
         )
 
-        info.processAllIrModuleFragments(module) { irModuleFragment, moduleName ->
+        val builder = baseDumper.builderForModule(module.name)
-            val builder = baseDumper.builderForModule(moduleName)
+        val testFileToIrFile = info.irModuleFragment.files.groupWithTestFiles(module)
-            val testFileToIrFile = irModuleFragment.files.groupWithTestFiles(module)
-
         for ((testFile, irFile) in testFileToIrFile) {
             if (testFile?.directives?.contains(EXTERNAL_FILE) == true) continue
             var actualDump = irFile.dumpTreesFromLineNumber(lineNumber = 0, dumpOptions)
@@ -88,7 +86,6 @@ class IrTextDumpHandler(
             }
             builder.append(actualDump)
         }
-        }
 
         compareDumpsOfExternalClasses(module, info)
     }

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/backend/handlers/IrTreeVerifierHandler.kt

@@ -6,6 +6,7 @@
 package org.jetbrains.kotlin.test.backend.handlers
 
 import org.jetbrains.kotlin.ir.IrVerifier
+import org.jetbrains.kotlin.ir.declarations.IrFile
 import org.jetbrains.kotlin.ir.util.DumpIrTreeOptions
 import org.jetbrains.kotlin.ir.util.deepCopyWithSymbols
 import org.jetbrains.kotlin.ir.util.dumpTreesFromLineNumber
@@ -29,8 +30,7 @@ class IrTreeVerifierHandler(
     override fun processModule(module: TestModule, info: IrBackendInput) {
         if (CodegenTestDirectives.DUMP_IR !in module.directives) return
 
-        info.processAllIrModuleFragments(module) { irModuleFragment, _ ->
+        val irFiles = info.irModuleFragment.files
-            val irFiles = irModuleFragment.files
         val testFileToIrFile = irFiles.groupWithTestFiles(module)
         for ((testFile, irFile) in testFileToIrFile) {
             if (testFile?.directives?.contains(EXTERNAL_FILE) == true) continue
@@ -39,12 +39,11 @@ class IrTreeVerifierHandler(
 
             val actualDump = irFile.dumpTreesFromLineNumber(lineNumber = 0, DumpIrTreeOptions(normalizeNames = true))
 
-                val irFileCopy = irFile.deepCopyWithSymbols()
+            val irFileCopy = irFile.deepCopyWithSymbols<IrFile>()
             val dumpOfCopy = irFileCopy.dumpTreesFromLineNumber(lineNumber = 0, DumpIrTreeOptions(normalizeNames = true))
             assertions.assertEquals(actualDump, dumpOfCopy) { "IR dump mismatch after deep copy with symbols" }
         }
     }
-    }
 
     override fun processAfterAllModules(someAssertionWasFailed: Boolean) {}
 }