演習9.2

app/sns/src/test/java/jp/co/mixi/androidtraining/sns/data/repository/FakeTimelineRepository.kt

@@ -0,0 +1,11 @@
+package jp.co.mixi.androidtraining.sns.data.repository
+
+import jp.co.mixi.androidtraining.sns.data.entity.Post
+import kotlinx.coroutines.delay
+
+class FakeTimelineRepository(private val timeline: List<Post>) : TimelineRepository {
+    override suspend fun getTimeline(): List<Post> {
+        delay(2000)
+        return timeline
+    }
+}

app/sns/src/test/java/jp/co/mixi/androidtraining/sns/ui/TimelineViewModelTest.kt

@@ -0,0 +1,33 @@
+package jp.co.mixi.androidtraining.sns.ui
+
+import jp.co.mixi.androidtraining.sns.data.entity.Post
+import jp.co.mixi.androidtraining.sns.data.entity.User
+import jp.co.mixi.androidtraining.sns.data.repository.FakeTimelineRepository
+import jp.co.mixi.androidtraining.sns.utils.MainDispatcherRule
+import kotlinx.coroutines.test.advanceUntilIdle
+import kotlinx.coroutines.test.runTest
+import org.junit.Assert.assertEquals
+import org.junit.Assert.assertFalse
+import org.junit.Rule
+import org.junit.Test
+
+class TimelineViewModelTest {
+
+    @get:Rule
+    val dispatcherRule = MainDispatcherRule()
+
+    private val fakeTimeline = listOf(
+        Post(author = User(name = "User Name", photoUrl = ""), content = "Content"),
+    )
+
+    private val viewModel = TimelineViewModel(FakeTimelineRepository(fakeTimeline))
+
+    @Test
+    fun testGetTimeline() = runTest {
+        assertEquals(emptyList<Post>(), viewModel.uiState.posts)
+        viewModel.getTimeline()
+        advanceUntilIdle()
+        assertEquals(fakeTimeline, viewModel.uiState.posts)
+        assertFalse(viewModel.uiState.isLoading)
+    }
+}

app/sns/src/test/java/jp/co/mixi/androidtraining/sns/utils/MainDispatcherRule.kt

@@ -0,0 +1,18 @@
+package jp.co.mixi.androidtraining.sns.utils
+
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.test.StandardTestDispatcher
+import kotlinx.coroutines.test.resetMain
+import kotlinx.coroutines.test.setMain
+import org.junit.rules.TestWatcher
+import org.junit.runner.Description
+
+class MainDispatcherRule : TestWatcher() {
+    override fun starting(description: Description) {
+        Dispatchers.setMain(StandardTestDispatcher())
+    }
+
+    override fun finished(description: Description) {
+        Dispatchers.resetMain()
+    }
+}