@avalix/chroma 0.0.8 → 0.0.9

package/dist/index.d.ts

@@ -21,6 +21,7 @@ interface BaseWalletInstance {
   approveTx: (options?: {
     password?: string;
   }) => Promise<void>;
+  rejectTx: () => Promise<void>;
 }
 interface PolkadotJsWalletInstance extends BaseWalletInstance {
   type: 'polkadot-js';

package/dist/index.js

@@ -9,14 +9,22 @@ const POLKADOT_JS_CONFIG = {
 	extensionName: "polkadot-extension-0.61.7"
 };
 async function findExtensionPopup$1(context, extensionId) {
-	const pages = context.pages();
-	for (const p of pages) if (p.url().includes(`chrome-extension://${extensionId}/`)) return p;
+	const maxAttempts = 10;
+	const retryDelay = 500;
+	for (let attempt = 0; attempt < maxAttempts; attempt++) {
+		const pages = context.pages();
+		for (const p of pages) if (p.url().includes(`chrome-extension://${extensionId}/`)) {
+			await p.waitForLoadState("domcontentloaded");
+			return p;
+		}
+		if (attempt < maxAttempts - 1) await new Promise((resolve) => setTimeout(resolve, retryDelay));
+	}
 	throw new Error(`Extension popup not found for ID: ${extensionId}`);
 }
 async function getPolkadotJSExtensionPath() {
 	const extensionsDir = path.resolve(process.cwd(), ".chroma");
 	const extensionDir = path.join(extensionsDir, POLKADOT_JS_CONFIG.extensionName);
-	if (!fs.existsSync(extensionDir) || fs.readdirSync(extensionDir).length === 0) throw new Error(`Polkadot-JS extension not found at: ${extensionDir}\n\nPlease download the extension first by running:\n  npx @avalix/chroma download-extensions\n\nOr if you're using this as a dependency:\n  npm run chroma:download\n`);
+	if (!fs.existsSync(extensionDir) || fs.readdirSync(extensionDir).length === 0) throw new Error(`Polkadot-JS extension not found at: ${extensionDir}\n\nPlease download the extension first by running:\n  npx @avalix/chroma download-extensions\n`);
 	console.log(`✅ Found Polkadot-JS extension at: ${extensionDir}`);
 	return extensionDir;
 }
@@ -46,7 +54,6 @@ async function importPolkadotJSAccount(page, { seed, name = "Test Account", pass
 async function authorizePolkadotJS(page) {
 	const context = page.__extensionContext;
 	const extensionId = page.__extensionId;
-	await new Promise((resolve) => setTimeout(resolve, 1e3));
 	const extensionPopup = await findExtensionPopup$1(context, extensionId);
 	await extensionPopup.getByText("Select all").click();
 	await extensionPopup.getByRole("button", { name: /Connect \d+ account\(s\)/ }).click();
@@ -56,12 +63,17 @@ async function approvePolkadotJSTx(page, options = {}) {
 	const { password = "h3llop0lkadot!" } = options;
 	const context = page.__extensionContext;
 	const extensionId = page.__extensionId;
-	await new Promise((resolve) => setTimeout(resolve, 1e3));
 	const extensionPopup = await findExtensionPopup$1(context, extensionId);
 	await extensionPopup.getByRole("textbox").fill(password);
 	await extensionPopup.getByRole("button", { name: "Sign the transaction" }).click();
 	console.log("✅ Polkadot-JS transaction signed successfully");
 }
+async function rejectPolkadotJSTx(page) {
+	const context = page.__extensionContext;
+	const extensionId = page.__extensionId;
+	await (await findExtensionPopup$1(context, extensionId)).getByRole("link", { name: "Cancel" }).click();
+	console.log("✅ Polkadot-JS transaction rejected successfully");
+}
 
 //#endregion
 //#region src/wallets/talisman.ts
@@ -70,45 +82,55 @@ const TALISMAN_CONFIG = {
 	extensionName: "talisman-extension-3.0.5"
 };
 async function findExtensionPopup(context, extensionId) {
-	await new Promise((resolve) => setTimeout(resolve, 1e3));
-	const pages = context.pages();
-	for (const p of pages) if (p.url().includes(`chrome-extension://${extensionId}/`)) {
-		p.setViewportSize({
-			width: 400,
-			height: 600
-		});
-		p.waitForLoadState("domcontentloaded");
-		return p;
+	const maxAttempts = 10;
+	const retryDelay = 500;
+	for (let attempt = 0; attempt < maxAttempts; attempt++) {
+		const pages = context.pages();
+		for (const p of pages) if (p.url().includes(`chrome-extension://${extensionId}/`)) {
+			await p.setViewportSize({
+				width: 400,
+				height: 600
+			});
+			await p.waitForLoadState("domcontentloaded");
+			return p;
+		}
+		if (attempt < maxAttempts - 1) await new Promise((resolve) => setTimeout(resolve, retryDelay));
 	}
 	throw new Error(`Extension popup not found for ID: ${extensionId}`);
 }
 async function getTalismanExtensionPath() {
 	const extensionsDir = path.resolve(process.cwd(), ".chroma");
 	const extensionDir = path.join(extensionsDir, TALISMAN_CONFIG.extensionName);
-	if (!fs.existsSync(extensionDir) || fs.readdirSync(extensionDir).length === 0) throw new Error(`Talisman extension not found at: ${extensionDir}\n\nPlease download the extension first by running:\n  npx @avalix/chroma download-extensions\n\nOr if you're using this as a dependency:\n  npm run chroma:download\n`);
+	if (!fs.existsSync(extensionDir) || fs.readdirSync(extensionDir).length === 0) throw new Error(`Talisman extension not found at: ${extensionDir}\n\nPlease download the extension first by running:\n  npx @avalix/chroma download-extensions\n`);
 	console.log(`✅ Found Talisman extension at: ${extensionDir}`);
 	return extensionDir;
 }
 async function importEthPrivateKey(page, { seed, name = "Test Account", password = "h3llop0lkadot!" }) {
 	const context = page.__extensionContext;
 	const extensionId = page.__extensionId;
-	await new Promise((resolve) => setTimeout(resolve, 2e3));
+	const maxAttempts = 20;
+	const retryDelay = 500;
 	let extensionPage = null;
-	const pages = context.pages();
-	for (const page$1 of pages) {
-		const url = page$1.url();
-		console.log(`📄 Found page: ${url}`);
-		if (url.includes("onboarding.html") || url.includes(`chrome-extension://${extensionId}/`)) {
-			extensionPage = page$1;
-			console.log(`✅ Found Talisman onboarding page: ${url}`);
-			break;
+	for (let attempt = 0; attempt < maxAttempts; attempt++) {
+		const pages = context.pages();
+		for (const p of pages) {
+			const url = p.url();
+			console.log(`📄 Found page: ${url}`);
+			if (url.includes("onboarding.html") || url.includes(`chrome-extension://${extensionId}/`)) {
+				extensionPage = p;
+				console.log(`✅ Found Talisman onboarding page: ${url}`);
+				break;
+			}
+		}
+		if (extensionPage) break;
+		if (attempt < maxAttempts - 1) {
+			console.log(`⏳ Attempt ${attempt + 1}/${maxAttempts}: Waiting for onboarding page...`);
+			await new Promise((resolve) => setTimeout(resolve, retryDelay));
 		}
 	}
-	if (!extensionPage) throw new Error(`Talisman onboarding page not found`);
+	if (!extensionPage) throw new Error(`Talisman onboarding page not found after ${maxAttempts} attempts`);
 	try {
 		await extensionPage.bringToFront();
-		console.log("🔄 Reloading onboarding page for fresh state...");
-		await extensionPage.reload();
 		await extensionPage.waitForLoadState("domcontentloaded");
 		await extensionPage.getByTestId("onboarding-get-started-button").click();
 		await extensionPage.getByRole("textbox", { name: "Enter password" }).fill(password);
@@ -135,7 +157,6 @@ async function authorizeTalisman(page, options = {}) {
 	const { accountName = "Test Account" } = options;
 	const context = page.__extensionContext;
 	const extensionId = page.__extensionId;
-	await new Promise((resolve) => setTimeout(resolve, 1e3));
 	const extensionPopup = await findExtensionPopup(context, extensionId);
 	await extensionPopup.getByRole("button", { name: accountName }).click();
 	await extensionPopup.getByTestId("connection-connect-button").click();
@@ -148,9 +169,13 @@ async function authorizeTalisman(page, options = {}) {
 async function approveTalismanTx(page) {
 	const context = page.__extensionContext;
 	const extensionId = page.__extensionId;
-	await new Promise((resolve) => setTimeout(resolve, 1e3));
 	await (await findExtensionPopup(context, extensionId)).getByRole("button", { name: "Approve" }).click();
 }
+async function rejectTalismanTx(page) {
+	const context = page.__extensionContext;
+	const extensionId = page.__extensionId;
+	await (await findExtensionPopup(context, extensionId)).getByTestId("connection-reject-button").click();
+}
 
 //#endregion
 //#region src/context-playwright/types.ts
@@ -206,6 +231,19 @@ function createWalletInstance(walletType, extensionId, context) {
 					break;
 				default: throw new Error(`Unsupported wallet type: ${walletType}`);
 			}
+		},
+		rejectTx: async () => {
+			const page = context.pages()[0] || await context.newPage();
+			const extPage = createExtendedPage(page, context, extensionId);
+			switch (walletType) {
+				case "polkadot-js":
+					await rejectPolkadotJSTx(extPage);
+					break;
+				case "talisman":
+					await rejectTalismanTx(extPage);
+					break;
+				default: throw new Error(`Unsupported wallet type: ${walletType}`);
+			}
 		}
 	};
 	switch (walletType) {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@avalix/chroma",
   "type": "module",
-  "version": "0.0.8",
+  "version": "0.0.9",
   "description": "End-to-end testing library for Polkadot wallet interactions",
   "author": "Avalix Labs",
   "license": "MIT",

package/src/context-playwright/types.ts

@@ -25,6 +25,7 @@ export interface BaseWalletInstance {
   importMnemonic: (options: WalletAccount) => Promise<void>
   authorize: (options?: { accountName?: string }) => Promise<void>
   approveTx: (options?: { password?: string }) => Promise<void>
+  rejectTx: () => Promise<void>
 }
 
 // Polkadot-JS specific wallet instance

package/src/context-playwright/wallet-factory.ts

@@ -10,11 +10,13 @@ import {
   approvePolkadotJSTx,
   authorizePolkadotJS,
   importPolkadotJSAccount,
+  rejectPolkadotJSTx,
 } from '../wallets/polkadot-js.js'
 import {
   approveTalismanTx,
   authorizeTalisman,
   importEthPrivateKey,
+  rejectTalismanTx,
 } from '../wallets/talisman.js'
 
 // Helper to create extended page with wallet context
@@ -87,6 +89,21 @@ export function createWalletInstance(
           throw new Error(`Unsupported wallet type: ${walletType}`)
       }
     },
+    rejectTx: async () => {
+      const page = context.pages()[0] || await context.newPage()
+      const extPage = createExtendedPage(page, context, extensionId)
+
+      switch (walletType) {
+        case 'polkadot-js':
+          await rejectPolkadotJSTx(extPage)
+          break
+        case 'talisman':
+          await rejectTalismanTx(extPage)
+          break
+        default:
+          throw new Error(`Unsupported wallet type: ${walletType}`)
+      }
+    },
   }
 
   // Return wallet-specific instance with type discriminator

package/src/wallets/polkadot-js.ts

@@ -12,12 +12,25 @@ export const POLKADOT_JS_CONFIG = {
 
 // Helper function to find extension popup
 async function findExtensionPopup(context: BrowserContext, extensionId: string): Promise<Page> {
-  const pages = context.pages()
-  for (const p of pages) {
-    if (p.url().includes(`chrome-extension://${extensionId}/`)) {
-      return p
+  // Wait for extension popup to appear with retry logic
+  const maxAttempts = 10
+  const retryDelay = 500
+
+  for (let attempt = 0; attempt < maxAttempts; attempt++) {
+    const pages = context.pages()
+    for (const p of pages) {
+      if (p.url().includes(`chrome-extension://${extensionId}/`)) {
+        await p.waitForLoadState('domcontentloaded')
+        return p
+      }
+    }
+
+    // If not found, wait a bit before retrying
+    if (attempt < maxAttempts - 1) {
+      await new Promise(resolve => setTimeout(resolve, retryDelay))
     }
   }
+
   throw new Error(`Extension popup not found for ID: ${extensionId}`)
 }
 
@@ -31,9 +44,7 @@ export async function getPolkadotJSExtensionPath(): Promise<string> {
     throw new Error(
       `Polkadot-JS extension not found at: ${extensionDir}\n\n`
       + `Please download the extension first by running:\n`
-      + `  npx @avalix/chroma download-extensions\n\n`
-      + `Or if you're using this as a dependency:\n`
-      + `  npm run chroma:download\n`,
+      + `  npx @avalix/chroma download-extensions\n`,
     )
   }
 
@@ -86,7 +97,6 @@ export async function authorizePolkadotJS(
 ): Promise<void> {
   const context = page.__extensionContext
   const extensionId = page.__extensionId
-  await new Promise(resolve => setTimeout(resolve, 1000))
 
   const extensionPopup = await findExtensionPopup(context, extensionId)
   await extensionPopup.getByText('Select all').click()
@@ -104,11 +114,22 @@ export async function approvePolkadotJSTx(
   const context = page.__extensionContext
   const extensionId = page.__extensionId
 
-  await new Promise(resolve => setTimeout(resolve, 1000))
   const extensionPopup = await findExtensionPopup(context, extensionId)
-
   await extensionPopup.getByRole('textbox').fill(password)
   await extensionPopup.getByRole('button', { name: 'Sign the transaction' }).click()
 
   console.log('✅ Polkadot-JS transaction signed successfully')
 }
+
+// Polkadot-JS specific transaction rejection implementation
+export async function rejectPolkadotJSTx(
+  page: Page & { __extensionContext: BrowserContext, __extensionId: string },
+): Promise<void> {
+  const context = page.__extensionContext
+  const extensionId = page.__extensionId
+
+  const extensionPopup = await findExtensionPopup(context, extensionId)
+  await extensionPopup.getByRole('link', { name: 'Cancel' }).click()
+
+  console.log('✅ Polkadot-JS transaction rejected successfully')
+}

package/src/wallets/talisman.ts

@@ -12,15 +12,23 @@ export const TALISMAN_CONFIG = {
 
 // Helper function to find extension popup
 async function findExtensionPopup(context: BrowserContext, extensionId: string): Promise<Page> {
-  // delay for 1 second
-  await new Promise(resolve => setTimeout(resolve, 1000))
-
-  const pages = context.pages()
-  for (const p of pages) {
-    if (p.url().includes(`chrome-extension://${extensionId}/`)) {
-      p.setViewportSize({ width: 400, height: 600 })
-      p.waitForLoadState('domcontentloaded')
-      return p
+  // Wait for extension popup to appear with retry logic
+  const maxAttempts = 10
+  const retryDelay = 500
+
+  for (let attempt = 0; attempt < maxAttempts; attempt++) {
+    const pages = context.pages()
+    for (const p of pages) {
+      if (p.url().includes(`chrome-extension://${extensionId}/`)) {
+        await p.setViewportSize({ width: 400, height: 600 })
+        await p.waitForLoadState('domcontentloaded')
+        return p
+      }
+    }
+
+    // If not found, wait a bit before retrying
+    if (attempt < maxAttempts - 1) {
+      await new Promise(resolve => setTimeout(resolve, retryDelay))
     }
   }
 
@@ -37,9 +45,7 @@ export async function getTalismanExtensionPath(): Promise<string> {
     throw new Error(
       `Talisman extension not found at: ${extensionDir}\n\n`
       + `Please download the extension first by running:\n`
-      + `  npx @avalix/chroma download-extensions\n\n`
-      + `Or if you're using this as a dependency:\n`
-      + `  npm run chroma:download\n`,
+      + `  npx @avalix/chroma download-extensions\n`,
     )
   }
 
@@ -55,38 +61,45 @@ export async function importEthPrivateKey(
   const context = page.__extensionContext
   const extensionId = page.__extensionId
 
-  // Wait for Talisman to open its onboarding tab
-  await new Promise(resolve => setTimeout(resolve, 2000))
-
-  // Find the onboarding tab
+  // Wait for Talisman to open its onboarding tab with retry logic
+  const maxAttempts = 20
+  const retryDelay = 500
   let extensionPage: Page | null = null
-  const pages = context.pages()
-
-  for (const page of pages) {
-    const url = page.url()
-    console.log(`📄 Found page: ${url}`)
-    if (url.includes('onboarding.html') || url.includes(`chrome-extension://${extensionId}/`)) {
-      extensionPage = page
-      console.log(`✅ Found Talisman onboarding page: ${url}`)
+
+  for (let attempt = 0; attempt < maxAttempts; attempt++) {
+    const pages = context.pages()
+
+    for (const p of pages) {
+      const url = p.url()
+      console.log(`📄 Found page: ${url}`)
+      if (url.includes('onboarding.html') || url.includes(`chrome-extension://${extensionId}/`)) {
+        extensionPage = p
+        console.log(`✅ Found Talisman onboarding page: ${url}`)
+        break
+      }
+    }
+
+    if (extensionPage) {
       break
     }
+
+    // If not found, wait before retrying
+    if (attempt < maxAttempts - 1) {
+      console.log(`⏳ Attempt ${attempt + 1}/${maxAttempts}: Waiting for onboarding page...`)
+      await new Promise(resolve => setTimeout(resolve, retryDelay))
+    }
   }
 
   if (!extensionPage) {
-    throw new Error(`Talisman onboarding page not found`)
+    throw new Error(`Talisman onboarding page not found after ${maxAttempts} attempts`)
   }
 
   try {
     // Bring the onboarding page to front
     await extensionPage.bringToFront()
 
-    // Reload the onboarding page to ensure fresh state
-    console.log('🔄 Reloading onboarding page for fresh state...')
-    await extensionPage.reload()
-
     // Wait for the page to load and become interactive
     await extensionPage.waitForLoadState('domcontentloaded')
-    // await extensionPage.waitForTimeout(20000) // Give Talisman more time to initialize
 
     // Click the get started button
     await extensionPage.getByTestId('onboarding-get-started-button').click()
@@ -128,7 +141,6 @@ export async function authorizeTalisman(
   const { accountName = 'Test Account' } = options
   const context = page.__extensionContext
   const extensionId = page.__extensionId
-  await new Promise(resolve => setTimeout(resolve, 1000))
 
   const extensionPopup = await findExtensionPopup(context, extensionId)
 
@@ -152,8 +164,17 @@ export async function approveTalismanTx(
   const context = page.__extensionContext
   const extensionId = page.__extensionId
 
-  await new Promise(resolve => setTimeout(resolve, 1000))
   const extensionPopup = await findExtensionPopup(context, extensionId)
-
   await extensionPopup.getByRole('button', { name: 'Approve' }).click()
 }
+
+// Talisman specific transaction rejection implementation
+export async function rejectTalismanTx(
+  page: Page & { __extensionContext: BrowserContext, __extensionId: string },
+): Promise<void> {
+  const context = page.__extensionContext
+  const extensionId = page.__extensionId
+
+  const extensionPopup = await findExtensionPopup(context, extensionId)
+  await extensionPopup.getByTestId('connection-reject-button').click()
+}