From e1f4e4f934ee7ab88f73f5bc8fdf9f223d4d2275 Mon Sep 17 00:00:00 2001
From: ganfra <francoisg@element.io>
Date: Fri, 22 Oct 2021 18:38:48 +0200
Subject: [PATCH] Hilt: fix test

---
 .../quads/SharedSecureStorageViewModelTest.kt    | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/vector/src/test/java/im/vector/app/features/crypto/quads/SharedSecureStorageViewModelTest.kt b/vector/src/test/java/im/vector/app/features/crypto/quads/SharedSecureStorageViewModelTest.kt
index 0fff663972..506ac9c7d0 100644
--- a/vector/src/test/java/im/vector/app/features/crypto/quads/SharedSecureStorageViewModelTest.kt
+++ b/vector/src/test/java/im/vector/app/features/crypto/quads/SharedSecureStorageViewModelTest.kt
@@ -46,6 +46,7 @@ class SharedSecureStorageViewModelTest {
 
     private val stringProvider = FakeStringProvider()
     private val session = FakeSession()
+    val args = SharedSecureStorageActivity.Args(keyId = null, emptyList(), "alias")
 
     @Test
     fun `given a key info with passphrase when initialising then step is EnterPassphrase`() {
@@ -123,14 +124,15 @@ class SharedSecureStorageViewModelTest {
         test.assertEvents(SharedSecureStorageViewEvent.Dismiss)
     }
 
-    private fun createViewModel() = SharedSecureStorageViewModel(
-            SharedSecureStorageViewState(),
-            SharedSecureStorageActivity.Args(keyId = null, emptyList(), "alias"),
-            stringProvider.instance,
-            session
-    )
+    private fun createViewModel(): SharedSecureStorageViewModel {
+        return SharedSecureStorageViewModel(
+                SharedSecureStorageViewState(args),
+                stringProvider.instance,
+                session
+        )
+    }
 
-    private fun aViewState(hasPassphrase: Boolean, step: SharedSecureStorageViewState.Step) = SharedSecureStorageViewState(
+    private fun aViewState(hasPassphrase: Boolean, step: SharedSecureStorageViewState.Step) = SharedSecureStorageViewState(args).copy(
             ready = true,
             hasPassphrase = hasPassphrase,
             checkingSSSSAction = Uninitialized,