From 086133ab8276a54f71403d78a7ba24a42e168710 Mon Sep 17 00:00:00 2001
From: Trask Stalnaker <trask.stalnaker@gmail.com>
Date: Wed, 19 Nov 2025 18:17:33 -0800
Subject: [PATCH] Gradle config cache: listTestsInPartition

---
 build.gradle.kts | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/build.gradle.kts b/build.gradle.kts
index e5a931272e71..cd7a981ec234 100644
--- a/build.gradle.kts
+++ b/build.gradle.kts
@@ -76,7 +76,7 @@ if (gradle.startParameter.taskNames.contains("listTestsInPartition")) {
         throw GradleException("Invalid test partition")
       }
 
-      val partitionTasks = ArrayList<Test>()
+      val partitionTasks = ArrayList<String>()
       var testPartitionCounter = 0
       subprojects {
         // relying on predictable ordering of subprojects
@@ -84,16 +84,17 @@ if (gradle.startParameter.taskNames.contains("listTestsInPartition")) {
         // since we are splitting these tasks across different github action jobs
         val enabled = testPartitionCounter++ % 4 == testPartition
         if (enabled) {
+          val projectPath = this.path
           tasks.withType<Test>().configureEach {
-            partitionTasks.add(this)
+            val taskPath = projectPath + ":" + this.name
+            partitionTasks.add(taskPath)
           }
         }
       }
 
       doLast {
         File("test-tasks.txt").printWriter().use { writer ->
-          partitionTasks.forEach { task ->
-            var taskPath = task.project.path + ":" + task.name
+          partitionTasks.forEach { taskPath ->
             // smoke tests are run separately
             // :instrumentation:test runs all instrumentation tests
             if (taskPath != ":smoke-tests:test" && taskPath != ":instrumentation:test") {