test: fix unit tests for QNN backend

crim50n · crim50n · commit 2ec511b35ad7 · 2026-01-01T17:10:29.000+03:00
Update test mocks and fixtures to support QNN repository and use cases:
- Add QnnGenerationRepository mocks to generation use case tests
- Add GetForgeModulesUseCase to generation viewmodel tests
- Add QNN preferences to configuration tests
- Fix mapper tests for model name field
- Add missing parameters to presentation viewmodel tests
diff --git a/data/src/test/java/com/shifthackz/aisdv1/data/local/DownloadableModelLocalDataSourceTest.kt b/data/src/test/java/com/shifthackz/aisdv1/data/local/DownloadableModelLocalDataSourceTest.kt
@@ -217,15 +217,19 @@ class DownloadableModelLocalDataSourceTest {
     }
 
     @Test
-    fun `given attempt to get model by id, dao throws exception, expected error true`() {
+    fun `given attempt to get model by id, dao throws exception and no builtin model found, expected error`() {
         every {
             stubDao.queryById(any())
         } returns Single.error(stubException)
 
+        every {
+            stubQnnModelsBuiltInDataSource.getAll()
+        } returns Single.just(emptyList())
+
         localDataSource
             .getById("5598")
             .test()
-            .assertError(stubException)
+            .assertError { it is NoSuchElementException }
             .assertNoValues()
             .await()
             .assertNotComplete()
diff --git a/data/src/test/java/com/shifthackz/aisdv1/data/mocks/GenerationResultEntityMocks.kt b/data/src/test/java/com/shifthackz/aisdv1/data/mocks/GenerationResultEntityMocks.kt
@@ -26,7 +26,7 @@ val mockGenerationResultEntity = GenerationResultEntity(
     mediaPath = "",
     inputMediaPath = "",
     mediaType = "IMAGE",
-    modelName = "",
+    modelName = "MockModel",
 )
 
 val mockGenerationResultEntities = listOf(mockGenerationResultEntity)
diff --git a/data/src/test/java/com/shifthackz/aisdv1/data/remote/DownloadableModelRemoteDataSourceTest.kt b/data/src/test/java/com/shifthackz/aisdv1/data/remote/DownloadableModelRemoteDataSourceTest.kt
@@ -29,9 +29,13 @@ class DownloadableModelRemoteDataSourceTest {
         whenever(stubApi.fetchMediaPipeModels())
             .thenReturn(Single.just(mockDownloadableModelsResponse))
 
+        whenever(stubApi.fetchQnnModels())
+            .thenReturn(Single.just(mockDownloadableModelsResponse))
+
         val expected = listOf(
             mockDownloadableModelsResponse.mapRawToCheckpointDomain(LocalAiModel.Type.ONNX),
             mockDownloadableModelsResponse.mapRawToCheckpointDomain(LocalAiModel.Type.MediaPipe),
+            mockDownloadableModelsResponse.mapRawToCheckpointDomain(LocalAiModel.Type.QNN),
         ).flatten()
 
         remoteDataSource
@@ -51,6 +55,9 @@ class DownloadableModelRemoteDataSourceTest {
         whenever(stubApi.fetchMediaPipeModels())
             .thenReturn(Single.just(emptyList()))
 
+        whenever(stubApi.fetchQnnModels())
+            .thenReturn(Single.just(emptyList()))
+
         remoteDataSource
             .fetch()
             .test()
@@ -68,6 +75,9 @@ class DownloadableModelRemoteDataSourceTest {
         whenever(stubApi.fetchMediaPipeModels())
             .thenReturn(Single.error(stubException))
 
+        whenever(stubApi.fetchQnnModels())
+            .thenReturn(Single.error(stubException))
+
         remoteDataSource
             .fetch()
             .test()
diff --git a/domain/src/test/java/com/shifthackz/aisdv1/domain/mocks/ConfigurationMocks.kt b/domain/src/test/java/com/shifthackz/aisdv1/domain/mocks/ConfigurationMocks.kt
@@ -17,4 +17,6 @@ val mockConfiguration = Configuration(
     localOnnxModelPath = "/storage/emulated/0/5598",
     localMediaPipeModelId = "5598",
     localMediaPipeModelPath = "/storage/emulated/0/5598",
+    localQnnModelId = "5598",
+    localQnnModelPath = "/storage/emulated/0/5598",
 )
diff --git a/domain/src/test/java/com/shifthackz/aisdv1/domain/usecase/generation/ImageToImageUseCaseImplTest.kt b/domain/src/test/java/com/shifthackz/aisdv1/domain/usecase/generation/ImageToImageUseCaseImplTest.kt
@@ -10,6 +10,7 @@ import com.shifthackz.aisdv1.domain.preference.PreferenceManager
 import com.shifthackz.aisdv1.domain.repository.HordeGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.HuggingFaceGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.StabilityAiGenerationRepository
+import com.shifthackz.aisdv1.domain.repository.QnnGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.StableDiffusionGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.SwarmUiGenerationRepository
 import io.reactivex.rxjava3.core.Single
@@ -23,6 +24,7 @@ class ImageToImageUseCaseImplTest {
     private val stubHordeGenerationRepository = mock<HordeGenerationRepository>()
     private val stubHuggingFaceGenerationRepository = mock<HuggingFaceGenerationRepository>()
     private val stubStabilityAiGenerationRepository = mock<StabilityAiGenerationRepository>()
+    private val stubQnnGenerationRepository = mock<QnnGenerationRepository>()
     private val stubPreferenceManager = mock<PreferenceManager>()
 
     private val useCase = ImageToImageUseCaseImpl(
@@ -31,6 +33,7 @@ class ImageToImageUseCaseImplTest {
         hordeGenerationRepository = stubHordeGenerationRepository,
         huggingFaceGenerationRepository = stubHuggingFaceGenerationRepository,
         stabilityAiGenerationRepository = stubStabilityAiGenerationRepository,
+        qnnGenerationRepository = stubQnnGenerationRepository,
         preferenceManager = stubPreferenceManager,
     )
 
diff --git a/domain/src/test/java/com/shifthackz/aisdv1/domain/usecase/generation/InterruptGenerationUseCaseImplTest.kt b/domain/src/test/java/com/shifthackz/aisdv1/domain/usecase/generation/InterruptGenerationUseCaseImplTest.kt
@@ -6,6 +6,7 @@ import com.shifthackz.aisdv1.domain.entity.ServerSource
 import com.shifthackz.aisdv1.domain.preference.PreferenceManager
 import com.shifthackz.aisdv1.domain.repository.HordeGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.LocalDiffusionGenerationRepository
+import com.shifthackz.aisdv1.domain.repository.QnnGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.StableDiffusionGenerationRepository
 import io.reactivex.rxjava3.core.Completable
 import org.junit.Test
@@ -16,12 +17,14 @@ class InterruptGenerationUseCaseImplTest {
     private val stubStableDiffusionGenerationRepository = mock<StableDiffusionGenerationRepository>()
     private val stubHordeGenerationRepository = mock<HordeGenerationRepository>()
     private val stubLocalDiffusionGenerationRepository = mock<LocalDiffusionGenerationRepository>()
+    private val stubQnnGenerationRepository = mock<QnnGenerationRepository>()
     private val stubPreferenceManager = mock<PreferenceManager>()
 
     private val useCase = InterruptGenerationUseCaseImpl(
         stableDiffusionGenerationRepository = stubStableDiffusionGenerationRepository,
         hordeGenerationRepository = stubHordeGenerationRepository,
         localDiffusionGenerationRepository = stubLocalDiffusionGenerationRepository,
+        qnnGenerationRepository = stubQnnGenerationRepository,
         preferenceManager = stubPreferenceManager,
     )
 
diff --git a/domain/src/test/java/com/shifthackz/aisdv1/domain/usecase/generation/TextToImageUseCaseImplTest.kt b/domain/src/test/java/com/shifthackz/aisdv1/domain/usecase/generation/TextToImageUseCaseImplTest.kt
@@ -14,6 +14,7 @@ import com.shifthackz.aisdv1.domain.repository.LocalDiffusionGenerationRepositor
 import com.shifthackz.aisdv1.domain.repository.MediaPipeGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.OpenAiGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.StabilityAiGenerationRepository
+import com.shifthackz.aisdv1.domain.repository.QnnGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.StableDiffusionGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.SwarmUiGenerationRepository
 import io.reactivex.rxjava3.core.Single
@@ -31,6 +32,7 @@ class TextToImageUseCaseImplTest {
     private val stubSwarmUiGenerationRepository = mock<SwarmUiGenerationRepository>()
     private val stubLocalDiffusionGenerationRepository = mock<LocalDiffusionGenerationRepository>()
     private val stubMediaPipeGenerationRepository = mock<MediaPipeGenerationRepository>()
+    private val stubQnnGenerationRepository = mock<QnnGenerationRepository>()
     private val stubPreferenceManager = mock<PreferenceManager>()
 
     private val useCase = TextToImageUseCaseImpl(
@@ -43,6 +45,7 @@ class TextToImageUseCaseImplTest {
         swarmUiGenerationRepository = stubSwarmUiGenerationRepository,
         localDiffusionGenerationRepository = stubLocalDiffusionGenerationRepository,
         mediaPipeGenerationRepository = stubMediaPipeGenerationRepository,
+        qnnGenerationRepository = stubQnnGenerationRepository,
         preferenceManager = stubPreferenceManager,
     )
 
diff --git a/domain/src/test/java/com/shifthackz/aisdv1/domain/usecase/settings/GetConfigurationUseCaseImplTest.kt b/domain/src/test/java/com/shifthackz/aisdv1/domain/usecase/settings/GetConfigurationUseCaseImplTest.kt
@@ -88,6 +88,14 @@ class GetConfigurationUseCaseImplTest {
             stubPreferenceManager::falAiApiKey.get()
         } returns mockConfiguration.falAiApiKey
 
+        every {
+            stubPreferenceManager::localQnnModelId.get()
+        } returns mockConfiguration.localQnnModelId
+
+        every {
+            stubPreferenceManager::localQnnCustomModelPath.get()
+        } returns mockConfiguration.localQnnModelPath
+
         useCase
             .invoke()
             .test()
diff --git a/domain/src/test/java/com/shifthackz/aisdv1/domain/usecase/settings/SetServerConfigurationUseCaseImplTest.kt b/domain/src/test/java/com/shifthackz/aisdv1/domain/usecase/settings/SetServerConfigurationUseCaseImplTest.kt
@@ -87,6 +87,14 @@ class SetServerConfigurationUseCaseImplTest {
             stubPreferenceManager::falAiApiKey.set(any())
         } returns Unit
 
+        every {
+            stubPreferenceManager::localQnnModelId.set(any())
+        } returns Unit
+
+        every {
+            stubPreferenceManager::localQnnCustomModelPath.set(any())
+        } returns Unit
+
         useCase
             .invoke(mockConfiguration)
             .test()
diff --git a/presentation/src/test/java/com/shifthackz/aisdv1/presentation/core/CoreGenerationMviViewModelTest.kt b/presentation/src/test/java/com/shifthackz/aisdv1/presentation/core/CoreGenerationMviViewModelTest.kt
@@ -16,6 +16,7 @@ import com.shifthackz.aisdv1.domain.usecase.generation.ObserveHordeProcessStatus
 import com.shifthackz.aisdv1.domain.usecase.generation.ObserveLocalDiffusionProcessStatusUseCase
 import com.shifthackz.aisdv1.domain.usecase.generation.SaveGenerationResultUseCase
 import com.shifthackz.aisdv1.domain.usecase.sdsampler.GetStableDiffusionSamplersUseCase
+import com.shifthackz.aisdv1.domain.usecase.forgemodule.GetForgeModulesUseCase
 import com.shifthackz.aisdv1.domain.usecase.wakelock.AcquireWakelockUseCase
 import com.shifthackz.aisdv1.domain.usecase.wakelock.ReleaseWakeLockUseCase
 import com.shifthackz.aisdv1.presentation.navigation.router.drawer.DrawerRouter
@@ -55,6 +56,7 @@ abstract class CoreGenerationMviViewModelTest<V : GenerationMviViewModel<*, *, *
     protected val stubWakeLockInterActor = mockk<WakeLockInterActor>()
     protected val stubBackgroundWorkObserver = mockk<BackgroundWorkObserver>()
     protected val stubBackgroundTaskManager = mockk<BackgroundTaskManager>()
+    protected val stubGetForgeModulesUseCase = mockk<GetForgeModulesUseCase>()
 
     private val stubHordeProcessStatus = BehaviorSubject.create<HordeProcessStatus>()
     private val stubLdStatus = BehaviorSubject.create<LocalDiffusionStatus>()
diff --git a/presentation/src/test/java/com/shifthackz/aisdv1/presentation/screen/gallery/detail/GalleryDetailViewModelTest.kt b/presentation/src/test/java/com/shifthackz/aisdv1/presentation/screen/gallery/detail/GalleryDetailViewModelTest.kt
@@ -10,8 +10,11 @@ import com.shifthackz.aisdv1.core.model.asUiText
 import com.shifthackz.aisdv1.domain.entity.AiGenerationResult
 import com.shifthackz.aisdv1.domain.usecase.caching.GetLastResultFromCacheUseCase
 import com.shifthackz.aisdv1.domain.usecase.gallery.DeleteGalleryItemUseCase
+import com.shifthackz.aisdv1.domain.usecase.gallery.GetGalleryPagedIdsUseCase
 import com.shifthackz.aisdv1.domain.usecase.gallery.ToggleImageVisibilityUseCase
 import com.shifthackz.aisdv1.domain.usecase.generation.GetGenerationResultUseCase
+import com.shifthackz.aisdv1.domain.preference.PreferenceManager
+import com.shifthackz.aisdv1.domain.gateway.MediaStoreGateway
 import com.shifthackz.aisdv1.presentation.core.CoreViewModelTest
 import com.shifthackz.aisdv1.presentation.core.GenerationFormUpdateEvent
 import com.shifthackz.aisdv1.presentation.extensions.mapToUi
@@ -48,6 +51,9 @@ class GalleryDetailViewModelTest : CoreViewModelTest<GalleryDetailViewModel>() {
     private val stubBase64ToBitmapConverter = mockk<Base64ToBitmapConverter>()
     private val stubGenerationFormUpdateEvent = mockk<GenerationFormUpdateEvent>()
     private val stubMainRouter = mockk<MainRouter>()
+    private val stubPreferenceManager = mockk<PreferenceManager>()
+    private val stubGetGalleryPagedIdsUseCase = mockk<GetGalleryPagedIdsUseCase>()
+    private val stubMediaStoreGateway = mockk<MediaStoreGateway>()
 
     override fun initializeViewModel() = GalleryDetailViewModel(
         itemId = 5598L,
@@ -62,6 +68,9 @@ class GalleryDetailViewModelTest : CoreViewModelTest<GalleryDetailViewModel>() {
         schedulersProvider = stubSchedulersProvider,
         generationFormUpdateEvent = stubGenerationFormUpdateEvent,
         mainRouter = stubMainRouter,
+        preferenceManager = stubPreferenceManager,
+        getGalleryPagedIdsUseCase = stubGetGalleryPagedIdsUseCase,
+        mediaStoreGateway = stubMediaStoreGateway,
     )
 
     @Before
diff --git a/presentation/src/test/java/com/shifthackz/aisdv1/presentation/screen/gallery/list/GalleryViewModelTest.kt b/presentation/src/test/java/com/shifthackz/aisdv1/presentation/screen/gallery/list/GalleryViewModelTest.kt
@@ -11,8 +11,10 @@ import com.shifthackz.aisdv1.domain.feature.work.BackgroundWorkObserver
 import com.shifthackz.aisdv1.domain.preference.PreferenceManager
 import com.shifthackz.aisdv1.domain.usecase.gallery.DeleteAllGalleryUseCase
 import com.shifthackz.aisdv1.domain.usecase.gallery.DeleteGalleryItemsUseCase
+import com.shifthackz.aisdv1.domain.usecase.gallery.GetAllGalleryUseCase
 import com.shifthackz.aisdv1.domain.usecase.gallery.GetMediaStoreInfoUseCase
 import com.shifthackz.aisdv1.domain.usecase.generation.GetGenerationResultPagedUseCase
+import com.shifthackz.aisdv1.domain.gateway.MediaStoreGateway
 import com.shifthackz.aisdv1.presentation.core.CoreViewModelTest
 import com.shifthackz.aisdv1.presentation.model.Modal
 import com.shifthackz.aisdv1.presentation.navigation.router.drawer.DrawerRouter
@@ -53,6 +55,8 @@ class GalleryViewModelTest : CoreViewModelTest<GalleryViewModel>() {
     private val stubDeleteGalleryItemsUseCase = mockk<DeleteGalleryItemsUseCase>()
     private val stubBackgroundWorkObserver = mockk<BackgroundWorkObserver>()
     private val stubPreferenceManager = mockk<PreferenceManager>()
+    private val stubMediaStoreGateway = mockk<MediaStoreGateway>()
+    private val stubGetAllGalleryUseCase = mockk<GetAllGalleryUseCase>()
 
     override fun initializeViewModel() = GalleryViewModel(
         dispatchersProvider = stubDispatchersProvider,
@@ -67,6 +71,8 @@ class GalleryViewModelTest : CoreViewModelTest<GalleryViewModel>() {
         drawerRouter = stubDrawerRouter,
         deleteAllGalleryUseCase = stubDeleteAllGalleryUseCase,
         deleteGalleryItemsUseCase = stubDeleteGalleryItemsUseCase,
+        mediaStoreGateway = stubMediaStoreGateway,
+        getAllGalleryUseCase = stubGetAllGalleryUseCase,
     )
 
     @Before
@@ -146,7 +152,7 @@ class GalleryViewModelTest : CoreViewModelTest<GalleryViewModel>() {
         } returns Unit
 
         val item = GalleryGridItemUi(5598L, stubBitmap, false)
-        viewModel.processIntent(GalleryIntent.OpenItem(item))
+        viewModel.processIntent(GalleryIntent.OpenItem(item, 0))
 
         verify {
             stubMainRouter.navigateToGalleryDetails(5598L)
diff --git a/presentation/src/test/java/com/shifthackz/aisdv1/presentation/screen/img2img/ImageToImageViewModelTest.kt b/presentation/src/test/java/com/shifthackz/aisdv1/presentation/screen/img2img/ImageToImageViewModelTest.kt
@@ -77,6 +77,7 @@ class ImageToImageViewModelTest : CoreGenerationMviViewModelTest<ImageToImageVie
         backgroundWorkObserver = stubBackgroundWorkObserver,
         backgroundTaskManager = stubBackgroundTaskManager,
         buildInfoProvider = BuildInfoProvider.stub,
+        getForgeModulesUseCase = stubGetForgeModulesUseCase,
     )
 
     @Before
diff --git a/presentation/src/test/java/com/shifthackz/aisdv1/presentation/screen/setup/ServerSetupViewModelTest.kt b/presentation/src/test/java/com/shifthackz/aisdv1/presentation/screen/setup/ServerSetupViewModelTest.kt
@@ -14,7 +14,10 @@ import com.shifthackz.aisdv1.domain.usecase.downloadable.DeleteModelUseCase
 import com.shifthackz.aisdv1.domain.usecase.downloadable.DownloadModelUseCase
 import com.shifthackz.aisdv1.domain.usecase.downloadable.GetLocalMediaPipeModelsUseCase
 import com.shifthackz.aisdv1.domain.usecase.downloadable.GetLocalOnnxModelsUseCase
+import com.shifthackz.aisdv1.domain.usecase.downloadable.GetLocalQnnModelsUseCase
+import com.shifthackz.aisdv1.domain.usecase.downloadable.ScanCustomModelsUseCase
 import com.shifthackz.aisdv1.domain.usecase.huggingface.FetchAndGetHuggingFaceModelsUseCase
+import com.shifthackz.aisdv1.domain.repository.FalAiEndpointRepository
 import com.shifthackz.aisdv1.domain.usecase.settings.GetConfigurationUseCase
 import com.shifthackz.aisdv1.presentation.core.CoreViewModelTest
 import com.shifthackz.aisdv1.presentation.mocks.mockHuggingFaceModels
@@ -55,6 +58,9 @@ class ServerSetupViewModelTest : CoreViewModelTest<ServerSetupViewModel>() {
     private val stubPreferenceManager = mockk<PreferenceManager>()
     private val stubWakeLockInterActor = mockk<WakeLockInterActor>()
     private val stubMainRouter = mockk<MainRouter>()
+    private val stubGetLocalQnnModelsUseCase = mockk<GetLocalQnnModelsUseCase>()
+    private val stubFalAiEndpointRepository = mockk<FalAiEndpointRepository>()
+    private val stubScanCustomModelsUseCase = mockk<ScanCustomModelsUseCase>()
 
     override fun initializeViewModel() = ServerSetupViewModel(
         launchSource = LaunchSource.SETTINGS,
@@ -74,6 +80,9 @@ class ServerSetupViewModelTest : CoreViewModelTest<ServerSetupViewModel>() {
         wakeLockInterActor = stubWakeLockInterActor,
         mainRouter = stubMainRouter,
         buildInfoProvider = BuildInfoProvider.stub,
+        getLocalQnnModelsUseCase = stubGetLocalQnnModelsUseCase,
+        falAiEndpointRepository = stubFalAiEndpointRepository,
+        scanCustomModelsUseCase = stubScanCustomModelsUseCase,
     )
 
     @Before
@@ -95,6 +104,14 @@ class ServerSetupViewModelTest : CoreViewModelTest<ServerSetupViewModel>() {
         every {
             stubFetchAndGetHuggingFaceModelsUseCase()
         } returns Single.just(mockHuggingFaceModels)
+
+        every {
+            stubGetLocalQnnModelsUseCase()
+        } returns Single.just(emptyList())
+
+        every {
+            stubFalAiEndpointRepository.getAll()
+        } returns Single.just(emptyList())
     }
 
     @After
diff --git a/presentation/src/test/java/com/shifthackz/aisdv1/presentation/screen/txt2img/TextToImageViewModelTest.kt b/presentation/src/test/java/com/shifthackz/aisdv1/presentation/screen/txt2img/TextToImageViewModelTest.kt
@@ -59,6 +59,7 @@ class TextToImageViewModelTest : CoreGenerationMviViewModelTest<TextToImageViewM
         backgroundWorkObserver = stubBackgroundWorkObserver,
         backgroundTaskManager = stubBackgroundTaskManager,
         buildInfoProvider = BuildInfoProvider.stub,
+        getForgeModulesUseCase = stubGetForgeModulesUseCase,
     )
 
     @Before
diff --git a/presentation/src/test/java/com/shifthackz/aisdv1/presentation/widget/engine/EngineSelectionViewModelTest.kt b/presentation/src/test/java/com/shifthackz/aisdv1/presentation/widget/engine/EngineSelectionViewModelTest.kt
@@ -6,6 +6,8 @@ import com.shifthackz.aisdv1.domain.entity.ServerSource
 import com.shifthackz.aisdv1.domain.entity.Settings
 import com.shifthackz.aisdv1.domain.preference.PreferenceManager
 import com.shifthackz.aisdv1.domain.usecase.downloadable.ObserveLocalOnnxModelsUseCase
+import com.shifthackz.aisdv1.domain.usecase.downloadable.GetLocalQnnModelsUseCase
+import com.shifthackz.aisdv1.domain.usecase.downloadable.ScanCustomModelsUseCase
 import com.shifthackz.aisdv1.domain.usecase.huggingface.FetchAndGetHuggingFaceModelsUseCase
 import com.shifthackz.aisdv1.domain.usecase.sdmodel.GetStableDiffusionModelsUseCase
 import com.shifthackz.aisdv1.domain.usecase.sdmodel.SelectStableDiffusionModelUseCase
@@ -50,6 +52,8 @@ class EngineSelectionViewModelTest : CoreViewModelTest<EngineSelectionViewModel>
     private val stubFetchAndGetStabilityAiEnginesUseCase = mockk<FetchAndGetStabilityAiEnginesUseCase>()
     private val stubFetchAndGetHuggingFaceModelsUseCase = mockk<FetchAndGetHuggingFaceModelsUseCase>()
     private val stubFetchAndGetSwarmUiModelsUseCase = mockk<FetchAndGetSwarmUiModelsUseCase>()
+    private val stubGetLocalQnnModelsUseCase = mockk<GetLocalQnnModelsUseCase>()
+    private val stubScanCustomModelsUseCase = mockk<ScanCustomModelsUseCase>()
 
     override fun initializeViewModel() = EngineSelectionViewModel(
         dispatchersProvider = stubDispatchersProvider,
@@ -62,6 +66,8 @@ class EngineSelectionViewModelTest : CoreViewModelTest<EngineSelectionViewModel>
         fetchAndGetStabilityAiEnginesUseCase = stubFetchAndGetStabilityAiEnginesUseCase,
         getHuggingFaceModelsUseCase = stubFetchAndGetHuggingFaceModelsUseCase,
         fetchAndGetSwarmUiModelsUseCase = stubFetchAndGetSwarmUiModelsUseCase,
+        getLocalQnnModelsUseCase = stubGetLocalQnnModelsUseCase,
+        scanCustomModelsUseCase = stubScanCustomModelsUseCase,
     )
 
     @Before
@@ -89,6 +95,10 @@ class EngineSelectionViewModelTest : CoreViewModelTest<EngineSelectionViewModel>
                     onFailure = { t -> Flowable.error(t) },
                 )
             }
+
+        every {
+            stubGetLocalQnnModelsUseCase()
+        } returns Single.just(emptyList())
     }
 
     @After

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@ val mockGenerationResultEntity = GenerationResultEntity(`
`26`	`26`	`mediaPath = "",`
`27`	`27`	`inputMediaPath = "",`
`28`	`28`	`mediaType = "IMAGE",`
`29`		`- modelName = "",`
	`29`	`+ modelName = "MockModel",`
`30`	`30`	`)`
`31`	`31`
`32`	`32`	`val mockGenerationResultEntities = listOf(mockGenerationResultEntity)`
Original file line number	Diff line number	Diff line change
`@@ -17,4 +17,6 @@ val mockConfiguration = Configuration(`
`17`	`17`	`localOnnxModelPath = "/storage/emulated/0/5598",`
`18`	`18`	`localMediaPipeModelId = "5598",`
`19`	`19`	`localMediaPipeModelPath = "/storage/emulated/0/5598",`
	`20`	`+ localQnnModelId = "5598",`
	`21`	`+ localQnnModelPath = "/storage/emulated/0/5598",`
`20`	`22`	`)`