Gradle, testing test execution: support asserting reports from multiple tasks at once

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

@@ -602,25 +602,30 @@ abstract class BaseGradleIT {
      */
     fun CompiledProject.assertTestResults(
         @TestDataFile assertionFileName: String,
-            testReportName: String
+        vararg testReportNames: String
     ) {
         val projectDir = project.projectDir
-        val testReportDir = projectDir.resolve("build/test-results/$testReportName")
+        val testReportDirs = testReportNames.map { projectDir.resolve("build/test-results/$it") }
 
-        if (!testReportDir.isDirectory) {
+        testReportDirs.forEach {
-            error("Test report dir $testReportDir was not created")
+            if (!it.isDirectory) {
+                error("Test report dir $it was not created")
+            }
         }
 
-        val actualTestResults = readAndCleanupTestResults(testReportDir, projectDir)
+        val actualTestResults = readAndCleanupTestResults(testReportDirs, projectDir)
         val expectedTestResults = prettyPrintXml(resourcesRootFile.resolve(assertionFileName).readText())
 
         assertEquals(expectedTestResults, actualTestResults)
     }
 
-    private fun readAndCleanupTestResults(testReportDir: File, projectDir: File): String {
+    private fun readAndCleanupTestResults(testReportDirs: List<File>, projectDir: File): String {
-        val files = testReportDir
+        val files = testReportDirs
-                .listFiles()
+            .flatMap {
-                .filter { it.isFile && it.name.endsWith(".xml") }
+                (it.listFiles() ?: arrayOf()).filter {
+                    it.isFile && it.name.endsWith(".xml")
+                }
+            }
             .sortedBy {
                 // let containing test suite be first
                 it.name.replace(".xml", ".A.xml")