msb-public
/
nowinandroid
mirror of https://github.com/android/nowinandroid.git
1
0
Code Issues Projects Releases Wiki Activity

Merge "Update Main dispatcher rule, assert on StateFlow's value property" into main

Browse Source
pull/152/head
Jolanda Verhoef 2 years ago committed by Gerrit Code Review
parent b32be52649 e220c66329
commit 70806f6104
5 changed files with 1452 additions and 1571 deletions
Show Stats Download Patch File Download Diff File

34
core-testing/src/main/java/com/google/samples/apps/nowinandroid/core/testing/util/TestDispatcherRule.kt → core-testing/src/main/java/com/google/samples/apps/nowinandroid/core/testing/util/MainDispatcherRule.kt
Unescape View File

@ -16,31 +16,27 @@
package com.google.samples.apps.nowinandroid.core.testing.util package com.google.samples.apps.nowinandroid.core.testing.util
import kotlinx.coroutines.CoroutineDispatcher
import kotlinx.coroutines.Dispatchers import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.test.StandardTestDispatcher import kotlinx.coroutines.test.TestDispatcher
import kotlinx.coroutines.test.UnconfinedTestDispatcher
import kotlinx.coroutines.test.resetMain import kotlinx.coroutines.test.resetMain
import kotlinx.coroutines.test.setMain import kotlinx.coroutines.test.setMain
import org.junit.rules.TestRule import org.junit.rules.TestRule
import org.junit.rules.TestWatcher
import org.junit.runner.Description import org.junit.runner.Description
import org.junit.runners.model.Statement
/** /**
* A [TestRule] that initializes the main dispatcher to [dispatcher], which defaults to a * A JUnit [TestRule] that sets the Main dispatcher to [testDispatcher]
* [StandardTestDispatcher]. * for the duration of the test.
*/ */
class TestDispatcherRule( class MainDispatcherRule(
private val dispatcher: CoroutineDispatcher = StandardTestDispatcher() val testDispatcher: TestDispatcher = UnconfinedTestDispatcher(),
) : TestRule { ) : TestWatcher() {
override fun apply(base: Statement, description: Description): Statement = override fun starting(description: Description) {
object : Statement() { Dispatchers.setMain(testDispatcher)
override fun evaluate() { }
Dispatchers.setMain(dispatcher)
try { override fun finished(description: Description) {
base.evaluate() Dispatchers.resetMain()
} finally { }
Dispatchers.resetMain()
}
}
}
} }

123
feature-author/src/test/java/com/google/samples/apps/nowinandroid/feature/author/AuthorViewModelTest.kt
Unescape View File

