Roman Elizarov | b170819 | 2017-12-22 12:14:05 +0300 | [diff] [blame] | 1 | /* |
Roman Elizarov | 1f74a2d | 2018-06-29 19:19:45 +0300 | [diff] [blame] | 2 | * Copyright 2016-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license. |
Roman Elizarov | b170819 | 2017-12-22 12:14:05 +0300 | [diff] [blame] | 3 | */ |
| 4 | |
| 5 | package kotlinx.coroutines.experimental |
| 6 | |
Roman Elizarov | 9fe5f46 | 2018-02-21 19:05:52 +0300 | [diff] [blame] | 7 | import kotlin.coroutines.experimental.* |
| 8 | import kotlin.js.* |
Roman Elizarov | b170819 | 2017-12-22 12:14:05 +0300 | [diff] [blame] | 9 | import kotlin.test.* |
| 10 | |
Roman Elizarov | b170819 | 2017-12-22 12:14:05 +0300 | [diff] [blame] | 11 | class PromiseTest : TestBase() { |
| 12 | @Test |
Roman Elizarov | 592de52 | 2018-09-10 20:06:38 +0300 | [diff] [blame^] | 13 | fun testPromiseResolvedAsDeferred() = GlobalScope.promise { |
Roman Elizarov | b170819 | 2017-12-22 12:14:05 +0300 | [diff] [blame] | 14 | val promise = Promise<String> { resolve, _ -> |
| 15 | resolve("OK") |
| 16 | } |
| 17 | val deferred = promise.asDeferred() |
| 18 | assertEquals("OK", deferred.await()) |
| 19 | } |
| 20 | |
| 21 | @Test |
Roman Elizarov | 592de52 | 2018-09-10 20:06:38 +0300 | [diff] [blame^] | 22 | fun testPromiseRejectedAsDeferred() = GlobalScope.promise { |
Roman Elizarov | 1bb8c1c | 2018-01-11 10:39:46 +0300 | [diff] [blame] | 23 | lateinit var promiseReject: (Throwable) -> Unit |
Roman Elizarov | b170819 | 2017-12-22 12:14:05 +0300 | [diff] [blame] | 24 | val promise = Promise<String> { _, reject -> |
Roman Elizarov | 1bb8c1c | 2018-01-11 10:39:46 +0300 | [diff] [blame] | 25 | promiseReject = reject |
Roman Elizarov | b170819 | 2017-12-22 12:14:05 +0300 | [diff] [blame] | 26 | } |
| 27 | val deferred = promise.asDeferred() |
Roman Elizarov | 1bb8c1c | 2018-01-11 10:39:46 +0300 | [diff] [blame] | 28 | // reject after converting to deferred to avoid "Unhandled promise rejection" warnings |
| 29 | promiseReject(TestException("Rejected")) |
Roman Elizarov | b170819 | 2017-12-22 12:14:05 +0300 | [diff] [blame] | 30 | try { |
| 31 | deferred.await() |
| 32 | expectUnreached() |
| 33 | } catch (e: Throwable) { |
| 34 | assertTrue(e is TestException) |
| 35 | assertEquals("Rejected", e.message) |
| 36 | } |
| 37 | } |
| 38 | |
| 39 | @Test |
Roman Elizarov | 592de52 | 2018-09-10 20:06:38 +0300 | [diff] [blame^] | 40 | fun testCompletedDeferredAsPromise() = GlobalScope.promise { |
Roman Elizarov | b170819 | 2017-12-22 12:14:05 +0300 | [diff] [blame] | 41 | val deferred = async(coroutineContext, CoroutineStart.UNDISPATCHED) { |
| 42 | // completed right away |
| 43 | "OK" |
| 44 | } |
| 45 | val promise = deferred.asPromise() |
| 46 | assertEquals("OK", promise.await()) |
| 47 | } |
| 48 | |
| 49 | @Test |
Roman Elizarov | 592de52 | 2018-09-10 20:06:38 +0300 | [diff] [blame^] | 50 | fun testWaitForDeferredAsPromise() = GlobalScope.promise { |
Roman Elizarov | b170819 | 2017-12-22 12:14:05 +0300 | [diff] [blame] | 51 | val deferred = async(coroutineContext) { |
| 52 | // will complete later |
| 53 | "OK" |
| 54 | } |
| 55 | val promise = deferred.asPromise() |
| 56 | assertEquals("OK", promise.await()) // await yields main thread to deferred coroutine |
| 57 | } |
| 58 | |
| 59 | @Test |
Roman Elizarov | 592de52 | 2018-09-10 20:06:38 +0300 | [diff] [blame^] | 60 | fun testCancellableAwaitPromise() = GlobalScope.promise { |
Roman Elizarov | b170819 | 2017-12-22 12:14:05 +0300 | [diff] [blame] | 61 | lateinit var r: (String) -> Unit |
| 62 | val toAwait = Promise<String> { resolve, _ -> r = resolve } |
| 63 | val job = launch(coroutineContext, CoroutineStart.UNDISPATCHED) { |
| 64 | toAwait.await() // suspends |
| 65 | } |
| 66 | job.cancel() // cancel the job |
| 67 | r("fail") // too late, the waiting job was already cancelled |
| 68 | } |
| 69 | |
| 70 | @Test |
Roman Elizarov | 592de52 | 2018-09-10 20:06:38 +0300 | [diff] [blame^] | 71 | fun testAsPromiseAsDeferred() = GlobalScope.promise { |
Roman Elizarov | b170819 | 2017-12-22 12:14:05 +0300 | [diff] [blame] | 72 | val deferred = async { "OK" } |
| 73 | val promise = deferred.asPromise() |
| 74 | val d2 = promise.asDeferred() |
| 75 | assertTrue(d2 === deferred) |
| 76 | assertEquals("OK", d2.await()) |
| 77 | } |
| 78 | |
| 79 | private class TestException(message: String) : Exception(message) |
| 80 | } |