@@ -25,10 +25,10 @@ import com.intellij.testFramework.TestDataFile
2525import org.jetbrains.kotlin.analysis.test.framework.services.TargetPlatformDirectives
2626import org.jetbrains.kotlin.analysis.test.framework.services.TargetPlatformProviderForAnalysisApiTests
2727import org.jetbrains.kotlin.cli.common.disposeRootInWriteAction
28- import org.jetbrains.kotlin.cli.jvm.compiler.EnvironmentConfigFiles
29- import org.jetbrains.kotlin.cli.jvm.compiler.KotlinCoreEnvironment
28+ import org.jetbrains.kotlin.cli.common.output.writeAllTo
3029import org.jetbrains.kotlin.cli.jvm.config.addJavaSourceRoot
3130import org.jetbrains.kotlin.cli.jvm.config.addJvmClasspathRoots
31+ import org.jetbrains.kotlin.codegen.ClassBuilderFactories
3232import org.jetbrains.kotlin.codegen.GenerationUtils
3333import org.jetbrains.kotlin.config.JvmTarget
3434import org.jetbrains.kotlin.platform.jvm.JvmPlatforms
@@ -176,13 +176,17 @@ abstract class AbstractKSPTest(frontend: FrontendKind<*>) : DisposableTest() {
176176 compilerConfiguration.addJavaSourceRoot(module.javaDir)
177177
178178 // TODO: other platforms
179- val kotlinCoreEnvironment = KotlinCoreEnvironment .createForTests(
180- disposable,
179+ val configurationProvider = testServices.compilerConfigurationProvider
180+ val project = configurationProvider.getProject(module)
181+ val ktFiles = module.loadKtFiles(project)
182+ GenerationUtils .compileFiles(
183+ ktFiles,
181184 compilerConfiguration,
182- EnvironmentConfigFiles .JVM_CONFIG_FILES
183- )
184- val ktFiles = module.loadKtFiles(kotlinCoreEnvironment.project)
185- GenerationUtils .compileFilesTo(ktFiles, kotlinCoreEnvironment, module.outDir)
185+ ClassBuilderFactories .TEST ,
186+ configurationProvider.getPackagePartProviderFactory(module)
187+ ).first.factory.apply {
188+ writeAllTo(module.outDir)
189+ }
186190
187191 if (module.javaFiles.isEmpty())
188192 return
0 commit comments