[JS IR] Enable directives for IR box tests

compiler/ir/backend.js/src/org/jetbrains/kotlin/ir/backend/js/compiler.kt

@@ -20,6 +20,7 @@ import org.jetbrains.kotlin.ir.declarations.StageController
 import org.jetbrains.kotlin.ir.declarations.persistent.PersistentIrFactory
 import org.jetbrains.kotlin.ir.util.ExternalDependenciesGenerator
 import org.jetbrains.kotlin.ir.util.noUnboundLeft
+import org.jetbrains.kotlin.js.backend.ast.JsProgram
 import org.jetbrains.kotlin.js.config.RuntimeDiagnostic
 import org.jetbrains.kotlin.name.FqName
 
@@ -31,6 +32,7 @@ class CompilerResult(
 
 class CompilationOutputs(
     val jsCode: String,
+    val jsProgram: JsProgram? = null,
     val sourceMap: String? = null,
     val dependencies: Iterable<Pair<String, CompilationOutputs>> = emptyList()
 )

compiler/ir/backend.js/src/org/jetbrains/kotlin/ir/backend/js/transformers/irToJs/IrModuleToJsTransformer.kt

@@ -113,7 +113,7 @@ class IrModuleToJsTransformer(
                 )
             }.reversed()
 
-            return CompilationOutputs(mainModule.jsCode, mainModule.sourceMap, dependencies)
+            return CompilationOutputs(mainModule.jsCode, mainModule.jsProgram, mainModule.sourceMap, dependencies)
         } else {
             return generateWrappedModuleBody2(
                 modules,
@@ -228,6 +228,7 @@ class IrModuleToJsTransformer(
 
         return CompilationOutputs(
             jsCode.toString(),
+            program,
             if(sourceMapsEnabled) sourceMapBuilder.build() else null
         )
     }

js/js.tests/test/org/jetbrains/kotlin/js/test/BasicBoxTest.kt

@@ -790,7 +790,7 @@ abstract class BasicBoxTest(
             incrementalData.header = incrementalService.headerMetadata
         }
 
-        processJsProgram(translationResult.program, units.filterIsInstance<TranslationUnit.SourceFile>().map { it.file })
+        processJsProgram(translationResult.program, units)
         checkSourceMap(outputFile, translationResult.program, remap)
     }
 
@@ -816,11 +816,15 @@ abstract class BasicBoxTest(
         }
     }
 
-    private fun processJsProgram(program: JsProgram, psiFiles: List<KtFile>) {
+    protected fun processJsProgram(program: JsProgram, units: List<TranslationUnit>) {
-        psiFiles.asSequence()
+        units.filterIsInstance<TranslationUnit.SourceFile>()
-            .map { it.text }
+            .forEach { DirectiveTestUtils.processDirectives(program, it.file.text, targetBackend) }
-            .forEach { DirectiveTestUtils.processDirectives(program, it) }
+
-        program.verifyAst()
+        // TODO: For now the IR backend generates JS code that doesn't pass verification,
+        // TODO: so we temporarily disabled AST verification.
+        if (targetBackend == TargetBackend.JS) {
+            program.verifyAst()
+        }
     }
 
     private fun checkSourceMap(outputFile: File, program: JsProgram, remap: Boolean) {

js/js.tests/test/org/jetbrains/kotlin/js/test/BasicIrBoxTest.kt

@@ -347,6 +347,8 @@ abstract class BasicIrBoxTest(
                     logger.logFile("Output d.ts", dtsFile)
                     dtsFile.write(compiledModule.tsDefinitions ?: error("No ts definitions"))
                 }
+
+                compiledOutput.jsProgram?.let { processJsProgram(it, units) }
             }
 
             if (runIrPir) {
@@ -368,6 +370,7 @@ abstract class BasicIrBoxTest(
                     verifySignatures = !skipMangleVerification,
                 )
                 compiledModule.outputs!!.writeTo(pirOutputFile, config)
+                processJsProgram(compiledModule.outputs!!.jsProgram!!, units)
             }
         }
     }