phantom

wallets/phantom/src/PhantomWallet.ts

@@ -1,16 +1,12 @@
 import { type BrowserContext, type Page } from '@playwright/test'
-import { ConfirmationPage } from './pages/ConfirmationPage/page'
 import { HomePage } from './pages/HomePage/page'
 import { LockPage } from './pages/LockPage/page'
 import { NotificationPage } from './pages/NotificationPage/page'
-import { SettingsPage } from './pages/SettingsPage/page'
 
 export class PhantomWallet {
   readonly lockPage: LockPage
   readonly homePage: HomePage
   readonly notificationPage: NotificationPage
-  readonly settingsPage: SettingsPage
-  readonly confirmationPage: ConfirmationPage
 
   constructor(
     readonly page: Page,
@@ -21,8 +17,6 @@ export class PhantomWallet {
     this.lockPage = new LockPage(page)
     this.homePage = new HomePage(page)
     this.notificationPage = new NotificationPage(page)
-    this.settingsPage = new SettingsPage(page)
-    this.confirmationPage = new ConfirmationPage(page)
   }
   /**
    * Does initial setup for the wallet.
@@ -86,4 +80,12 @@ export class PhantomWallet {
   async rejectTransaction() {
     await this.notificationPage.rejectTransaction(this.extensionId!)
   }
+
+  async backToMain() {
+    await this.homePage.backToMain()
+  }
+
+  async selectDefaultWallet() {
+    await this.homePage.selectDefaultWallet(this.extensionId!, 'phantom')
+  }
 }

wallets/phantom/src/pages/HomePage/actions/backToMain.ts

@@ -0,0 +1,8 @@
+import type { Page } from '@playwright/test'
+import { homePageElements } from '../selectors'
+
+export const backToMain = async (page: Page) => {
+  await page.waitForLoadState('domcontentloaded')
+  await page.click(homePageElements.connectedSites.trustedAppsBackButton)
+  await page.click(homePageElements.settingsMenu.settingsSidebarCloseButton)
+}

wallets/phantom/src/pages/HomePage/actions/index.ts

@@ -2,3 +2,5 @@ export * from './changeAccount'
 export * from './closePopupAndTooltips'
 export * from './getWalletAddress'
 export * from './disconnectFromApp'
+export * from './selectDefaultWallet'
+export * from './backToMain'

wallets/phantom/src/pages/HomePage/actions/selectDefaultWallet.ts

@@ -0,0 +1,15 @@
+import type { Page } from '@playwright/test'
+import { getNotificationPageAndWaitForLoad } from '../../../utils/getNotificationAndWaitForLoads'
+import { homePageElements } from '../selectors'
+
+export const selectDefaultWallet = async (page: Page, extensionId: string, wallet: string) => {
+  console.log(wallet)
+  const notificationPage = await getNotificationPageAndWaitForLoad(page.context(), extensionId)
+  const walletSelector = homePageElements.defaultWallet['phantom']
+  await notificationPage.click(homePageElements.settingsMenu.settingsMenuButton)
+  await notificationPage.click(homePageElements.settingsMenu.settingsSidebarButton)
+  await notificationPage.click(homePageElements.settingsMenu.settingsPreferencesButton)
+  await notificationPage.click(homePageElements.settingsMenu.defaultAppWalletRow)
+  await notificationPage.click(walletSelector)
+  await notificationPage.click(homePageElements.connectedSites.trustedAppsBackButton)
+}

wallets/phantom/src/pages/HomePage/page.ts

@@ -1,5 +1,5 @@
 import type { Page } from '@playwright/test'
-import { changeAccount, closePopupAndTooltips, disconnectFromApp, getWalletAddress } from './actions'
+import { changeAccount, closePopupAndTooltips, disconnectFromApp, getWalletAddress, selectDefaultWallet, backToMain } from './actions'
 import { homePageElements } from './selectors'
 
 export class HomePage {
@@ -27,4 +27,12 @@ export class HomePage {
   async disconnectFromApp(extensionId: string) {
     await disconnectFromApp(this.page, extensionId)
   }
+
+  async selectDefaultWallet(extensionId: string, wallet: string) {
+    await selectDefaultWallet(this.page, extensionId, wallet)
+  }
+
+  async backToMain() {
+    await backToMain(this.page)
+  }
 }

wallets/phantom/src/pages/NotificationPage/actions/lock.ts

@@ -1,6 +1,6 @@
 import type { Page } from '@playwright/test'
 import { closePopupAndTooltips } from '../../HomePage/actions/closePopupAndTooltips'
-import { settingsPageElements } from '../../SettingsPage/selectors'
+import { settingsPageElements } from '../selectors'
 import { notificationPageElements } from '../selectors'
 
 export const lock = async (page: Page, extensionId: string) => {

wallets/phantom/src/pages/NotificationPage/selectors/index.ts

@@ -174,6 +174,13 @@ const signaturePageElements = {
   }
 }
 
+export const settingsPageElements = {
+  buttons: {
+    lockWallet: '[data-testid="lock-menu-item"]'
+  }
+}
+
+
 const transactionPageElements = {
   buttons: {
     confirmTransaction: '[data-testid="primary-button"]',