blob: 41d14d0841bd859d4479f193c6a9e8ac2bd407f1 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
/*
* Copyright 2016-2019 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
*/
package kotlinx.coroutines.test
import kotlinx.coroutines.*
import kotlin.coroutines.*
/**
* Access uncaught coroutine exceptions captured during test execution.
*/
@ExperimentalCoroutinesApi // Since 1.2.1, tentatively till 1.3.0
public interface UncaughtExceptionCaptor {
/**
* List of uncaught coroutine exceptions.
*
* The returned list is a copy of the currently caught exceptions.
* During [cleanupTestCoroutines] the first element of this list is rethrown if it is not empty.
*/
public val uncaughtExceptions: List<Throwable>
/**
* Call after the test completes to ensure that there were no uncaught exceptions.
*
* The first exception in uncaughtExceptions is rethrown. All other exceptions are
* printed using [Throwable.printStackTrace].
*
* @throws Throwable the first uncaught exception, if there are any uncaught exceptions.
*/
public fun cleanupTestCoroutines()
}
/**
* An exception handler that captures uncaught exceptions in tests.
*/
@ExperimentalCoroutinesApi // Since 1.2.1, tentatively till 1.3.0
public class TestCoroutineExceptionHandler :
AbstractCoroutineContextElement(CoroutineExceptionHandler), UncaughtExceptionCaptor, CoroutineExceptionHandler
{
private val _exceptions = mutableListOf<Throwable>()
/** @suppress **/
override fun handleException(context: CoroutineContext, exception: Throwable) {
synchronized(_exceptions) {
_exceptions += exception
}
}
/** @suppress **/
override val uncaughtExceptions
get() = synchronized(_exceptions) { _exceptions.toList() }
/** @suppress **/
override fun cleanupTestCoroutines() {
synchronized(_exceptions) {
val exception = _exceptions.firstOrNull() ?: return
// log the rest
_exceptions.drop(1).forEach { it.printStackTrace() }
throw exception
}
}
}
|
