Add tests for WalletAccountRepository in CreateAccountHandlerTest

Author: AleksandarIlic

app/src/test/kotlin/net/primal/android/auth/repository/CreateAccountHandlerTest.kt

@@ -22,6 +22,7 @@ import net.primal.android.user.domain.Credential
 import net.primal.android.user.repository.BlossomRepository
 import net.primal.android.user.repository.RelayRepository
 import net.primal.android.user.repository.UserRepository
+import net.primal.domain.account.WalletAccountRepository
 import net.primal.domain.common.exception.NetworkException
 import net.primal.domain.nostr.NostrEvent
 import net.primal.domain.nostr.NostrEventKind
@@ -43,6 +44,7 @@ class CreateAccountHandlerTest {
         blossomRepository: BlossomRepository = mockk(relaxed = true),
         settingsRepository: SettingsRepository = mockk(relaxed = true),
         credentialsStore: CredentialsStore = mockk(relaxed = true),
+        walletAccountRepository: WalletAccountRepository = mockk(relaxed = true),
         eventsSignatureHandler: NostrEventSignatureHandler = FakeNostrNotary(
             expectedSignedNostrEvent = mockk(relaxed = true),
         ),
@@ -56,6 +58,7 @@ class CreateAccountHandlerTest {
             dispatchers = coroutinesTestRule.dispatcherProvider,
             eventsSignatureHandler = eventsSignatureHandler,
             blossomRepository = blossomRepository,
+            walletAccountRepository = walletAccountRepository,
         )
     }
 
@@ -261,6 +264,58 @@ class CreateAccountHandlerTest {
             }
         }
 
+    @Test
+    fun createNostrAccount_callsFetchWalletAccountInfo() =
+        runTest {
+            val keyPair = CryptoUtils.generateHexEncodedKeypair()
+            val walletAccountRepository = mockk<WalletAccountRepository>(relaxed = true)
+            val credentialsStore = mockk<CredentialsStore>(relaxed = true) {
+                coEvery { saveNsec(any()) } returns keyPair.pubKey
+            }
+
+            val handler = createAccountHandler(
+                authRepository = createAuthRepository(),
+                walletAccountRepository = walletAccountRepository,
+                credentialsStore = credentialsStore,
+            )
+
+            handler.createNostrAccount(
+                privateKey = keyPair.privateKey,
+                profileMetadata = ProfileMetadata(displayName = "Test", username = null),
+                interests = emptyList(),
+            )
+
+            coVerify {
+                walletAccountRepository.fetchWalletAccountInfo(keyPair.pubKey)
+            }
+        }
+
+    @Test
+    fun createNostrAccount_setsActiveWalletId() =
+        runTest {
+            val keyPair = CryptoUtils.generateHexEncodedKeypair()
+            val walletAccountRepository = mockk<WalletAccountRepository>(relaxed = true)
+            val credentialsStore = mockk<CredentialsStore>(relaxed = true) {
+                coEvery { saveNsec(any()) } returns keyPair.pubKey
+            }
+
+            val handler = createAccountHandler(
+                authRepository = createAuthRepository(),
+                walletAccountRepository = walletAccountRepository,
+                credentialsStore = credentialsStore,
+            )
+
+            handler.createNostrAccount(
+                privateKey = keyPair.privateKey,
+                profileMetadata = ProfileMetadata(displayName = "Test", username = null),
+                interests = emptyList(),
+            )
+
+            coVerify {
+                walletAccountRepository.setActiveWallet(keyPair.pubKey, keyPair.pubKey)
+            }
+        }
+
     @Test
     fun createNostrAccount_revertsAuthData_ifAnyOfApiCallsFail() =
         runTest {