@ -17,7 +17,6 @@
package com.google.samples.apps.nowinandroid.feature.author package com.google.samples.apps.nowinandroid.feature.author
import androidx.lifecycle.SavedStateHandle import androidx.lifecycle.SavedStateHandle
import app.cash.turbine.test
import com.google.samples.apps.nowinandroid.core.model.data.Author import com.google.samples.apps.nowinandroid.core.model.data.Author
import com.google.samples.apps.nowinandroid.core.model.data.FollowableAuthor import com.google.samples.apps.nowinandroid.core.model.data.FollowableAuthor
import com.google.samples.apps.nowinandroid.core.model.data.NewsResource import com.google.samples.apps.nowinandroid.core.model.data.NewsResource
@ -25,9 +24,12 @@ import com.google.samples.apps.nowinandroid.core.model.data.NewsResourceType.Vid
import com.google.samples.apps.nowinandroid.core.testing.repository.TestAuthorsRepository import com.google.samples.apps.nowinandroid.core.testing.repository.TestAuthorsRepository
import com.google.samples.apps.nowinandroid.core.testing.repository.TestNewsRepository import com.google.samples.apps.nowinandroid.core.testing.repository.TestNewsRepository
import com.google.samples.apps.nowinandroid.core.testing.repository.TestUserDataRepository import com.google.samples.apps.nowinandroid.core.testing.repository.TestUserDataRepository
import com.google.samples.apps.nowinandroid.core.testing.util.TestDispatcherRule import com.google.samples.apps.nowinandroid.core.testing.util.MainDispatcherRule
import com.google.samples.apps.nowinandroid.feature.author.navigation.AuthorDestination import com.google.samples.apps.nowinandroid.feature.author.navigation.AuthorDestination
import kotlinx.coroutines.flow.collect
import kotlinx.coroutines.flow.first import kotlinx.coroutines.flow.first
import kotlinx.coroutines.launch
import kotlinx.coroutines.test.UnconfinedTestDispatcher
import kotlinx.coroutines.test.runTest import kotlinx.coroutines.test.runTest
import kotlinx.datetime.Instant import kotlinx.datetime.Instant
import org.junit.Assert.assertEquals import org.junit.Assert.assertEquals
@ -36,10 +38,14 @@ import org.junit.Before
import org.junit.Rule import org.junit.Rule
import org.junit.Test import org.junit.Test
/**
* To learn more about how this test handles Flows created with stateIn, see
* https://developer.android.com/kotlin/flow/test#statein
*/
class AuthorViewModelTest { class AuthorViewModelTest {
@get:Rule @get:Rule
val dispatcherRule = TestDispatcherRule() val dispatcherRule = MainDispatcherRule()
private val userDataRepository = TestUserDataRepository() private val userDataRepository = TestUserDataRepository()
private val authorsRepository = TestAuthorsRepository() private val authorsRepository = TestAuthorsRepository()
@ -62,90 +68,91 @@ class AuthorViewModelTest {
@Test @Test
fun uiStateAuthor_whenSuccess_matchesAuthorFromRepository() = runTest { fun uiStateAuthor_whenSuccess_matchesAuthorFromRepository() = runTest {
viewModel.uiState.test { val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
awaitItem()
// To make sure AuthorUiState is success
authorsRepository.sendAuthors(testInputAuthors.map(FollowableAuthor::author))
userDataRepository.setFollowedAuthorIds(setOf(testInputAuthors[1].author.id))
val item = awaitItem() // To make sure AuthorUiState is success
assertTrue(item.authorState is AuthorUiState.Success) authorsRepository.sendAuthors(testInputAuthors.map(FollowableAuthor::author))
userDataRepository.setFollowedAuthorIds(setOf(testInputAuthors[1].author.id))
val successAuthorUiState = item.authorState as AuthorUiState.Success val item = viewModel.uiState.value
val authorFromRepository = authorsRepository.getAuthorStream( assertTrue(item.authorState is AuthorUiState.Success)
id = testInputAuthors[0].author.id
).first()
successAuthorUiState.followableAuthor.author val successAuthorUiState = item.authorState as AuthorUiState.Success
assertEquals(authorFromRepository, successAuthorUiState.followableAuthor.author) val authorFromRepository = authorsRepository.getAuthorStream(
} id = testInputAuthors[0].author.id
).first()
successAuthorUiState.followableAuthor.author
assertEquals(authorFromRepository, successAuthorUiState.followableAuthor.author)
collectJob.cancel()
} }
@Test @Test
fun uiStateNews_whenInitialized_thenShowLoading() = runTest { fun uiStateNews_whenInitialized_thenShowLoading() = runTest {
viewModel.uiState.test { assertEquals(NewsUiState.Loading, viewModel.uiState.value.newsState)
assertEquals(NewsUiState.Loading, awaitItem().newsState)
}
} }
@Test @Test
fun uiStateAuthor_whenInitialized_thenShowLoading() = runTest { fun uiStateAuthor_whenInitialized_thenShowLoading() = runTest {
viewModel.uiState.test { assertEquals(AuthorUiState.Loading, viewModel.uiState.value.authorState)
assertEquals(AuthorUiState.Loading, awaitItem().authorState)
}
} }
@Test @Test
fun uiStateAuthor_whenFollowedIdsSuccessAndAuthorLoading_thenShowLoading() = runTest { fun uiStateAuthor_whenFollowedIdsSuccessAndAuthorLoading_thenShowLoading() = runTest {
viewModel.uiState.test { val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
userDataRepository.setFollowedAuthorIds(setOf(testInputAuthors[1].author.id))
assertEquals(AuthorUiState.Loading, awaitItem().authorState) userDataRepository.setFollowedAuthorIds(setOf(testInputAuthors[1].author.id))
} assertEquals(AuthorUiState.Loading, viewModel.uiState.value.authorState)
collectJob.cancel()
} }
@Test @Test
fun uiStateAuthor_whenFollowedIdsSuccessAndAuthorSuccess_thenAuthorSuccessAndNewsLoading() = fun uiStateAuthor_whenFollowedIdsSuccessAndAuthorSuccess_thenAuthorSuccessAndNewsLoading() =
runTest { runTest {
viewModel.uiState.test { val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
awaitItem()
authorsRepository.sendAuthors(testInputAuthors.map { it.author }) authorsRepository.sendAuthors(testInputAuthors.map { it.author })
userDataRepository.setFollowedAuthorIds(setOf(testInputAuthors[1].author.id)) userDataRepository.setFollowedAuthorIds(setOf(testInputAuthors[1].author.id))
val item = awaitItem() val item = viewModel.uiState.value
assertTrue(item.authorState is AuthorUiState.Success) assertTrue(item.authorState is AuthorUiState.Success)
assertTrue(item.newsState is NewsUiState.Loading) assertTrue(item.newsState is NewsUiState.Loading)
}
collectJob.cancel()
} }
@Test @Test
fun uiStateAuthor_whenFollowedIdsSuccessAndAuthorSuccessAndNewsIsSuccess_thenAllSuccess() = fun uiStateAuthor_whenFollowedIdsSuccessAndAuthorSuccessAndNewsIsSuccess_thenAllSuccess() =
runTest { runTest {
viewModel.uiState.test { val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
awaitItem()
authorsRepository.sendAuthors(testInputAuthors.map { it.author }) authorsRepository.sendAuthors(testInputAuthors.map { it.author })
userDataRepository.setFollowedAuthorIds(setOf(testInputAuthors[1].author.id)) userDataRepository.setFollowedAuthorIds(setOf(testInputAuthors[1].author.id))
newsRepository.sendNewsResources(sampleNewsResources) newsRepository.sendNewsResources(sampleNewsResources)
val item = awaitItem() val item = viewModel.uiState.value
assertTrue(item.authorState is AuthorUiState.Success) assertTrue(item.authorState is AuthorUiState.Success)
assertTrue(item.newsState is NewsUiState.Success) assertTrue(item.newsState is NewsUiState.Success)
}
collectJob.cancel()
} }
@Test @Test
fun uiStateAuthor_whenFollowingAuthor_thenShowUpdatedAuthor() = runTest { fun uiStateAuthor_whenFollowingAuthor_thenShowUpdatedAuthor() = runTest {
viewModel.uiState val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
.test {
awaitItem() authorsRepository.sendAuthors(testInputAuthors.map { it.author })
authorsRepository.sendAuthors(testInputAuthors.map { it.author }) // Set which author IDs are followed, not including 0.
// Set which author IDs are followed, not including 0. userDataRepository.setFollowedAuthorIds(setOf(testInputAuthors[1].author.id))
userDataRepository.setFollowedAuthorIds(setOf(testInputAuthors[1].author.id))
viewModel.followAuthorToggle(true)
viewModel.followAuthorToggle(true)
assertEquals(
assertEquals( AuthorUiState.Success(followableAuthor = testOutputAuthors[0]),
AuthorUiState.Success(followableAuthor = testOutputAuthors[0]), viewModel.uiState.value.authorState
awaitItem().authorState )
)
} collectJob.cancel()
} }
} }

2528
feature-foryou/src/test/java/com/google/samples/apps/nowinandroid/feature/foryou/ForYouViewModelTest.kt
View File

File diff suppressed because it is too large Load Diff

219
feature-interests/src/test/java/com/google/samples/apps/nowinandroid/interests/InterestsViewModelTest.kt
Unescape View File

@ -16,7 +16,6 @@
package com.google.samples.apps.nowinandroid.interests package com.google.samples.apps.nowinandroid.interests
import app.cash.turbine.test
import com.google.samples.apps.nowinandroid.core.model.data.Author import com.google.samples.apps.nowinandroid.core.model.data.Author
import com.google.samples.apps.nowinandroid.core.model.data.FollowableAuthor import com.google.samples.apps.nowinandroid.core.model.data.FollowableAuthor
import com.google.samples.apps.nowinandroid.core.model.data.FollowableTopic import com.google.samples.apps.nowinandroid.core.model.data.FollowableTopic
@ -24,19 +23,26 @@ import com.google.samples.apps.nowinandroid.core.model.data.Topic
import com.google.samples.apps.nowinandroid.core.testing.repository.TestAuthorsRepository import com.google.samples.apps.nowinandroid.core.testing.repository.TestAuthorsRepository
import com.google.samples.apps.nowinandroid.core.testing.repository.TestTopicsRepository import com.google.samples.apps.nowinandroid.core.testing.repository.TestTopicsRepository
import com.google.samples.apps.nowinandroid.core.testing.repository.TestUserDataRepository import com.google.samples.apps.nowinandroid.core.testing.repository.TestUserDataRepository
import com.google.samples.apps.nowinandroid.core.testing.util.TestDispatcherRule import com.google.samples.apps.nowinandroid.core.testing.util.MainDispatcherRule
import com.google.samples.apps.nowinandroid.feature.interests.InterestsUiState import com.google.samples.apps.nowinandroid.feature.interests.InterestsUiState
import com.google.samples.apps.nowinandroid.feature.interests.InterestsViewModel import com.google.samples.apps.nowinandroid.feature.interests.InterestsViewModel
import kotlinx.coroutines.flow.collect
import kotlinx.coroutines.launch
import kotlinx.coroutines.test.UnconfinedTestDispatcher
import kotlinx.coroutines.test.runTest import kotlinx.coroutines.test.runTest
import org.junit.Assert.assertEquals import org.junit.Assert.assertEquals
import org.junit.Before import org.junit.Before
import org.junit.Rule import org.junit.Rule
import org.junit.Test import org.junit.Test
/**
* To learn more about how this test handles Flows created with stateIn, see
* https://developer.android.com/kotlin/flow/test#statein
*/
class InterestsViewModelTest { class InterestsViewModelTest {
@get:Rule @get:Rule
val dispatcherRule = TestDispatcherRule() val mainDispatcherRule = MainDispatcherRule()
private val userDataRepository = TestUserDataRepository() private val userDataRepository = TestUserDataRepository()
private val authorsRepository = TestAuthorsRepository() private val authorsRepository = TestAuthorsRepository()
@ -54,135 +60,136 @@ class InterestsViewModelTest {
@Test @Test
fun uiState_whenInitialized_thenShowLoading() = runTest { fun uiState_whenInitialized_thenShowLoading() = runTest {
viewModel.uiState.test { assertEquals(InterestsUiState.Loading, viewModel.uiState.value)
assertEquals(InterestsUiState.Loading, awaitItem())
}
} }
@Test @Test
fun uiState_whenFollowedTopicsAreLoading_thenShowLoading() = runTest { fun uiState_whenFollowedTopicsAreLoading_thenShowLoading() = runTest {
viewModel.uiState.test { val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
assertEquals(InterestsUiState.Loading, awaitItem())
userDataRepository.setFollowedAuthorIds(setOf("1")) userDataRepository.setFollowedAuthorIds(setOf("1"))
userDataRepository.setFollowedTopicIds(emptySet()) userDataRepository.setFollowedTopicIds(emptySet())
} assertEquals(InterestsUiState.Loading, viewModel.uiState.value)
collectJob.cancel()
} }
@Test @Test
fun uiState_whenFollowedAuthorsAreLoading_thenShowLoading() = runTest { fun uiState_whenFollowedAuthorsAreLoading_thenShowLoading() = runTest {
viewModel.uiState.test { val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
assertEquals(InterestsUiState.Loading, awaitItem())
userDataRepository.setFollowedAuthorIds(emptySet()) userDataRepository.setFollowedAuthorIds(emptySet())
userDataRepository.setFollowedTopicIds(setOf("1")) userDataRepository.setFollowedTopicIds(setOf("1"))
} assertEquals(InterestsUiState.Loading, viewModel.uiState.value)
collectJob.cancel()
} }
@Test @Test
fun uiState_whenFollowingNewTopic_thenShowUpdatedTopics() = runTest { fun uiState_whenFollowingNewTopic_thenShowUpdatedTopics() = runTest {
val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
val toggleTopicId = testOutputTopics[1].topic.id val toggleTopicId = testOutputTopics[1].topic.id
viewModel.uiState authorsRepository.sendAuthors(emptyList())
.test { userDataRepository.setFollowedAuthorIds(emptySet())
awaitItem() topicsRepository.sendTopics(testInputTopics.map { it.topic })
authorsRepository.sendAuthors(emptyList()) userDataRepository.setFollowedTopicIds(setOf(testInputTopics[0].topic.id))
userDataRepository.setFollowedAuthorIds(emptySet())
topicsRepository.sendTopics(testInputTopics.map { it.topic }) assertEquals(
userDataRepository.setFollowedTopicIds(setOf(testInputTopics[0].topic.id)) false,
(viewModel.uiState.value as InterestsUiState.Interests)
assertEquals( .topics.first { it.topic.id == toggleTopicId }.isFollowed
false, )
(awaitItem() as InterestsUiState.Interests)
.topics.first { it.topic.id == toggleTopicId }.isFollowed viewModel.followTopic(
) followedTopicId = toggleTopicId,
true
viewModel.followTopic( )
followedTopicId = toggleTopicId,
true assertEquals(
) InterestsUiState.Interests(topics = testOutputTopics, authors = emptyList()),
viewModel.uiState.value
assertEquals( )
InterestsUiState.Interests(topics = testOutputTopics, authors = emptyList()),
awaitItem() collectJob.cancel()
)
}
} }
@Test @Test
fun uiState_whenFollowingNewAuthor_thenShowUpdatedAuthors() = runTest { fun uiState_whenFollowingNewAuthor_thenShowUpdatedAuthors() = runTest {
viewModel.uiState val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
.test {
awaitItem() authorsRepository.sendAuthors(testInputAuthors.map { it.author })
authorsRepository.sendAuthors(testInputAuthors.map { it.author }) userDataRepository.setFollowedAuthorIds(setOf(testInputAuthors[0].author.id))
userDataRepository.setFollowedAuthorIds(setOf(testInputAuthors[0].author.id)) topicsRepository.sendTopics(listOf())
topicsRepository.sendTopics(listOf()) userDataRepository.setFollowedTopicIds(setOf())
userDataRepository.setFollowedTopicIds(setOf())
viewModel.followAuthor(
awaitItem() followedAuthorId = testInputAuthors[1].author.id,
viewModel.followAuthor( followed = true
followedAuthorId = testInputAuthors[1].author.id, )
followed = true
) assertEquals(
InterestsUiState.Interests(topics = emptyList(), authors = testOutputAuthors),
assertEquals( viewModel.uiState.value
InterestsUiState.Interests(topics = emptyList(), authors = testOutputAuthors), )
awaitItem()
) collectJob.cancel()
}
} }
@Test @Test
fun uiState_whenUnfollowingTopics_thenShowUpdatedTopics() = runTest { fun uiState_whenUnfollowingTopics_thenShowUpdatedTopics() = runTest {
val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
val toggleTopicId = testOutputTopics[1].topic.id val toggleTopicId = testOutputTopics[1].topic.id
viewModel.uiState
.test { authorsRepository.sendAuthors(emptyList())
awaitItem() userDataRepository.setFollowedAuthorIds(emptySet())
authorsRepository.sendAuthors(emptyList()) topicsRepository.sendTopics(testOutputTopics.map { it.topic })
userDataRepository.setFollowedAuthorIds(emptySet()) userDataRepository.setFollowedTopicIds(
topicsRepository.sendTopics(testOutputTopics.map { it.topic }) setOf(testOutputTopics[0].topic.id, testOutputTopics[1].topic.id)
userDataRepository.setFollowedTopicIds( )
setOf(testOutputTopics[0].topic.id, testOutputTopics[1].topic.id)
) assertEquals(
true,
assertEquals( (viewModel.uiState.value as InterestsUiState.Interests)
true, .topics.first { it.topic.id == toggleTopicId }.isFollowed
(awaitItem() as InterestsUiState.Interests) )
.topics.first { it.topic.id == toggleTopicId }.isFollowed
) viewModel.followTopic(
followedTopicId = toggleTopicId,
viewModel.followTopic( false
followedTopicId = toggleTopicId, )
false
) assertEquals(
InterestsUiState.Interests(topics = testInputTopics, authors = emptyList()),
assertEquals( viewModel.uiState.value
InterestsUiState.Interests(topics = testInputTopics, authors = emptyList()), )
awaitItem()
) collectJob.cancel()
}
} }
@Test @Test
fun uiState_whenUnfollowingAuthors_thenShowUpdatedAuthors() = runTest { fun uiState_whenUnfollowingAuthors_thenShowUpdatedAuthors() = runTest {
viewModel.uiState val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
.test {
awaitItem() authorsRepository.sendAuthors(testOutputAuthors.map { it.author })
authorsRepository.sendAuthors(testOutputAuthors.map { it.author }) userDataRepository.setFollowedAuthorIds(
userDataRepository.setFollowedAuthorIds( setOf(testOutputAuthors[0].author.id, testOutputAuthors[1].author.id)
setOf(testOutputAuthors[0].author.id, testOutputAuthors[1].author.id) )
) topicsRepository.sendTopics(listOf())
topicsRepository.sendTopics(listOf()) userDataRepository.setFollowedTopicIds(setOf())
userDataRepository.setFollowedTopicIds(setOf())
viewModel.followAuthor(
awaitItem() followedAuthorId = testOutputAuthors[1].author.id,
viewModel.followAuthor( followed = false
followedAuthorId = testOutputAuthors[1].author.id, )
followed = false
) assertEquals(
InterestsUiState.Interests(topics = emptyList(), authors = testInputAuthors),
assertEquals( viewModel.uiState.value
InterestsUiState.Interests(topics = emptyList(), authors = testInputAuthors), )
awaitItem()
) collectJob.cancel()
}
} }
} }

119
feature-topic/src/test/java/com/google/samples/apps/nowinandroid/feature/topic/TopicViewModelTest.kt
Unescape View File

@ -17,7 +17,6 @@
package com.google.samples.apps.nowinandroid.feature.topic package com.google.samples.apps.nowinandroid.feature.topic
import androidx.lifecycle.SavedStateHandle import androidx.lifecycle.SavedStateHandle
import app.cash.turbine.test
import com.google.samples.apps.nowinandroid.core.model.data.FollowableTopic import com.google.samples.apps.nowinandroid.core.model.data.FollowableTopic
import com.google.samples.apps.nowinandroid.core.model.data.NewsResource import com.google.samples.apps.nowinandroid.core.model.data.NewsResource
import com.google.samples.apps.nowinandroid.core.model.data.NewsResourceType.Video import com.google.samples.apps.nowinandroid.core.model.data.NewsResourceType.Video
@ -25,9 +24,12 @@ import com.google.samples.apps.nowinandroid.core.model.data.Topic
import com.google.samples.apps.nowinandroid.core.testing.repository.TestNewsRepository import com.google.samples.apps.nowinandroid.core.testing.repository.TestNewsRepository
import com.google.samples.apps.nowinandroid.core.testing.repository.TestTopicsRepository import com.google.samples.apps.nowinandroid.core.testing.repository.TestTopicsRepository
import com.google.samples.apps.nowinandroid.core.testing.repository.TestUserDataRepository import com.google.samples.apps.nowinandroid.core.testing.repository.TestUserDataRepository
import com.google.samples.apps.nowinandroid.core.testing.util.TestDispatcherRule import com.google.samples.apps.nowinandroid.core.testing.util.MainDispatcherRule
import com.google.samples.apps.nowinandroid.feature.topic.navigation.TopicDestination import com.google.samples.apps.nowinandroid.feature.topic.navigation.TopicDestination
import kotlinx.coroutines.flow.collect
import kotlinx.coroutines.flow.first import kotlinx.coroutines.flow.first
import kotlinx.coroutines.launch
import kotlinx.coroutines.test.UnconfinedTestDispatcher
import kotlinx.coroutines.test.runTest import kotlinx.coroutines.test.runTest
import kotlinx.datetime.Instant import kotlinx.datetime.Instant
import org.junit.Assert.assertEquals import org.junit.Assert.assertEquals
@ -36,10 +38,14 @@ import org.junit.Before
import org.junit.Rule import org.junit.Rule
import org.junit.Test import org.junit.Test
/**
* To learn more about how this test handles Flows created with stateIn, see
* https://developer.android.com/kotlin/flow/test#statein
*/
class TopicViewModelTest { class TopicViewModelTest {
@get:Rule @get:Rule
val dispatcherRule = TestDispatcherRule() val dispatcherRule = MainDispatcherRule()
private val userDataRepository = TestUserDataRepository() private val userDataRepository = TestUserDataRepository()
private val topicsRepository = TestTopicsRepository() private val topicsRepository = TestTopicsRepository()
@ -59,87 +65,88 @@ class TopicViewModelTest {
@Test @Test
fun uiStateAuthor_whenSuccess_matchesTopicFromRepository() = runTest { fun uiStateAuthor_whenSuccess_matchesTopicFromRepository() = runTest {
viewModel.uiState.test { val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
awaitItem()
topicsRepository.sendTopics(testInputTopics.map(FollowableTopic::topic))
userDataRepository.setFollowedTopicIds(setOf(testInputTopics[1].topic.id))
val item = awaitItem()
assertTrue(item.topicState is TopicUiState.Success)
val successTopicState = item.topicState as TopicUiState.Success topicsRepository.sendTopics(testInputTopics.map(FollowableTopic::topic))
val topicFromRepository = topicsRepository.getTopic( userDataRepository.setFollowedTopicIds(setOf(testInputTopics[1].topic.id))
testInputTopics[0].topic.id val item = viewModel.uiState.value
).first() assertTrue(item.topicState is TopicUiState.Success)
assertEquals(topicFromRepository, successTopicState.followableTopic.topic) val successTopicState = item.topicState as TopicUiState.Success
} val topicFromRepository = topicsRepository.getTopic(
testInputTopics[0].topic.id
).first()
assertEquals(topicFromRepository, successTopicState.followableTopic.topic)
collectJob.cancel()
} }
@Test @Test
fun uiStateNews_whenInitialized_thenShowLoading() = runTest { fun uiStateNews_whenInitialized_thenShowLoading() = runTest {
viewModel.uiState.test { assertEquals(NewsUiState.Loading, viewModel.uiState.value.newsState)
assertEquals(NewsUiState.Loading, awaitItem().newsState)
}
} }
@Test @Test
fun uiStateTopic_whenInitialized_thenShowLoading() = runTest { fun uiStateTopic_whenInitialized_thenShowLoading() = runTest {
viewModel.uiState.test { assertEquals(TopicUiState.Loading, viewModel.uiState.value.topicState)
assertEquals(TopicUiState.Loading, awaitItem().topicState)
}
} }
@Test @Test
fun uiStateTopic_whenFollowedIdsSuccessAndTopicLoading_thenShowLoading() = runTest { fun uiStateTopic_whenFollowedIdsSuccessAndTopicLoading_thenShowLoading() = runTest {
viewModel.uiState.test { val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
userDataRepository.setFollowedTopicIds(setOf(testInputTopics[1].topic.id))
assertEquals(TopicUiState.Loading, awaitItem().topicState) userDataRepository.setFollowedTopicIds(setOf(testInputTopics[1].topic.id))
} assertEquals(TopicUiState.Loading, viewModel.uiState.value.topicState)
collectJob.cancel()
} }
@Test @Test
fun uiStateTopic_whenFollowedIdsSuccessAndTopicSuccess_thenTopicSuccessAndNewsLoading() = fun uiStateTopic_whenFollowedIdsSuccessAndTopicSuccess_thenTopicSuccessAndNewsLoading() =
runTest { runTest {
viewModel.uiState.test { val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
awaitItem()
topicsRepository.sendTopics(testInputTopics.map { it.topic }) topicsRepository.sendTopics(testInputTopics.map { it.topic })
userDataRepository.setFollowedTopicIds(setOf(testInputTopics[1].topic.id)) userDataRepository.setFollowedTopicIds(setOf(testInputTopics[1].topic.id))
val item = awaitItem() val item = viewModel.uiState.value
assertTrue(item.topicState is TopicUiState.Success) assertTrue(item.topicState is TopicUiState.Success)
assertTrue(item.newsState is NewsUiState.Loading) assertTrue(item.newsState is NewsUiState.Loading)
}
collectJob.cancel()
} }
@Test @Test
fun uiStateTopic_whenFollowedIdsSuccessAndTopicSuccessAndNewsIsSuccess_thenAllSuccess() = fun uiStateTopic_whenFollowedIdsSuccessAndTopicSuccessAndNewsIsSuccess_thenAllSuccess() =
runTest { runTest {
viewModel.uiState.test { val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
awaitItem()
topicsRepository.sendTopics(testInputTopics.map { it.topic }) topicsRepository.sendTopics(testInputTopics.map { it.topic })
userDataRepository.setFollowedTopicIds(setOf(testInputTopics[1].topic.id)) userDataRepository.setFollowedTopicIds(setOf(testInputTopics[1].topic.id))
newsRepository.sendNewsResources(sampleNewsResources) newsRepository.sendNewsResources(sampleNewsResources)
val item = awaitItem() val item = viewModel.uiState.value
assertTrue(item.topicState is TopicUiState.Success) assertTrue(item.topicState is TopicUiState.Success)
assertTrue(item.newsState is NewsUiState.Success) assertTrue(item.newsState is NewsUiState.Success)
}
collectJob.cancel()
} }
@Test @Test
fun uiStateTopic_whenFollowingTopic_thenShowUpdatedTopic() = runTest { fun uiStateTopic_whenFollowingTopic_thenShowUpdatedTopic() = runTest {
viewModel.uiState val collectJob = launch(UnconfinedTestDispatcher()) { viewModel.uiState.collect() }
.test {
awaitItem() topicsRepository.sendTopics(testInputTopics.map { it.topic })
topicsRepository.sendTopics(testInputTopics.map { it.topic }) // Set which topic IDs are followed, not including 0.
// Set which topic IDs are followed, not including 0. userDataRepository.setFollowedTopicIds(setOf(testInputTopics[1].topic.id))
userDataRepository.setFollowedTopicIds(setOf(testInputTopics[1].topic.id))
viewModel.followTopicToggle(true)
viewModel.followTopicToggle(true)
assertEquals(
assertEquals( TopicUiState.Success(followableTopic = testOutputTopics[0]),
TopicUiState.Success(followableTopic = testOutputTopics[0]), viewModel.uiState.value.topicState
awaitItem().topicState )
)
} collectJob.cancel()
} }
} }

Write Preview
Loading…
Cancel
Save