Fix location cloud data source test

Signed-off-by: Yurii Surzhykov <yuriisurzhykov@gmail.com>
yuriisurzhykov · May 28, 2024 · 1fd0ef3 · 1fd0ef3
1 parent 08f2b19
commit 1fd0ef3
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 9 deletions.
diff --git a/.../src/test/kotlin/com/github/yuriisurzhykov/purs/data/cloud/LocationCloudDataSourceTest.kt b/.../src/test/kotlin/com/github/yuriisurzhykov/purs/data/cloud/LocationCloudDataSourceTest.kt
@@ -4,7 +4,7 @@ import com.github.yuriisurzhykov.purs.core.RequestResult
 import com.github.yuriisurzhykov.purs.data.cloud.model.LocationCloud
 import io.mockk.coEvery
 import io.mockk.mockk
-import kotlinx.coroutines.flow.first
+import kotlinx.coroutines.flow.toList
 import kotlinx.coroutines.test.runTest
 import okhttp3.MediaType.Companion.toMediaType
 import okhttp3.ResponseBody.Companion.toResponseBody
@@ -22,8 +22,8 @@ class LocationCloudDataSourceTest {
         coEvery { locationService.fetchLocation() } returns Response.success(mockLocationCloud)
         val dataSource = LocationCloudDataSource.Base(locationService)
 
-        val actual = dataSource.fetchLocation().first()
-        val expected = RequestResult.Success(mockLocationCloud)
+        val actual = dataSource.fetchLocation().toList()
+        val expected = listOf(RequestResult.InProgress(), RequestResult.Success(mockLocationCloud))
 
         assertEquals(expected, actual)
     }
@@ -34,8 +34,8 @@ class LocationCloudDataSourceTest {
         coEvery { locationService.fetchLocation() } returns Response.success(null)
         val dataSource = LocationCloudDataSource.Base(locationService)
 
-        val actual = dataSource.fetchLocation().first()
-        val expected = RequestResult.Error(null, null)
+        val actual = dataSource.fetchLocation().toList()
+        val expected = listOf(RequestResult.InProgress(), RequestResult.Error(null, null))
 
         assertEquals(expected, actual)
     }
@@ -49,8 +49,11 @@ class LocationCloudDataSourceTest {
         )
         val dataSource = LocationCloudDataSource.Base(locationService)
 
-        val actual = dataSource.fetchLocation().first()
-        val expected = RequestResult.Error(null, ServerError(404, "Page not found"))
+        val actual = dataSource.fetchLocation().toList()
+        val expected = listOf(
+            RequestResult.InProgress(),
+            RequestResult.Error(null, ServerError(404, "Page not found"))
+        )
 
         assertEquals(expected, actual)
     }
@@ -62,8 +65,9 @@ class LocationCloudDataSourceTest {
         coEvery { locationService.fetchLocation() } throws exceptionToThrow
         val dataSource = LocationCloudDataSource.Base(locationService)
 
-        val actual = dataSource.fetchLocation().first()
-        val expected = RequestResult.Error(null, exceptionToThrow)
+        val actual = dataSource.fetchLocation().toList()
+        val expected =
+            listOf(RequestResult.InProgress(), RequestResult.Error(null, exceptionToThrow))
 
         assertEquals(expected, actual)
     }

diff --git a/...b.yuriisurzhykov.purs.PursDatabase/1.json → ...iisurzhykov.purs.data.PursDatabase/1.json b/...b.yuriisurzhykov.purs.PursDatabase/1.json → ...iisurzhykov.purs.data.PursDatabase/1.json