fix: broken tests after merge

WalletConnect · Mar 27, 2024 · 1854f0a · 1854f0a
1 parent 39cae7a
commit 1854f0a
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 7 deletions.
diff --git a/packages/core/src/controllers/AccountController.ts b/packages/core/src/controllers/AccountController.ts
@@ -27,7 +27,8 @@ type StateKey = keyof AccountControllerState
 const state = proxy<AccountControllerState>({
   isConnected: false,
   currentTab: 0,
-  tokenBalance: []
+  tokenBalance: [],
+  smartAccountDeployed: false
 })
 
 // -- Controller ---------------------------------------- //
@@ -108,5 +109,6 @@ export const AccountController = {
     state.profileName = undefined
     state.profileImage = undefined
     state.addressExplorerUrl = undefined
+    state.tokenBalance = []
   }
 }
diff --git a/packages/core/tests/controllers/AccountController.test.ts b/packages/core/tests/controllers/AccountController.test.ts
@@ -12,7 +12,12 @@ const explorerUrl = 'https://some.explorer.com/explore'
 // -- Tests --------------------------------------------------------------------
 describe('AccountController', () => {
   it('should have valid default state', () => {
-    expect(AccountController.state).toEqual({ isConnected: false })
+    expect(AccountController.state).toEqual({
+      isConnected: false,
+      smartAccountDeployed: false,
+      currentTab: 0,
+      tokenBalance: []
+    })
   })
 
   it('should update state correctly on setIsConnected()', () => {
@@ -54,6 +59,18 @@ describe('AccountController', () => {
 
   it('should update state correctly on resetAccount()', () => {
     AccountController.resetAccount()
-    expect(AccountController.state).toEqual({ isConnected: false, smartAccountDeployed: false })
+    expect(AccountController.state).toEqual({
+      isConnected: false,
+      smartAccountDeployed: false,
+      currentTab: 0,
+      caipAddress: undefined,
+      address: undefined,
+      balance: undefined,
+      balanceSymbol: undefined,
+      profileName: undefined,
+      profileImage: undefined,
+      addressExplorerUrl: undefined,
+      tokenBalance: []
+    })
   })
 })
diff --git a/packages/core/tests/controllers/ConnectorController.test.ts b/packages/core/tests/controllers/ConnectorController.test.ts
@@ -15,13 +15,17 @@ const emailProvider = {
   syncTheme: (_args: { themeMode: ThemeMode; themeVariables: ThemeVariables }) => Promise.resolve()
 }
 
-const walletConnectConnector = { id: 'walletConnect', type: 'WALLET_CONNECT' } as const
+const walletConnectConnector = {
+  id: 'walletConnect',
+  explorerId: 'walletConnect',
+  type: 'WALLET_CONNECT'
+} as const
 const externalConnector = { id: 'external', type: 'EXTERNAL' } as const
 const emailConnector = { id: 'w3mEmail', type: 'EMAIL', provider: emailProvider } as const
 const announcedConnector = {
-  id: 'metamask',
+  id: 'announced',
   type: 'ANNOUNCED',
-  info: { rdns: 'metamask.io' }
+  info: { rdns: 'announced.io' }
 } as const
 
 const syncDappDataSpy = vi.spyOn(emailProvider, 'syncDappData')
@@ -88,6 +92,7 @@ describe('ConnectorController', () => {
     expect(ConnectorController.state.connectors).toEqual([
       walletConnectConnector,
       externalConnector,
+      metamaskConnector,
       emailConnector
     ])
 
@@ -105,13 +110,14 @@ describe('ConnectorController', () => {
 
   it('getAnnouncedConnectorRdns() should return corresponding info array', () => {
     ConnectorController.addConnector(announcedConnector)
-    expect(ConnectorController.getAnnouncedConnectorRdns()).toEqual(['metamask.io'])
+    expect(ConnectorController.getAnnouncedConnectorRdns()).toEqual(['announced.io'])
   })
 
   it('getConnnectors() should return all connectors', () => {
     expect(ConnectorController.getConnectors()).toEqual([
       walletConnectConnector,
       externalConnector,
+      metamaskConnector,
       emailConnector,
       announcedConnector
     ])