Update getMultiAccountBalancesEnabled specs

MetaMask · Dec 4, 2023 · af38a0f · af38a0f
1 parent f4a814f
commit af38a0f
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/packages/assets-controllers/src/AccountTrackerController.test.ts b/packages/assets-controllers/src/AccountTrackerController.test.ts
@@ -185,7 +185,9 @@ describe('AccountTrackerController', () => {
       });
 
       it('should update only selected address balance when multi-account is disabled', async () => {
-        mockedQuery.mockReturnValueOnce(Promise.resolve('0x10'));
+        mockedQuery
+          .mockReturnValueOnce(Promise.resolve('0x10'))
+          .mockReturnValueOnce(Promise.resolve('0x11'));
 
         const controller = new AccountTrackerController(
           {
@@ -221,8 +223,9 @@ describe('AccountTrackerController', () => {
       });
 
       it('should update all address balances when multi-account is enabled', async () => {
-        mockedQuery.mockReturnValueOnce(Promise.resolve('0x11'));
-        mockedQuery.mockReturnValueOnce(Promise.resolve('0x12'));
+        mockedQuery
+          .mockReturnValueOnce(Promise.resolve('0x11'))
+          .mockReturnValueOnce(Promise.resolve('0x12'));
 
         const controller = new AccountTrackerController(
           {
@@ -363,7 +366,9 @@ describe('AccountTrackerController', () => {
       });
 
       it('should update only selected address balance when multi-account is disabled', async () => {
-        mockedQuery.mockReturnValueOnce(Promise.resolve('0x10'));
+        mockedQuery
+          .mockReturnValueOnce(Promise.resolve('0x10'))
+          .mockReturnValueOnce(Promise.resolve('0x11'));
 
         const controller = new AccountTrackerController({
           onPreferencesStateChange: sinon.stub(),
@@ -405,8 +410,9 @@ describe('AccountTrackerController', () => {
       });
 
       it('should update all address balances when multi-account is enabled', async () => {
-        mockedQuery.mockReturnValueOnce(Promise.resolve('0x11'));
-        mockedQuery.mockReturnValueOnce(Promise.resolve('0x12'));
+        mockedQuery
+          .mockReturnValueOnce(Promise.resolve('0x11'))
+          .mockReturnValueOnce(Promise.resolve('0x12'));
 
         const controller = new AccountTrackerController({
           onPreferencesStateChange: sinon.stub(),