chore: Cell unit tests [WPB-15386] (#3979)

Author: sbakhtiarov

features/cells/build.gradle.kts

@@ -45,6 +45,7 @@ dependencies {
     testImplementation(libs.junit5.core)
     testImplementation(libs.coroutines.test)
     testImplementation(libs.mockk.core)
+    testImplementation(libs.turbine)
     testRuntimeOnly(libs.junit5.engine)
     androidTestImplementation(libs.androidx.test.extJunit)
     androidTestImplementation(libs.androidx.espresso.core)

features/cells/src/main/java/com/wire/android/feature/cells/ui/CellViewModel.kt

@@ -36,12 +36,14 @@ import com.wire.kalium.common.functional.onFailure
 import com.wire.kalium.common.functional.onSuccess
 import com.wire.kalium.logic.data.asset.KaliumFileSystem
 import dagger.hilt.android.lifecycle.HiltViewModel
+import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.channels.BufferOverflow
 import kotlinx.coroutines.channels.Channel
 import kotlinx.coroutines.flow.MutableSharedFlow
 import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.asSharedFlow
 import kotlinx.coroutines.flow.asStateFlow
+import kotlinx.coroutines.flow.flowOn
 import kotlinx.coroutines.flow.receiveAsFlow
 import kotlinx.coroutines.flow.update
 import kotlinx.coroutines.launch
@@ -77,7 +79,9 @@ class CellViewModel @Inject constructor(
         capacity = Channel.BUFFERED,
         onBufferOverflow = BufferOverflow.DROP_OLDEST
     )
-    internal val actions = _actions.receiveAsFlow()
+    internal val actions = _actions
+        .receiveAsFlow()
+        .flowOn(Dispatchers.Main.immediate)
 
     // Download progress value for each file being downloaded.
     private val uploadProgress = mutableStateMapOf<String, Float>()

features/cells/src/main/java/com/wire/android/feature/cells/ui/publiclink/PublicLinkViewModel.kt

@@ -30,10 +30,12 @@ import com.wire.kalium.cells.domain.usecase.publiclink.GetPublicLinkUseCase
 import com.wire.kalium.common.functional.onFailure
 import com.wire.kalium.common.functional.onSuccess
 import dagger.hilt.android.lifecycle.HiltViewModel
+import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.channels.BufferOverflow
 import kotlinx.coroutines.channels.Channel
 import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.asStateFlow
+import kotlinx.coroutines.flow.flowOn
 import kotlinx.coroutines.flow.receiveAsFlow
 import kotlinx.coroutines.flow.update
 import kotlinx.coroutines.launch
@@ -58,7 +60,9 @@ class PublicLinkViewModel @Inject constructor(
         onBufferOverflow = BufferOverflow.DROP_OLDEST
     )
 
-    internal val actions = _actions.receiveAsFlow()
+    internal val actions = _actions
+        .receiveAsFlow()
+        .flowOn(Dispatchers.Main.immediate)
 
     private var publicLink: PublicLink? = null
 

features/cells/src/test/kotlin/com/wire/android/config/NavigationTestExtension.kt

@@ -0,0 +1,49 @@
+/*
+ * Wire
+ * Copyright (C) 2025 Wire Swiss GmbH
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see http://www.gnu.org/licenses/.
+ */
+package com.wire.android.config
+
+import io.mockk.mockkStatic
+import io.mockk.unmockkStatic
+import org.junit.jupiter.api.extension.AfterEachCallback
+import org.junit.jupiter.api.extension.BeforeEachCallback
+import org.junit.jupiter.api.extension.ExtensionContext
+
+/**
+ * This extension provides a way to mock navigation arguments getters.
+ * It can be used to mock getting navigation arguments from savedStateHandle, like: savedStateHandle.navArgs()
+ * or to get arguments from the specific Destination object, like: ExampleScreenDestination.argsFrom(savedStateHandle).
+ *
+ * Add this JUnit 5 extension to your test class using
+ * @JvmField
+ * @RegisterExtension
+ * val navigationTestExtension = NavigationTestExtension()
+ *
+ * or:
+ *
+ * Annotating the class with
+ * @ExtendWith(NavigationTestExtension::class)
+ */
+class NavigationTestExtension : BeforeEachCallback, AfterEachCallback {
+    override fun beforeEach(context: ExtensionContext?) {
+        mockkStatic("com.wire.android.feature.cells.ui.NavArgsGettersKt")
+    }
+
+    override fun afterEach(context: ExtensionContext?) {
+        unmockkStatic("com.wire.android.feature.cells.ui.NavArgsGettersKt")
+    }
+}