1 files changed, 73 insertions, 0 deletions
diff --git a/kotlinx-coroutines-core/jvm/test/exceptions/Exceptions.kt b/kotlinx-coroutines-core/jvm/test/exceptions/Exceptions.kt
new file mode 100644
index 00000000..13023e31
--- /dev/null
+++ b/kotlinx-coroutines-core/jvm/test/exceptions/Exceptions.kt
@@ -0,0 +1,73 @@
+/*
+ * Copyright 2016-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+package kotlinx.coroutines.exceptions
+
+import kotlinx.coroutines.*
+import java.io.*
+import java.util.*
+import kotlin.coroutines.*
+import kotlin.test.*
+
+/**
+ * Proxy for [Throwable.getSuppressed] for tests, which are compiled for both JDK 1.6 and JDK 1.8,
+ * but run only under JDK 1.8
+ */
+@Suppress("ConflictingExtensionProperty")
+val Throwable.suppressed: Array<Throwable> get() {
+    val method = this::class.java.getMethod("getSuppressed") ?: error("This test can only be run using JDK 1.7")
+    @Suppress("UNCHECKED_CAST")
+    return method.invoke(this) as Array<Throwable>
+}
+
+internal inline fun <reified T : Throwable> checkException(exception: Throwable): Boolean {
+    assertTrue(exception is T)
+    assertTrue(exception.suppressed.isEmpty())
+    assertNull(exception.cause)
+    return true
+}
+
+internal fun checkCycles(t: Throwable) {
+    val sw = StringWriter()
+    t.printStackTrace(PrintWriter(sw))
+    assertFalse(sw.toString().contains("CIRCULAR REFERENCE"))
+}
+
+class CapturingHandler : AbstractCoroutineContextElement(CoroutineExceptionHandler),
+    CoroutineExceptionHandler
+{
+    private var unhandled: ArrayList<Throwable>? = ArrayList()
+
+    override fun handleException(context: CoroutineContext, exception: Throwable) = synchronized<Unit>(this) {
+        unhandled!!.add(exception)
+    }
+
+    fun getExceptions(): List<Throwable> = synchronized(this) {
+        return unhandled!!.also { unhandled = null }
+    }
+
+    fun getException(): Throwable = synchronized(this) {
+        val size = unhandled!!.size
+        assert(size == 1) { "Expected one unhandled exception, but have $size: $unhandled" }
+        return unhandled!![0].also { unhandled = null }
+    }
+}
+
+internal fun captureExceptionsRun(
+    context: CoroutineContext = EmptyCoroutineContext,
+    block: suspend CoroutineScope.() -> Unit
+): Throwable {
+    val handler = CapturingHandler()
+    runBlocking(context + handler, block = block)
+    return handler.getException()
+}
+
+internal fun captureMultipleExceptionsRun(
+    context: CoroutineContext = EmptyCoroutineContext,
+    block: suspend CoroutineScope.() -> Unit
+): List<Throwable> {
+    val handler = CapturingHandler()
+    runBlocking(context + handler, block = block)
+    return handler.getExceptions()
+}