Diagnostics tests with JVM backend

2020-02-03 23:08:07 +03:00
parent a17027e330
commit 4e6c3b6b43
6 changed files with 161 additions and 0 deletions
@@ -0,0 +1,70 @@
+/*
+ * Copyright 2010-2020 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.checkers
+
+import org.jetbrains.kotlin.analyzer.AnalysisResult
+import org.jetbrains.kotlin.backend.common.phaser.PhaseConfig
+import org.jetbrains.kotlin.backend.jvm.JvmIrCodegenFactory
+import org.jetbrains.kotlin.backend.jvm.jvmPhases
+import org.jetbrains.kotlin.cli.common.CLIConfigurationKeys
+import org.jetbrains.kotlin.codegen.ClassBuilderFactories
+import org.jetbrains.kotlin.codegen.DefaultCodegenFactory
+import org.jetbrains.kotlin.codegen.KotlinCodegenFacade
+import org.jetbrains.kotlin.codegen.state.GenerationState
+import org.jetbrains.kotlin.config.JvmTarget
+import org.jetbrains.kotlin.config.LanguageVersionSettings
+import org.jetbrains.kotlin.context.ModuleContext
+import org.jetbrains.kotlin.psi.KtFile
+import org.jetbrains.kotlin.resolve.BindingTrace
+
+abstract class AbstractDiagnosticsTestWithJvmBackend : AbstractDiagnosticsTest() {
+
+    override fun analyzeModuleContents(
+        moduleContext: ModuleContext,
+        files: List<KtFile>,
+        moduleTrace: BindingTrace,
+        languageVersionSettings: LanguageVersionSettings,
+        separateModules: Boolean,
+        jvmTarget: JvmTarget
+    ): AnalysisResult {
+        val analysisResult =
+            super.analyzeModuleContents(
+                moduleContext, files, moduleTrace, languageVersionSettings, separateModules, jvmTarget
+            )
+
+        val generationState =
+            GenerationState.Builder(
+                project, ClassBuilderFactories.TEST, analysisResult.moduleDescriptor, analysisResult.bindingContext,
+                files, environment.configuration
+            ).setupGenerationState().build()
+
+        KotlinCodegenFacade.compileCorrectFiles(generationState)
+
+        for (diagnostic in generationState.collectedExtraJvmDiagnostics.all()) {
+            moduleTrace.report(diagnostic)
+        }
+        return analysisResult
+    }
+
+    protected abstract fun GenerationState.Builder.setupGenerationState(): GenerationState.Builder
+}
+
+abstract class AbstractDiagnosticsTestWithOldJvmBackend : AbstractDiagnosticsTestWithJvmBackend() {
+
+    override fun GenerationState.Builder.setupGenerationState(): GenerationState.Builder =
+        codegenFactory(DefaultCodegenFactory).isIrBackend(false)
+}
+
+abstract class AbstractDiagnosticsTestWithJvmIrBackend : AbstractDiagnosticsTestWithJvmBackend() {
+
+    override fun GenerationState.Builder.setupGenerationState(): GenerationState.Builder =
+        codegenFactory(
+            JvmIrCodegenFactory(
+                environment.configuration.get(CLIConfigurationKeys.PHASE_CONFIG)
+                    ?: PhaseConfig(jvmPhases)
+            )
+        ).isIrBackend(true)
+}