[Test/AA] Pass project disposable to getOrCreateApplicationEnvironment instead of application disposable

- The disposable passed to `getOrCreateApplicationEnvironment` should
  not actually be the application environment's disposable, which is
  created inside the function. Instead, it should be the project's
  disposable, which is used to track how many projects still rely on the
  shared application environment.
- This issue wasn't apparent before because there is no visible
  consequence when an application isn't disposed after all projects have
  been disposed (during tests). However, the solution for KT-63650
  relies on application environments being disposed after all projects
  are disposed, so that a new application environment with a different
  configuration can be created. (Only one shared application environment
  may be active at the same time.)

^KT-63650

analysis/analysis-api-standalone/analysis-api-standalone-base/src/org/jetbrains/kotlin/analysis/api/standalone/base/project/structure/StandaloneProjectFactory.kt

@@ -69,18 +69,17 @@ import java.nio.file.Paths
 object StandaloneProjectFactory {
     fun createProjectEnvironment(
         projectDisposable: Disposable,
-        applicationDisposable: Disposable,
         applicationEnvironmentMode: KotlinCoreApplicationEnvironmentMode,
         compilerConfiguration: CompilerConfiguration = CompilerConfiguration(),
         classLoader: ClassLoader = MockProject::class.java.classLoader,
     ): KotlinCoreProjectEnvironment {
         val applicationEnvironment = KotlinCoreEnvironment.getOrCreateApplicationEnvironment(
-            applicationDisposable,
+            projectDisposable = projectDisposable,
             compilerConfiguration,
             applicationEnvironmentMode,
         )
 
-        registerApplicationExtensionPoints(applicationEnvironment, applicationDisposable)
+        registerApplicationExtensionPoints(applicationEnvironment)
 
         registerApplicationServices(applicationEnvironment)
 
@@ -138,10 +137,7 @@ object StandaloneProjectFactory {
         }
     }
 
-    private fun registerApplicationExtensionPoints(
-        applicationEnvironment: KotlinCoreApplicationEnvironment,
-        applicationDisposable: Disposable,
-    ) {
+    private fun registerApplicationExtensionPoints(applicationEnvironment: KotlinCoreApplicationEnvironment) {
         val applicationArea = applicationEnvironment.application.extensionArea
 
         if (!applicationArea.hasExtensionPoint(AdditionalKDocResolutionProvider.EP_NAME)) {
@@ -162,7 +158,7 @@ object StandaloneProjectFactory {
                     ClassTypePointerFactory::class.java
                 )
                 applicationArea.getExtensionPoint(ClassTypePointerFactory.EP_NAME)
-                    .registerExtension(PsiClassReferenceTypePointerFactory(), applicationDisposable)
+                    .registerExtension(PsiClassReferenceTypePointerFactory(), applicationEnvironment.application)
             }
         }
     }

analysis/analysis-api-standalone/analysis-api-standalone-native/build.gradle.kts

@@ -8,6 +8,7 @@ dependencies {
     testImplementation(projectTests(":compiler:tests-common"))
     testImplementation(project(":analysis:analysis-api-standalone"))
     testImplementation(projectTests(":analysis:analysis-api-standalone"))
+    testImplementation(projectTests(":analysis:analysis-test-framework"))
     testImplementation(projectTests(":native:native.tests"))
     testImplementation(project(":native:kotlin-native-utils"))
     testImplementation(project(":kotlin-native:utilities:basic-utils"))

analysis/analysis-api-standalone/src/org/jetbrains/kotlin/analysis/api/standalone/StandaloneAnalysisAPISessionBuilder.kt

@@ -44,7 +44,6 @@ import kotlin.contracts.InvocationKind
 import kotlin.contracts.contract
 
 public class StandaloneAnalysisAPISessionBuilder(
-    applicationDisposable: Disposable,
     projectDisposable: Disposable,
     unitTestMode: Boolean,
     classLoader: ClassLoader = MockProject::class.java.classLoader
@@ -61,7 +60,6 @@ public class StandaloneAnalysisAPISessionBuilder(
     private val kotlinCoreProjectEnvironment: KotlinCoreProjectEnvironment =
         StandaloneProjectFactory.createProjectEnvironment(
             projectDisposable,
-            applicationDisposable,
             KotlinCoreApplicationEnvironmentMode.fromUnitTestModeFlag(unitTestMode),
             classLoader = classLoader
         )
@@ -229,7 +227,6 @@ public class StandaloneAnalysisAPISessionBuilder(
 
 @OptIn(ExperimentalContracts::class)
 public inline fun buildStandaloneAnalysisAPISession(
-    applicationDisposable: Disposable = Disposer.newDisposable("StandaloneAnalysisAPISession.application"),
     projectDisposable: Disposable = Disposer.newDisposable("StandaloneAnalysisAPISession.project"),
     unitTestMode: Boolean = false,
     withPsiDeclarationFromBinaryModuleProvider: Boolean = false,
@@ -240,7 +237,6 @@ public inline fun buildStandaloneAnalysisAPISession(
         callsInPlace(init, InvocationKind.EXACTLY_ONCE)
     }
     return StandaloneAnalysisAPISessionBuilder(
-        applicationDisposable,
         projectDisposable,
         unitTestMode,
         classLoader

analysis/analysis-api-standalone/tests/org/jetbrains/kotlin/analysis/api/standalone/fir/test/cases/session/builder/AbstractStandaloneSessionBuilderAgainstStdlibTest.kt

@@ -13,6 +13,7 @@ import org.jetbrains.kotlin.analysis.api.standalone.buildStandaloneAnalysisAPISe
 import org.jetbrains.kotlin.analysis.project.structure.KtSourceModule
 import org.jetbrains.kotlin.analysis.project.structure.builder.buildKtLibraryModule
 import org.jetbrains.kotlin.analysis.project.structure.builder.buildKtSourceModule
+import org.jetbrains.kotlin.analysis.test.framework.TestWithDisposable
 import org.jetbrains.kotlin.builtins.StandardNames
 import org.jetbrains.kotlin.name.CallableId
 import org.jetbrains.kotlin.name.ClassId
@@ -28,13 +29,13 @@ import org.junit.jupiter.api.Assertions
 import java.nio.file.Path
 
 @OptIn(KtAnalysisApiInternals::class)
-abstract class AbstractStandaloneSessionBuilderAgainstStdlibTest {
+abstract class AbstractStandaloneSessionBuilderAgainstStdlibTest : TestWithDisposable() {
     protected fun doTestKotlinStdLibResolve(
         targetPlatform: TargetPlatform, platformStdlibPath: Path,
         additionalStdlibRoots: List<Path> = emptyList(),
     ) {
         lateinit var sourceModule: KtSourceModule
-        val session = buildStandaloneAnalysisAPISession {
+        val session = buildStandaloneAnalysisAPISession(disposable) {
             registerProjectService(KtLifetimeTokenProvider::class.java, KtAlwaysAccessibleLifetimeTokenProvider())
 
             buildKtModuleProvider {

analysis/analysis-api-standalone/tests/org/jetbrains/kotlin/analysis/api/standalone/fir/test/cases/session/builder/StandaloneSessionBuilderTest.kt

@@ -19,6 +19,7 @@ import org.jetbrains.kotlin.analysis.project.structure.KtSourceModule
 import org.jetbrains.kotlin.analysis.project.structure.builder.buildKtLibraryModule
 import org.jetbrains.kotlin.analysis.project.structure.builder.buildKtSdkModule
 import org.jetbrains.kotlin.analysis.project.structure.builder.buildKtSourceModule
+import org.jetbrains.kotlin.analysis.test.framework.TestWithDisposable
 import org.jetbrains.kotlin.name.CallableId
 import org.jetbrains.kotlin.name.ClassId
 import org.jetbrains.kotlin.name.FqName
@@ -35,11 +36,11 @@ import org.junit.jupiter.api.Test
 import java.nio.file.Paths
 
 @OptIn(KtAnalysisApiInternals::class)
-class StandaloneSessionBuilderTest {
+class StandaloneSessionBuilderTest : TestWithDisposable() {
     @Test
     fun testJdkSessionBuilder() {
         lateinit var sourceModule: KtSourceModule
-        val session = buildStandaloneAnalysisAPISession {
+        val session = buildStandaloneAnalysisAPISession(disposable) {
             registerProjectService(KtLifetimeTokenProvider::class.java, KtAlwaysAccessibleLifetimeTokenProvider())
 
             buildKtModuleProvider {
@@ -76,7 +77,7 @@ class StandaloneSessionBuilderTest {
     @Test
     fun testResolveAgainstCommonKlib() {
         lateinit var sourceModule: KtSourceModule
-        val session = buildStandaloneAnalysisAPISession {
+        val session = buildStandaloneAnalysisAPISession(disposable) {
             registerProjectService(KtLifetimeTokenProvider::class.java, KtAlwaysAccessibleLifetimeTokenProvider())
 
             buildKtModuleProvider {
@@ -108,7 +109,7 @@ class StandaloneSessionBuilderTest {
     @Test
     fun testKotlinSourceModuleSessionBuilder() {
         lateinit var sourceModule: KtSourceModule
-        val session = buildStandaloneAnalysisAPISession {
+        val session = buildStandaloneAnalysisAPISession(disposable) {
             registerProjectService(KtLifetimeTokenProvider::class.java, KtAlwaysAccessibleLifetimeTokenProvider())
 
             buildKtModuleProvider {

analysis/analysis-test-framework/tests/org/jetbrains/kotlin/analysis/test/framework/services/AnalysisApiEnvironmentManager.kt

@@ -12,7 +12,6 @@ import org.jetbrains.kotlin.analysis.api.standalone.base.project.structure.KtSta
 import org.jetbrains.kotlin.analysis.api.standalone.base.project.structure.StandaloneProjectFactory
 import org.jetbrains.kotlin.analysis.project.structure.KtBuiltinsModule
 import org.jetbrains.kotlin.analysis.test.framework.project.structure.ktModuleProvider
-import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisApiTestConfigurator
 import org.jetbrains.kotlin.cli.jvm.compiler.KotlinCoreApplicationEnvironment
 import org.jetbrains.kotlin.cli.jvm.compiler.KotlinCoreApplicationEnvironmentMode
 import org.jetbrains.kotlin.cli.jvm.compiler.KotlinCoreProjectEnvironment
@@ -44,7 +43,6 @@ class AnalysisApiEnvironmentManagerImpl(
     private val _projectEnvironment: KotlinCoreProjectEnvironment by lazy {
         StandaloneProjectFactory.createProjectEnvironment(
             testRootDisposable,
-            testServices.applicationDisposableProvider.getApplicationRootDisposable(),
             KotlinCoreApplicationEnvironmentMode.UnitTest,
         )
     }