[TEST] Add Klib ABI test runner logic

- common and js-specific classes
2021-12-06 19:20:39 +03:00
parent 4ef316a590
commit c3c2bb5884
2 changed files with 312 additions and 0 deletions
@@ -0,0 +1,110 @@
+/*
+ * Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.js.test
+
+import com.intellij.openapi.project.Project
+import org.jetbrains.kotlin.backend.common.phaser.PhaseConfig
+import org.jetbrains.kotlin.cli.common.messages.AnalyzerWithCompilerReport
+import org.jetbrains.kotlin.config.CommonConfigurationKeys
+import org.jetbrains.kotlin.config.CompilerConfiguration
+import org.jetbrains.kotlin.ir.backend.js.*
+import org.jetbrains.kotlin.ir.backend.js.codegen.JsGenerationGranularity
+import org.jetbrains.kotlin.ir.backend.js.transformers.irToJs.IrModuleToJsTransformerTmp
+import org.jetbrains.kotlin.ir.declarations.impl.IrFactoryImpl
+import org.jetbrains.kotlin.ir.declarations.impl.IrFactoryImplForJsIC
+import org.jetbrains.kotlin.js.config.JSConfigurationKeys
+import org.jetbrains.kotlin.js.testOld.V8IrJsTestChecker
+import org.jetbrains.kotlin.klib.AbstractKlibABITestCase
+import org.jetbrains.kotlin.name.FqName
+import org.jetbrains.kotlin.psi.KtFile
+import org.jetbrains.kotlin.serialization.js.ModuleKind
+import java.io.File
+
+abstract class AbstractJsKLibABITestCase : AbstractKlibABITestCase() {
+
+    override fun compileBinaryAndRun(project: Project, configuration: CompilerConfiguration, libraries: Collection<String>, mainModulePath: String, buildDir: File) {
+        configuration.put(JSConfigurationKeys.PARTIAL_LINKAGE, true)
+        configuration.put(JSConfigurationKeys.MODULE_KIND, ModuleKind.PLAIN)
+        configuration.put(CommonConfigurationKeys.MODULE_NAME, MAIN_MODULE_NAME)
+        val kLib = MainModule.Klib(mainModulePath)
+        val moduleStructure = ModulesStructure(project, kLib, configuration, libraries, emptyList(), false, false, emptyMap())
+
+        val ir = compile(
+            moduleStructure,
+            PhaseConfig(jsPhases),
+            IrFactoryImplForJsIC(WholeWorldStageController()),
+            propertyLazyInitialization = true,
+            granularity = JsGenerationGranularity.PER_MODULE,
+            exportedDeclarations = setOf(FqName("box")),
+            icCompatibleIr2Js = true
+        )
+
+        val transformer = IrModuleToJsTransformerTmp(
+            ir.context,
+            emptyList(),
+            fullJs = true,
+            dceJs = true,
+            multiModule = true,
+            relativeRequirePath = false
+        )
+
+        val compiledResult = transformer.generateModule(ir.allModules)
+
+        val dceOutput = compiledResult.outputsAfterDce ?: error("No DCE output")
+
+        val binariesDir = File(buildDir, BIN_DIR_NAME).also { it.mkdirs() }
+
+        val binaries = ArrayList<File>(libraries.size)
+
+        for ((name, code) in dceOutput.dependencies) {
+            val depBinary = binariesDir.binJsFile(name)
+            depBinary.parentFile?.let { if (!it.exists()) it.mkdirs() }
+            depBinary.writeText(code.jsCode)
+            binaries.add(depBinary)
+        }
+
+        val mainBinary = binariesDir.binJsFile(MAIN_MODULE_NAME)
+        mainBinary.writeText(dceOutput.jsCode)
+        binaries.add(mainBinary)
+
+        executeAndCheckBinaries(MAIN_MODULE_NAME, binaries)
+    }
+
+    private fun File.binJsFile(name: String): File = File(this, "$name.js")
+
+    private fun executeAndCheckBinaries(mainModuleName: String, dependencies: Collection<File>) {
+        val checker = V8IrJsTestChecker
+
+        val filePaths = dependencies.map { it.canonicalPath }
+        checker.check(filePaths, mainModuleName, null, "box", "OK", withModuleSystem = false)
+    }
+
+    override fun buildKlibImpl(
+        project: Project,
+        configuration: CompilerConfiguration,
+        moduleName: String,
+        sources: Collection<KtFile>,
+        dependencies: Collection<String>,
+        outputFile: File
+    ) {
+        val sourceModule = prepareAnalyzedSourceModule(
+            project,
+            sources.toList(),
+            configuration,
+            dependencies.toList(),
+            emptyList(), // TODO
+            AnalyzerWithCompilerReport(configuration)
+        )
+
+        generateKLib(sourceModule, IrFactoryImpl, outputFile.canonicalPath, nopack = false, jsOutputName = moduleName)
+    }
+
+    override fun stdlibPath(): String = "libraries/stdlib/js-ir/build/classes/kotlin/js/main"
+
+    companion object {
+        private const val BIN_DIR_NAME = "_bins_js"
+    }
+}