diff options
| -rw-r--r-- | app/src/test/kotlin/com/pitchedapps/frost/views/FrostContentViewAsyncTest.kt | 62 |
1 files changed, 41 insertions, 21 deletions
diff --git a/app/src/test/kotlin/com/pitchedapps/frost/views/FrostContentViewAsyncTest.kt b/app/src/test/kotlin/com/pitchedapps/frost/views/FrostContentViewAsyncTest.kt index a179fb98..13c40aa3 100644 --- a/app/src/test/kotlin/com/pitchedapps/frost/views/FrostContentViewAsyncTest.kt +++ b/app/src/test/kotlin/com/pitchedapps/frost/views/FrostContentViewAsyncTest.kt @@ -1,19 +1,22 @@ package com.pitchedapps.frost.views import kotlinx.coroutines.Dispatchers -import kotlinx.coroutines.ExecutorCoroutineDispatcher import kotlinx.coroutines.ExperimentalCoroutinesApi import kotlinx.coroutines.asCoroutineDispatcher import kotlinx.coroutines.async import kotlinx.coroutines.channels.BroadcastChannel import kotlinx.coroutines.channels.ReceiveChannel +import kotlinx.coroutines.channels.count +import kotlinx.coroutines.joinAll +import kotlinx.coroutines.launch import kotlinx.coroutines.runBlocking import kotlinx.coroutines.withContext import java.util.concurrent.Executors -import kotlin.test.AfterTest -import kotlin.test.BeforeTest +import kotlin.coroutines.EmptyCoroutineContext import kotlin.test.Test import kotlin.test.assertEquals +import kotlin.test.assertFalse +import kotlin.test.assertTrue /** * Collection of tests around the view thread logic @@ -22,24 +25,6 @@ import kotlin.test.assertEquals class FrostContentViewAsyncTest { /** - * Single threaded dispatcher with thread name "main" - * Mimics the usage of Android's main dispatcher - */ - private lateinit var mainDispatcher: ExecutorCoroutineDispatcher - - @BeforeTest - fun before() { - mainDispatcher = Executors.newSingleThreadExecutor { r -> - Thread(r, "main") - }.asCoroutineDispatcher() - } - - @AfterTest - fun after() { - mainDispatcher.close() - } - - /** * Hooks onto the refresh channel for one true -> false cycle. * Returns the list of event ids that were emitted */ @@ -107,4 +92,39 @@ class FrostContentViewAsyncTest { ) } } + + /** + * Sanity check to ensure that contexts are being honoured + */ + @Test + fun contextSwitching() { + val mainTag = "main-test" + val mainDispatcher = Executors.newSingleThreadExecutor { r -> + Thread(r, mainTag) + }.asCoroutineDispatcher() + + val channel = BroadcastChannel<String>(100) + + runBlocking(Dispatchers.IO) { + val receiver1 = channel.openSubscription() + val receiver2 = channel.openSubscription() + launch(mainDispatcher) { + for (thread in receiver1) { + assertTrue( + Thread.currentThread().name.startsWith(mainTag), + "Channel should be received in main thread" + ) + assertFalse( + thread.startsWith(mainTag), + "Channel execution should not be in main thread" + ) + } + } + listOf(EmptyCoroutineContext, Dispatchers.IO, Dispatchers.Default, Dispatchers.IO).map { + async(it) { channel.send(Thread.currentThread().name) } + }.joinAll() + channel.close() + assertEquals(4, receiver2.count(), "Not all events received") + } + } }
\ No newline at end of file |