| /* |
| * 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 NonCancellableTest : TestBase() { |
| @Test |
| fun testNonCancellable() = runTest { |
| expect(1) |
| val job = async { |
| withContext(NonCancellable) { |
| expect(2) |
| yield() |
| expect(4) |
| } |
| |
| expect(5) |
| yield() |
| expectUnreached() |
| } |
| |
| yield() |
| job.cancel() |
| expect(3) |
| assertTrue(job.isCancelled) |
| try { |
| job.await() |
| expectUnreached() |
| } catch (e: JobCancellationException) { |
| if (RECOVER_STACK_TRACES) { |
| val cause = e.cause as JobCancellationException // shall be recovered JCE |
| assertNull(cause.cause) |
| } else { |
| assertNull(e.cause) |
| } |
| finish(6) |
| } |
| } |
| |
| @Test |
| fun testNonCancellableWithException() = runTest { |
| expect(1) |
| val deferred = async(NonCancellable) { |
| withContext(NonCancellable) { |
| expect(2) |
| yield() |
| expect(4) |
| } |
| |
| expect(5) |
| yield() |
| expectUnreached() |
| } |
| |
| yield() |
| deferred.cancel(TestCancellationException("TEST")) |
| expect(3) |
| assertTrue(deferred.isCancelled) |
| try { |
| deferred.await() |
| expectUnreached() |
| } catch (e: TestCancellationException) { |
| assertEquals("TEST", e.message) |
| finish(6) |
| } |
| } |
| |
| @Test |
| fun testNonCancellableFinally() = runTest { |
| expect(1) |
| val job = async { |
| try { |
| expect(2) |
| yield() |
| expectUnreached() |
| } finally { |
| withContext(NonCancellable) { |
| expect(4) |
| yield() |
| expect(5) |
| } |
| } |
| |
| expectUnreached() |
| } |
| |
| yield() |
| job.cancel() |
| expect(3) |
| assertTrue(job.isCancelled) |
| |
| try { |
| job.await() |
| expectUnreached() |
| } catch (e: CancellationException) { |
| finish(6) |
| } |
| } |
| |
| @Test |
| fun testNonCancellableTwice() = runTest { |
| expect(1) |
| val job = async { |
| withContext(NonCancellable) { |
| expect(2) |
| yield() |
| expect(4) |
| } |
| |
| withContext(NonCancellable) { |
| expect(5) |
| yield() |
| expect(6) |
| } |
| } |
| |
| yield() |
| job.cancel() |
| expect(3) |
| assertTrue(job.isCancelled) |
| try { |
| job.await() |
| expectUnreached() |
| } catch (e: JobCancellationException) { |
| if (RECOVER_STACK_TRACES) { |
| val cause = e.cause as JobCancellationException // shall be recovered JCE |
| assertNull(cause.cause) |
| } else { |
| assertNull(e.cause) |
| } |
| finish(7) |
| } |
| } |
| } |