@avalix/chroma 0.0.11 → 0.0.13

package/src/wallets/talisman.ts

@@ -12,7 +12,11 @@ export const TALISMAN_CONFIG = {
   extensionName: `talisman-extension-${VERSION}`,
 } as const
 
-// Helper function to find extension popup
+/*
+ * Helper function to find extension popup
+ * Coverage excluded: requires real browser context with Chrome extension APIs.
+ */
+/* c8 ignore start */
 async function findExtensionPopup(context: BrowserContext, extensionId: string): Promise<Page> {
   // Wait for extension popup to appear with retry logic
   const maxAttempts = 10
@@ -36,6 +40,7 @@ async function findExtensionPopup(context: BrowserContext, extensionId: string):
 
   throw new Error(`Extension popup not found for ID: ${extensionId}`)
 }
+/* c8 ignore stop */
 
 // Get Talisman extension path
 export async function getTalismanExtensionPath(): Promise<string> {
@@ -54,65 +59,121 @@ export async function getTalismanExtensionPath(): Promise<string> {
   return extensionDir
 }
 
-// Talisman specific Ethereum private key import implementation
-export async function importEthPrivateKey(
+/*
+ * Wallet interaction functions below are excluded from coverage because:
+ * - They require a real Chromium browser with extension support
+ * - They interact with Chrome extension popup pages
+ */
+/* c8 ignore start */
+
+// Helper function to find Talisman onboarding page
+async function findOnboardingPage(
+  context: BrowserContext,
+  extensionId: string,
+): Promise<Page> {
+  // Open new dashboard page
+  const popupUrl = `chrome-extension://${extensionId}/dashboard.html`
+  const newPage = await context.newPage()
+  await newPage.goto(popupUrl)
+  await newPage.waitForLoadState('domcontentloaded')
+
+  // Close any other extension tabs that may have been opened automatically
+  for (const p of context.pages()) {
+    if (p !== newPage && p.url().includes(`chrome-extension://${extensionId}/`)) {
+      await p.close()
+    }
+  }
+
+  return newPage
+}
+
+// Helper function to complete Talisman onboarding flow
+async function completeOnboarding(
+  extensionPage: Page,
+  password: string,
+): Promise<void> {
+  // Bring the onboarding page to front
+  await extensionPage.bringToFront()
+
+  // Wait for the page to load and become interactive
+  await extensionPage.waitForLoadState('domcontentloaded')
+
+  if (await extensionPage.getByRole('button', { name: 'Settings' }).isVisible()) {
+    await extensionPage.getByRole('button', { name: 'Settings' }).click({ force: true })
+    return
+  }
+
+  // Click the get started button
+  await extensionPage.getByTestId('onboarding-get-started-button').click()
+
+  // Fill the password
+  await extensionPage.getByRole('textbox', { name: 'Enter password' }).fill(password)
+  await extensionPage.getByRole('textbox', { name: 'Confirm password' }).fill(password)
+  await extensionPage.getByTestId('onboarding-password-confirm-button').click()
+
+  // Click the no thanks button
+  await extensionPage.getByRole('button', { name: 'No thanks' }).click()
+  await extensionPage.getByTestId('onboarding-enter-talisman-button').click()
+
+  // Navigate directly to settings/general page
+  const extensionId = extensionPage.url().match(/chrome-extension:\/\/([^/]+)/)?.[1]
+  await extensionPage.goto(`chrome-extension://${extensionId}/dashboard.html#/settings/general`)
+  await extensionPage.waitForLoadState('domcontentloaded')
+  await extensionPage.getByRole('link', { name: 'Security & Privacy' }).click()
+
+  // Toggle the risk scan setting
+  await extensionPage.getByTestId('component-toggle-button').first().click()
+}
+
+// Talisman specific Polkadot mnemonic import implementation
+export async function importPolkadotMnemonic(
   page: Page & { __extensionContext: BrowserContext, __extensionId: string },
   { seed, name = 'Test Account', password = 'h3llop0lkadot!' }: WalletAccount,
 ): Promise<void> {
   const context = page.__extensionContext
   const extensionId = page.__extensionId
 
-  // Wait for Talisman to open its onboarding tab with retry logic
-  const maxAttempts = 20
-  const retryDelay = 500
-  let extensionPage: Page | null = null
-
-  for (let attempt = 0; attempt < maxAttempts; attempt++) {
-    const pages = context.pages()
+  const extensionPage = await findOnboardingPage(context, extensionId)
 
-    for (const p of pages) {
-      const url = p.url()
-      if (url.includes('onboarding.html') || url.includes(`chrome-extension://${extensionId}/`)) {
-        extensionPage = p
-        break
-      }
-    }
+  try {
+    await completeOnboarding(extensionPage, password!)
 
-    if (extensionPage) {
-      break
-    }
+    // Import Polkadot account via Recovery Phrase
+    await extensionPage.getByRole('link', { name: 'Manage Accounts' }).click()
+    await extensionPage.getByRole('button', { name: 'Get Started' }).click()
+    await extensionPage.getByRole('button', { name: 'Add Account' }).click()
+    await extensionPage.getByRole('button', { name: 'Import Import an existing' }).click()
+    await extensionPage.getByRole('button', { name: 'Import via Recovery Phrase' }).click()
+    await extensionPage.getByRole('button', { name: 'Polkadot Relay Chain, Asset' }).click()
+    await extensionPage.getByRole('textbox', { name: 'Choose a name' }).fill(name!)
+    await extensionPage.getByRole('textbox', { name: 'Enter your 12 or 24 word' }).fill(seed!)
+    await extensionPage.getByTestId('account-add-mnemonic-import-button').click()
 
-    // If not found, wait before retrying
-    if (attempt < maxAttempts - 1) {
-      await new Promise(resolve => setTimeout(resolve, retryDelay))
-    }
+    await extensionPage.close()
   }
-
-  if (!extensionPage) {
-    throw new Error(`Talisman onboarding page not found after ${maxAttempts} attempts`)
+  catch (error) {
+    console.error('❌ Error during Talisman Polkadot account import:', error)
+    throw error
   }
+}
 
-  try {
-    // Bring the onboarding page to front
-    await extensionPage.bringToFront()
-
-    // Wait for the page to load and become interactive
-    await extensionPage.waitForLoadState('domcontentloaded')
-
-    // Click the get started button
-    await extensionPage.getByTestId('onboarding-get-started-button').click()
+// Talisman specific Ethereum private key import implementation
+export async function importEthPrivateKey(
+  page: Page & { __extensionContext: BrowserContext, __extensionId: string },
+  { seed, name = 'Test Account', password = 'h3llop0lkadot!' }: WalletAccount,
+): Promise<void> {
+  const context = page.__extensionContext
+  const extensionId = page.__extensionId
 
-    // Fill the password
-    await extensionPage.getByRole('textbox', { name: 'Enter password' }).fill(password!)
-    await extensionPage.getByRole('textbox', { name: 'Confirm password' }).fill(password!)
-    await extensionPage.getByTestId('onboarding-password-confirm-button').click()
+  const extensionPage = await findOnboardingPage(context, extensionId)
 
-    // Click the no thanks button
-    await extensionPage.getByRole('button', { name: 'No thanks' }).click()
-    await extensionPage.getByTestId('onboarding-enter-talisman-button').click()
+  try {
+    await completeOnboarding(extensionPage, password!)
 
-    // Import Ethereum account
-    await extensionPage.getByRole('button', { name: 'Add account Create or import' }).click()
+    // Import Ethereum account via Private Key
+    await extensionPage.getByRole('link', { name: 'Manage Accounts' }).click()
+    await extensionPage.getByRole('button', { name: 'Get Started' }).click()
+    await extensionPage.getByRole('button', { name: 'Add Account' }).click()
     await extensionPage.getByRole('button', { name: 'Import Import an existing' }).click()
     await extensionPage.getByRole('button', { name: 'Import via Private Key' }).click()
     await extensionPage.getByRole('button', { name: 'Select account platform' }).click()
@@ -124,7 +185,7 @@ export async function importEthPrivateKey(
     await extensionPage.close()
   }
   catch (error) {
-    console.error('❌ Error during Talisman account import:', error)
+    console.error('❌ Error during Talisman Ethereum account import:', error)
     throw error
   }
 }
@@ -134,14 +195,18 @@ export async function authorizeTalisman(
   page: Page & { __extensionContext: BrowserContext, __extensionId: string },
   options: { accountName?: string } = {},
 ): Promise<void> {
-  const { accountName = 'Test Account' } = options
   const context = page.__extensionContext
   const extensionId = page.__extensionId
+  const { accountName = 'Test Account' } = options
 
   const extensionPopup = await findExtensionPopup(context, extensionId)
+  await extensionPopup.waitForLoadState('domcontentloaded')
 
   // Authorize Talisman account
-  await extensionPopup.getByRole('button', { name: accountName }).click()
+  const accountButton = extensionPopup.getByRole('button', { name: accountName })
+  await accountButton.waitFor({ state: 'visible' })
+  await accountButton.scrollIntoViewIfNeeded()
+  await accountButton.click({ force: true })
   await extensionPopup.getByTestId('connection-connect-button').click()
 
   try {
@@ -161,12 +226,9 @@ export async function approveTalismanTx(
 
   const extensionPopup = await findExtensionPopup(context, extensionId)
 
-  try {
+  if (await extensionPopup.getByRole('button', { name: 'Yes' }).isVisible()) {
     await extensionPopup.getByRole('button', { name: 'Yes' }).click()
   }
-  catch {
-    console.log('No another popup found, skipping')
-  }
 
   await extensionPopup.getByRole('button', { name: 'Approve' }).click()
 }
@@ -179,5 +241,11 @@ export async function rejectTalismanTx(
   const extensionId = page.__extensionId
 
   const extensionPopup = await findExtensionPopup(context, extensionId)
-  await extensionPopup.getByTestId('connection-reject-button').click()
+
+  const rejectButton = extensionPopup.getByTestId('connection-reject-button')
+    .or(extensionPopup.getByRole('button', { name: 'Cancel' }))
+
+  await rejectButton.click()
 }
+
+/* c8 ignore stop */