test(web-extension): use behaviour subject to update store state between assertions

Author: greatertomi

packages/web-extension/test/walletManager/WalletRepository.test.ts

@@ -11,10 +11,10 @@ import {
   WalletRepositoryDependencies,
   WalletType
 } from '../../src';
+import { BehaviorSubject, firstValueFrom, of } from 'rxjs';
 import { Cardano, Serialization } from '@cardano-sdk/core';
 import { Hash28ByteBase16 } from '@cardano-sdk/crypto';
 import { KeyPurpose, KeyRole } from '@cardano-sdk/key-management';
-import { firstValueFrom, of } from 'rxjs';
 import { logger } from '@cardano-sdk/util-dev';
 import pick from 'lodash/pick';
 
@@ -210,6 +210,9 @@ describe('WalletRepository', () => {
     });
 
     it('allows creating 1852 and 1854 purpose account with the same index', async () => {
+      const storeSubject = new BehaviorSubject([storedLedgerWallet]);
+      store.observeAll.mockReturnValue(storeSubject.asObservable());
+
       const standardAccountProps = createAccount(
         0,
         storedLedgerWallet.accounts[storedLedgerWallet.accounts.length - 1].accountIndex + 1,
@@ -220,14 +223,14 @@ describe('WalletRepository', () => {
         walletId: storedLedgerWallet.walletId
       };
 
-      const multiSigAccountProps2 = createAccount(
+      const multiSigAccountProps = createAccount(
         0,
         storedLedgerWallet.accounts[storedLedgerWallet.accounts.length - 1].accountIndex + 1,
         KeyPurpose.MULTI_SIG
       );
 
       const multiSigProps = {
-        ...multiSigAccountProps2,
+        ...multiSigAccountProps,
         walletId: storedLedgerWallet.walletId
       };
 
@@ -239,11 +242,18 @@ describe('WalletRepository', () => {
         }
       ]);
 
+      storeSubject.next([
+        {
+          ...storedLedgerWallet,
+          accounts: [...storedLedgerWallet.accounts, standardAccountProps]
+        }
+      ]);
+
       await expect(repository.addAccount(multiSigProps)).resolves.toEqual(multiSigProps);
       expect(store.setAll).toBeCalledWith([
         {
           ...storedLedgerWallet,
-          accounts: [...storedLedgerWallet.accounts, multiSigAccountProps2]
+          accounts: [...storedLedgerWallet.accounts, standardAccountProps, multiSigAccountProps]
         }
       ]);
     });
@@ -333,7 +343,8 @@ describe('WalletRepository', () => {
     });
 
     it('does not update 1852 account metadata when updating 1854 account', async () => {
-      const newAccount = createAccount(0, 1, KeyPurpose.MULTI_SIG);
+      const storedAccount = storedLedgerWallet.accounts[0];
+      const newAccount = createAccount(0, storedAccount.accountIndex, KeyPurpose.MULTI_SIG);
       const accounts = [storedLedgerWallet.accounts[0], newAccount];
       store.observeAll.mockReturnValueOnce(of([{ ...storedLedgerWallet, accounts }]));