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