From aea09bc91a231d045885c9be2d544ff45a4092ec Mon Sep 17 00:00:00 2001
From: Nikita Klimenko <nikita.klimenko@jetbrains.com>
Date: Thu, 27 Jun 2024 19:40:16 +0300
Subject: [PATCH] Enable parallel forks for tests

---
 core/build.gradle.kts                            | 5 +++++
 plugins/dataframe-gradle-plugin/build.gradle.kts | 1 +
 2 files changed, 6 insertions(+)

diff --git a/core/build.gradle.kts b/core/build.gradle.kts
index 82a7f82161..a98ba17305 100644
--- a/core/build.gradle.kts
+++ b/core/build.gradle.kts
@@ -364,6 +364,7 @@ tasks.withType<KotlinCompile> {
 
 tasks.test {
     maxHeapSize = "2048m"
+    maxParallelForks = (Runtime.getRuntime().availableProcessors() / 2).coerceAtLeast(1)
     extensions.configure(kotlinx.kover.api.KoverTaskExtension::class) {
         excludes.set(
             listOf(
@@ -374,6 +375,10 @@ tasks.test {
     }
 }
 
+tasks.getByName<Test>("samplesTest") {
+    maxParallelForks = 1
+}
+
 tasks.processJupyterApiResources {
     libraryProducers = listOf("org.jetbrains.kotlinx.dataframe.jupyter.Integration")
 }
diff --git a/plugins/dataframe-gradle-plugin/build.gradle.kts b/plugins/dataframe-gradle-plugin/build.gradle.kts
index a134075364..6cc376494d 100644
--- a/plugins/dataframe-gradle-plugin/build.gradle.kts
+++ b/plugins/dataframe-gradle-plugin/build.gradle.kts
@@ -123,6 +123,7 @@ val integrationTestTask = task<Test>("integrationTest") {
     dependsOn(":core:publishCorePublicationToMavenLocal")
     description = "Runs integration tests."
     group = "verification"
+    maxParallelForks = (Runtime.getRuntime().availableProcessors() / 2).coerceAtLeast(1)
 
     testClassesDirs = sourceSets["integrationTest"].output.classesDirs
     classpath = sourceSets["integrationTest"].runtimeClasspath