[Test] Handle any Throwable from test instead of AssertionError

compiler/test-infrastructure-utils/tests/org/jetbrains/kotlin/test/Assertions.kt

@@ -44,7 +44,7 @@ abstract class Assertions {
         return collection.joinToString("\n")
     }
 
-    abstract fun assertAll(exceptions: List<AssertionError>)
+    abstract fun assertAll(exceptions: List<Throwable>)
 
     abstract fun fail(message: () -> String): Nothing
 }

compiler/test-infrastructure/tests/org/jetbrains/kotlin/test/ExceptionFromTestError.kt

@@ -8,4 +8,7 @@ package org.jetbrains.kotlin.test
 class ExceptionFromTestError(cause: Throwable) : AssertionError(cause) {
     override val message: String
         get() = "Exception was thrown"
+
+    override val cause: Throwable
+        get() = super.cause!!
 }

compiler/test-infrastructure/tests/org/jetbrains/kotlin/test/TestRunner.kt

@@ -12,7 +12,7 @@ import org.jetbrains.kotlin.test.services.*
 import java.io.IOException
 
 class TestRunner(private val testConfiguration: TestConfiguration) {
-    private val failedAssertions = mutableListOf<AssertionError>()
+    private val failedAssertions = mutableListOf<Throwable>()
 
     fun runTest(@TestDataFile testDataFileName: String) {
         try {
@@ -79,9 +79,10 @@ class TestRunner(private val testConfiguration: TestConfiguration) {
         }
 
         val filteredFailedAssertions = testConfiguration.afterAnalysisCheckers
-            .fold<AfterAnalysisChecker, List<AssertionError>>(failedAssertions) { assertions, checker ->
+            .fold<AfterAnalysisChecker, List<Throwable>>(failedAssertions) { assertions, checker ->
                 checker.suppressIfNeeded(assertions)
             }
+            .map { if (it is ExceptionFromTestError) it.cause else it }
 
         services.assertions.assertAll(filteredFailedAssertions)
     }
@@ -134,7 +135,7 @@ class TestRunner(private val testConfiguration: TestConfiguration) {
     private inline fun withAssertionCatching(insertExceptionInStart: Boolean = false, block: () -> Unit) {
         try {
             block()
-        } catch (e: AssertionError) {
+        } catch (e: Throwable) {
             if (insertExceptionInStart) {
                 failedAssertions.add(0, e)
             } else {

compiler/test-infrastructure/tests/org/jetbrains/kotlin/test/model/AfterAnalysisChecker.kt

@@ -12,7 +12,7 @@ abstract class AfterAnalysisChecker(protected val testServices: TestServices) {
     open val directives: List<DirectivesContainer>
         get() = emptyList()
 
-    open fun check(failedAssertions: List<AssertionError>) {}
+    open fun check(failedAssertions: List<Throwable>) {}
 
-    open fun suppressIfNeeded(failedAssertions: List<AssertionError>): List<AssertionError> = failedAssertions
+    open fun suppressIfNeeded(failedAssertions: List<Throwable>): List<Throwable> = failedAssertions
 }

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/backend/BlackBoxCodegenSuppressor.kt

@@ -22,7 +22,7 @@ class BlackBoxCodegenSuppressor(testServices: TestServices) : AfterAnalysisCheck
     override val directives: List<DirectivesContainer>
         get() = listOf(CodegenTestDirectives)
 
-    override fun suppressIfNeeded(failedAssertions: List<AssertionError>): List<AssertionError> {
+    override fun suppressIfNeeded(failedAssertions: List<Throwable>): List<Throwable> {
         val moduleStructure = testServices.moduleStructure
         val targetBackends = moduleStructure.modules.mapNotNull { it.targetBackend }
         return when (moduleStructure.modules.map { it.frontendKind }.first()) {
@@ -36,8 +36,8 @@ class BlackBoxCodegenSuppressor(testServices: TestServices) : AfterAnalysisCheck
         moduleStructure: TestModuleStructure,
         directive: ValueDirective<TargetBackend>,
         targetBackends: List<TargetBackend>,
-        failedAssertions: List<AssertionError>
+        failedAssertions: List<Throwable>
-    ): List<AssertionError> {
+    ): List<Throwable> {
         val ignoredBackends = moduleStructure.allDirectives[directive]
         if (ignoredBackends.isEmpty()) return failedAssertions
         val matchedBackend = ignoredBackends.intersect(targetBackends)
@@ -52,10 +52,10 @@ class BlackBoxCodegenSuppressor(testServices: TestServices) : AfterAnalysisCheck
 
 
     private fun processAssertions(
-        failedAssertions: List<AssertionError>,
+        failedAssertions: List<Throwable>,
         directive: ValueDirective<TargetBackend>,
         additionalMessage: String = ""
-    ): List<AssertionError> {
+    ): List<Throwable> {
         return if (failedAssertions.isNotEmpty()) emptyList()
         else {
             val message = buildString {

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/backend/handlers/FirIrDumpIdenticalCheckers.kt

@@ -39,7 +39,7 @@ class FirIrDumpIdenticalChecker(testServices: TestServices) : AfterAnalysisCheck
         }
     }
 
-    override fun check(failedAssertions: List<AssertionError>) {
+    override fun check(failedAssertions: List<Throwable>) {
         if (failedAssertions.isNotEmpty()) return
         val testDataFile = testServices.moduleStructure.originalTestDataFiles.first()
         if (FIR_IDENTICAL in testServices.moduleStructure.allDirectives) {

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/frontend/classic/handlers/FirTestDataConsistencyHandler.kt

@@ -21,7 +21,7 @@ class FirTestDataConsistencyHandler(testServices: TestServices) : AfterAnalysisC
     override val directives: List<DirectivesContainer>
         get() = listOf(FirDiagnosticsDirectives)
 
-    override fun check(failedAssertions: List<AssertionError>) {
+    override fun check(failedAssertions: List<Throwable>) {
         val moduleStructure = testServices.moduleStructure
         val testData = moduleStructure.originalTestDataFiles.first()
         if (testData.extension == "kts") return

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/frontend/fir/FirFailingTestSuppressor.kt

@@ -11,7 +11,7 @@ import org.jetbrains.kotlin.test.services.TestServices
 import org.jetbrains.kotlin.test.services.moduleStructure
 
 class FirFailingTestSuppressor(testServices: TestServices) : AfterAnalysisChecker(testServices) {
-    override fun suppressIfNeeded(failedAssertions: List<AssertionError>): List<AssertionError> {
+    override fun suppressIfNeeded(failedAssertions: List<Throwable>): List<Throwable> {
         val testFile = testServices.moduleStructure.originalTestDataFiles.first()
         val failFile = testFile.parentFile.resolve("${testFile.nameWithoutExtension}.fail")
         val exceptionFromFir = failedAssertions.firstOrNull { it is ExceptionFromTestError }

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/frontend/fir/handlers/FirIdenticalChecker.kt

@@ -25,7 +25,7 @@ class FirIdenticalChecker(testServices: TestServices) : AfterAnalysisChecker(tes
         }
     }
 
-    override fun check(failedAssertions: List<AssertionError>) {
+    override fun check(failedAssertions: List<Throwable>) {
         if (failedAssertions.isNotEmpty()) return
         val testDataFile = testServices.moduleStructure.originalTestDataFiles.first()
         if (testDataFile.isFirTestData) {

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/services/JUnit5Assertions.kt

@@ -51,7 +51,7 @@ object JUnit5Assertions : AssertionsService() {
         JUnit5PlatformAssertions.assertFalse(value, message?.invoke())
     }
 
-    override fun assertAll(exceptions: List<AssertionError>) {
+    override fun assertAll(exceptions: List<Throwable>) {
         exceptions.singleOrNull()?.let { throw it }
         JUnit5PlatformAssertions.assertAll(exceptions.map { Executable { throw it } })
     }

compiler/tests-common/tests/org/jetbrains/kotlin/test/util/JUnit4Assertions.kt

@@ -40,7 +40,7 @@ object JUnit4Assertions : Assertions() {
         KtUsefulTestCase.assertSameElements(message?.invoke() ?: "", expected, actual)
     }
 
-    override fun assertAll(exceptions: List<AssertionError>) {
+    override fun assertAll(exceptions: List<Throwable>) {
         exceptions.forEach { throw it }
     }