Put activePartitioning in ci-jobs plugin (#9912)

* Put activePartitioning in ci-jobs

* Empty commit to re-run CI benchmarks

Author: sarahchen6

buildSrc/src/main/kotlin/datadog.ci-jobs.gradle.kts

@@ -5,6 +5,7 @@
  */
 
 import datadog.gradle.plugin.ci.findAffectedTaskPath
+import org.gradle.api.tasks.testing.Test
 import java.io.File
 import kotlin.math.abs
 
@@ -20,6 +21,15 @@ allprojects {
     val currentTaskPartition = abs(project.path.hashCode() % taskPartitionCount.toInt())
     extra.set("activePartition", currentTaskPartition == taskPartition.toInt())
   }
+  
+  // Disable test tasks if not in active partition
+  val activePartitionProvider = providers.provider {
+    project.extra.properties["activePartition"] as? Boolean ?: true
+  }
+  
+  tasks.withType<Test>().configureEach {
+    enabled = activePartitionProvider.get()
+  }
 }
 
 fun relativeToGitRoot(f: File): File {

buildSrc/src/main/kotlin/datadog.configure-tests.gradle.kts

@@ -32,14 +32,9 @@ val skipTestsProvider = rootProject.providers.gradleProperty("skipTests")
 val skipForkedTestsProvider = rootProject.providers.gradleProperty("skipForkedTests")
 val skipFlakyTestsProvider = rootProject.providers.gradleProperty("skipFlakyTests")
 val runFlakyTestsProvider = rootProject.providers.gradleProperty("runFlakyTests")
-val activePartitionProvider = providers.provider {
-  project.extra.properties["activePartition"] as? Boolean ?: true
-}
 
 // Go through the Test tasks and configure them
 tasks.withType<Test>().configureEach {
-  enabled = activePartitionProvider.get()
-  
   // Disable all tests if skipTests property was specified
   onlyIf("skipTests are undefined or false") { !skipTestsProvider.isPresent }