Workaround coroutines hang due to Gradle dependencies downgrading stdlib

Since 2.2.20 uncaught exceptions in coroutines make the -ea executions
hang if the stdlib is downgraded without appropriate -api-version
downgrade.

In KGP-IT Test process we don't want stdlib downgrade, so this change filters
out the downgraded stdlib. In other Gradle-related tests we can just
disable coroutines stacktrace recovery if embedded stdlib is present.

^KT-79528

Author: timofey-solonin

libraries/tools/kotlin-gradle-plugin-integration-tests/build.gradle.kts

@@ -528,3 +528,5 @@ tasks.withType<Test>().configureEach {
         })
     }
 }
+
+excludeGradleEmbeddedStdlibFromTestTasksRuntimeClasspath()

libraries/tools/kotlin-gradle-plugin-integration-tests/src/test/kotlin/org/jetbrains/kotlin/gradle/KT79528StdlibDowngradeHangs.kt

@@ -0,0 +1,22 @@
+package org.jetbrains.kotlin.gradle
+
+import kotlin.test.Test
+import kotlinx.coroutines.async
+import kotlinx.coroutines.runBlocking
+import org.jetbrains.kotlin.util.assertThrows
+
+class KT79528StdlibDowngradeHangs {
+
+    @Test
+    fun `test KT-79528 - test Gradle doesn't downgrade stdlib and cause uncaught exceptions in coroutines to hang coroutines machinery`() {
+        class Foo : RuntimeException()
+        assertThrows<Foo> {
+            runBlocking {
+                async {
+                    throw Foo()
+                }.await()
+            }
+        }
+    }
+
+}

libraries/tools/kotlin-gradle-plugin/src/functionalTest/kotlin/org/jetbrains/kotlin/gradle/unitTests/KT79528StdlibDowngradeHangs.kt

@@ -0,0 +1,21 @@
+package org.jetbrains.kotlin.gradle.unitTests
+
+import kotlin.test.Test
+import kotlinx.coroutines.async
+import kotlinx.coroutines.runBlocking
+import org.jetbrains.kotlin.util.assertThrows
+
+class KT79528StdlibDowngradeHangs {
+
+    @Test
+    fun `test KT-79528 - test Gradle doesn't downgrade stdlib and cause uncaught exceptions in coroutines to hang coroutines machinery`() {
+        class Foo : RuntimeException()
+        assertThrows<Foo> {
+            runBlocking {
+                async {
+                    throw Foo()
+                }.await()
+            }
+        }
+    }
+}

repo/gradle-build-conventions/buildsrc-compat/src/main/kotlin/GradleCommon.kt

@@ -10,6 +10,7 @@ import org.gradle.api.Action
 import org.gradle.api.GradleException
 import org.gradle.api.Project
 import org.gradle.api.artifacts.Configuration
+import org.gradle.api.artifacts.FileCollectionDependency
 import org.gradle.api.artifacts.ModuleDependency
 import org.gradle.api.artifacts.VersionCatalogsExtension
 import org.gradle.api.artifacts.type.ArtifactTypeDefinition
@@ -31,6 +32,7 @@ import org.gradle.api.tasks.Copy
 import org.gradle.api.tasks.SourceSet
 import org.gradle.api.tasks.TaskProvider
 import org.gradle.api.tasks.compile.JavaCompile
+import org.gradle.api.tasks.testing.Test
 import org.gradle.jvm.tasks.Jar
 import org.gradle.jvm.toolchain.JavaToolchainService
 import org.gradle.kotlin.dsl.*
@@ -39,6 +41,7 @@ import org.gradle.plugin.devel.PluginDeclaration
 import org.gradle.plugin.devel.plugins.JavaGradlePluginPlugin
 import org.gradle.plugin.devel.tasks.ValidatePlugins
 import org.gradle.plugin.use.resolve.internal.ArtifactRepositoriesPluginResolver.PLUGIN_MARKER_SUFFIX
+import org.gradle.process.CommandLineArgumentProvider
 import org.gradle.util.GradleVersion
 import org.jetbrains.kotlin.buildtools.api.ExperimentalBuildToolsApi
 import org.jetbrains.kotlin.gradle.ExperimentalKotlinGradlePluginApi
@@ -51,6 +54,7 @@ import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
 import org.jetbrains.kotlin.gradle.tasks.KotlinCompilerExecutionStrategy
 import plugins.configureDefaultPublishing
 import plugins.configureKotlinPomAttributes
+import java.io.File
 
 /**
  * We have to handle the returned provider lazily, because the publication's artifactId
@@ -808,4 +812,33 @@ fun Project.registerKotlinSourceForVersionRange(
                 }
             }
         }
-}
+}
+
+// See KT-79528 and https://github.com/gradle/gradle/issues/34453
+fun Project.disableCoroutinesStacktraceRecoveryInTestsIfGradleEmbeddedStdlibIsInRuntimeClasspath() {
+    tasks.withType<Test>().configureEach {
+        val isEmbeddedStdlibInRuntimeClasspath = provider {
+            classpath.contains(gradleEmbeddedStdlib())
+        }
+        jvmArgumentProviders.add {
+            if (isEmbeddedStdlibInRuntimeClasspath.get()) {
+                listOf("-Dkotlinx.coroutines.stacktrace.recovery=false")
+            } else {
+                emptyList()
+            }
+        }
+    }
+}
+
+fun Project.excludeGradleEmbeddedStdlibFromTestTasksRuntimeClasspath() {
+    afterEvaluate {
+        tasks.withType<Test>().configureEach {
+            val embeddedStdlib = gradleEmbeddedStdlib()
+            classpath = classpath.filter {
+                it != embeddedStdlib
+            }
+        }
+    }
+}
+
+private fun Project.gradleEmbeddedStdlib(): File = (dependencies.gradleApi() as FileCollectionDependency).files.single { it.name.startsWith("kotlin-stdlib") }

repo/gradle-build-conventions/buildsrc-compat/src/main/kotlin/gradle-plugin-common-configuration.gradle.kts

@@ -123,3 +123,5 @@ if (!kotlinBuildProperties.isInJpsBuildIdeaSync) {
     )
     publishShadowedJar(gradle813SourceSet, commonSourceSet)
 }
+
+disableCoroutinesStacktraceRecoveryInTestsIfGradleEmbeddedStdlibIsInRuntimeClasspath()

repo/gradle-build-conventions/buildsrc-compat/src/main/kotlin/gradle-plugin-dependency-configuration.gradle.kts

@@ -86,3 +86,4 @@ publishing {
     }
 }
 
+disableCoroutinesStacktraceRecoveryInTestsIfGradleEmbeddedStdlibIsInRuntimeClasspath()