[TEST] Add ability to register compiler extensions in EnvironmentConfigurator

compiler/test-infrastructure/tests/org/jetbrains/kotlin/test/services/EnvironmentConfigurator.kt

@@ -5,6 +5,7 @@
 
 package org.jetbrains.kotlin.test.services
 
+import com.intellij.openapi.project.Project
 import org.jetbrains.kotlin.config.AnalysisFlag
 import org.jetbrains.kotlin.config.CompilerConfiguration
 import org.jetbrains.kotlin.config.CompilerConfigurationKey
@@ -38,6 +39,8 @@ abstract class EnvironmentConfigurator(protected val testServices: TestServices)
     open fun provideAdditionalAnalysisFlags(directives: RegisteredDirectives): Map<AnalysisFlag<*>, Any?> {
         return emptyMap()
     }
+
+    open fun registerCompilerExtensions(project: Project) {}
 }
 
 class DirectiveToConfigurationKeyExtractor {

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/services/CompilerConfigurationProvider.kt

@@ -84,6 +84,8 @@ open class CompilerConfigurationProviderImpl(
             CompilerConfiguration()
         )
         val projectEnv = KotlinCoreEnvironment.ProjectEnvironment(testRootDisposable, applicationEnvironment)
+        val project = projectEnv.project
+        configurators.forEach { it.registerCompilerExtensions(project) }
         return KotlinCoreEnvironment.createForTests(
             projectEnv,
             createCompilerConfiguration(module),