From 86e66fb4237ed7c923bebcdfbf9bc2580a57a6fb Mon Sep 17 00:00:00 2001
From: 0x330a <92654767+0x330a@users.noreply.github.com>
Date: Thu, 5 Oct 2023 10:09:57 +1100
Subject: [PATCH] feat: change storage to regular mocked injected in
 constructor

---
 .../loki/messenger/groups/ClosedGroupViewTests.kt        | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/app/src/androidTest/java/network/loki/messenger/groups/ClosedGroupViewTests.kt b/app/src/androidTest/java/network/loki/messenger/groups/ClosedGroupViewTests.kt
index e2f9fa25a8..2c20bac7cf 100644
--- a/app/src/androidTest/java/network/loki/messenger/groups/ClosedGroupViewTests.kt
+++ b/app/src/androidTest/java/network/loki/messenger/groups/ClosedGroupViewTests.kt
@@ -1,6 +1,7 @@
 package network.loki.messenger.groups
 
 import android.content.Context
+import androidx.test.platform.app.InstrumentationRegistry
 import junit.framework.TestCase.assertNotNull
 import network.loki.messenger.libsession_util.util.Sodium
 import org.junit.Before
@@ -16,6 +17,7 @@ import org.session.libsession.utilities.TextSecurePreferences
 import org.session.libsignal.utilities.Hex
 import org.session.libsignal.utilities.IdPrefix
 import org.session.libsignal.utilities.SessionId
+import org.thoughtcrime.securesms.ApplicationContext
 import org.thoughtcrime.securesms.database.ConfigDatabase
 import org.thoughtcrime.securesms.database.Storage
 import org.thoughtcrime.securesms.dependencies.ConfigFactory
@@ -39,17 +41,16 @@ class ClosedGroupViewTests {
 
     @Before
     fun setup() {
+        val applicationContext = InstrumentationRegistry.getInstrumentation().targetContext
         whenever(textSecurePreferences.getLocalNumber()).thenReturn(userSessionId.hexString())
         val context = mock<Context>()
         val emptyDb = mock<ConfigDatabase> { db ->
             whenever(db.retrieveConfigAndHashes(any(), any())).thenReturn(byteArrayOf())
         }
-        val overridenStorage = Storage(mock(), mock(), ConfigFactory(context, emptyDb) {
+        val overriddenStorage = Storage(applicationContext, mock(), ConfigFactory(context, emptyDb) {
             keyPair.secretKey to userSessionId.hexString()
         }, mock())
-        storage = spy(overridenStorage) { storage ->
-            whenever(storage.createNewGroup(any(), any(), any())).thenCallRealMethod()
-        }
+        storage = overriddenStorage
     }
 
     @Test