aboutsummaryrefslogtreecommitdiffstats
path: root/kotlinx-coroutines-core/jvm/test/DebugThreadNameTest.kt
blob: 9d683960f06fd558e93bfc7377b12bc73cc68c92 (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
64
65
66
67
68
69
70
71
72
73
/*
 * Copyright 2016-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
 */

package kotlinx.coroutines

import kotlin.test.*

class DebugThreadNameTest : TestBase() {
    @BeforeTest
    fun resetName() {
        resetCoroutineId()
    }

    @Test
    fun testLaunchId() = runTest {
        assertName("coroutine#1")
        launch {
            assertName("coroutine#2")
            yield()
            assertName("coroutine#2")
        }
        assertName("coroutine#1")
    }

    @Test
    fun testLaunchIdUndispatched() = runTest {
        assertName("coroutine#1")
        launch(start = CoroutineStart.UNDISPATCHED) {
            assertName("coroutine#2")
            yield()
            assertName("coroutine#2")
        }
        assertName("coroutine#1")
    }

    @Test
    fun testLaunchName() = runTest {
        assertName("coroutine#1")
        launch(CoroutineName("TEST")) {
            assertName("TEST#2")
            yield()
            assertName("TEST#2")
        }
        assertName("coroutine#1")
    }

    @Test
    fun testWithContext() = runTest {
        assertName("coroutine#1")
        withContext(Dispatchers.Default) {
            assertName("coroutine#1")
            yield()
            assertName("coroutine#1")
            withContext(CoroutineName("TEST")) {
                assertName("TEST#1")
                yield()
                assertName("TEST#1")
            }
            assertName("coroutine#1")
            yield()
            assertName("coroutine#1")
        }
        assertName("coroutine#1")
    }

    private fun assertName(expected: String) {
        val name = Thread.currentThread().name
        val split = name.split(Regex(" @"))
        assertEquals(2, split.size, "Thread name '$name' is expected to contain one coroutine name")
        assertEquals(expected, split[1], "Thread name '$name' is expected to end with coroutine name '$expected'")
    }
}