@avalix/chroma 0.0.16 → 1.0.0

package/dist/index.mjs

@@ -1,18 +1,29 @@
+import { cp, rm } from "node:fs/promises";
+import path, { resolve } from "node:path";
 import { chromium, expect, test as test$1 } from "@playwright/test";
 import fs from "node:fs";
-import path from "node:path";
 import process from "node:process";
 
+//#region src/utils/test-defaults.ts
+/**
+* Default password used by chroma to bootstrap wallets in tests.
+*
+* Wallet methods accept an explicit `password` option that overrides this
+* value, so callers that already supply their own password are unaffected.
+*/
+const DEFAULT_TEST_PASSWORD = "h3llop0lkadot!";
+
+//#endregion
 //#region src/wallets/metamask.ts
-const VERSION$2 = "13.17.0";
+const VERSION$2 = "13.28.0";
 const METAMASK_CONFIG = {
-	downloadUrl: `https://github.com/MetaMask/metamask-extension/releases/download/v${VERSION$2}/metamask-flask-chrome-${VERSION$2}-flask.0.zip`,
+	downloadUrl: `https://github.com/MetaMask/metamask-extension/releases/download/v${VERSION$2}/metamask-chrome-${VERSION$2}.zip`,
 	extensionName: `metamask-extension-${VERSION$2}`
 };
 async function getMetaMaskExtensionPath() {
 	const extensionsDir = path.resolve(process.cwd(), ".chroma");
 	const extensionDir = path.join(extensionsDir, METAMASK_CONFIG.extensionName);
-	if (!fs.existsSync(extensionDir) || fs.readdirSync(extensionDir).length === 0) throw new Error(`MetaMask extension not found at: ${extensionDir}\n\nPlease download the extension first by running:\n  npx @avalix/chroma download-extensions\n`);
+	if ((await fs.promises.readdir(extensionDir).catch(() => [])).length === 0) throw new Error(`MetaMask extension not found at: ${extensionDir}\n\nPlease download the extension first by running:\n  npx @avalix/chroma download-extensions\n`);
 	return extensionDir;
 }
 /* c8 ignore start */
@@ -23,7 +34,6 @@ async function findOnboardingPage$1(context, extensionId) {
 		const pages = context.pages();
 		for (const p of pages) if (p.url().includes(`chrome-extension://${extensionId}/`)) {
 			await p.waitForLoadState("domcontentloaded");
-			await p.getByText("I accept the risks").click();
 			return p;
 		}
 		if (attempt < maxAttempts - 1) await new Promise((resolve) => setTimeout(resolve, retryDelay));
@@ -50,7 +60,6 @@ async function findExtensionPopup$2(context, extensionId) {
 	}
 	throw new Error(`MetaMask side panel not found for ID: ${extensionId}`);
 }
-const METAMASK_PASSWORD = "h3llop0lkadot!";
 async function completeOnboarding$1(extensionPage, seedPhrase) {
 	await extensionPage.bringToFront();
 	await extensionPage.waitForLoadState("domcontentloaded");
@@ -58,18 +67,13 @@ async function completeOnboarding$1(extensionPage, seedPhrase) {
 	await extensionPage.getByTestId("onboarding-import-with-srp-button").click();
 	await extensionPage.getByTestId("srp-input-import__srp-note").pressSequentially(seedPhrase, { delay: 50 });
 	await extensionPage.getByTestId("import-srp-confirm").click();
-	await extensionPage.getByTestId("create-password-new-input").fill(METAMASK_PASSWORD);
-	await extensionPage.getByTestId("create-password-confirm-input").fill(METAMASK_PASSWORD);
+	await extensionPage.getByTestId("create-password-new-input").fill(DEFAULT_TEST_PASSWORD);
+	await extensionPage.getByTestId("create-password-confirm-input").fill(DEFAULT_TEST_PASSWORD);
 	await extensionPage.getByTestId("create-password-terms").click();
 	await extensionPage.getByTestId("create-password-submit").click();
 	await extensionPage.getByTestId("metametrics-checkbox").click();
 	await extensionPage.getByTestId("metametrics-i-agree").click();
 	await extensionPage.getByTestId("manage-default-settings").click();
-	await extensionPage.getByTestId("category-item-General").click();
-	await extensionPage.getByTestId("basic-functionality-toggle").locator(".toggle-button").click();
-	await extensionPage.getByText("I understand and want to continue").click();
-	await extensionPage.getByTestId("basic-configuration-modal-toggle-button").click();
-	await extensionPage.getByTestId("category-back-button").click();
 	await extensionPage.getByTestId("category-item-Assets").click();
 	await extensionPage.getByTestId("privacy-settings-settings").locator(".toggle-button").nth(0).click();
 	await extensionPage.getByTestId("privacy-settings-settings").locator(".toggle-button").nth(1).click();
@@ -81,40 +85,44 @@ async function completeOnboarding$1(extensionPage, seedPhrase) {
 	await extensionPage.getByTestId("onboarding-complete-done").click();
 	await extensionPage.close();
 }
-async function authorizeMetaMask(page) {
-	const context = page.__extensionContext;
-	const extensionId = page.__extensionId;
+async function approveMetaMask(context, extensionId) {
 	const extensionPopup = await findExtensionPopup$2(context, extensionId);
-	await extensionPopup.getByTestId("confirm-btn").click();
+	await extensionPopup.getByTestId("confirm-btn").or(extensionPopup.getByTestId("confirm-footer-button")).or(extensionPopup.getByTestId("confirm-sign-message-confirm-snap-footer-button")).first().click();
 	await extensionPopup.close();
 }
-async function confirmMetaMask(page) {
-	const context = page.__extensionContext;
-	const extensionId = page.__extensionId;
-	const extensionPopup = await findExtensionPopup$2(context, extensionId);
-	await extensionPopup.getByTestId("confirm-footer-button").click();
-	await extensionPopup.close();
-}
-async function rejectMetaMask(page) {
-	const context = page.__extensionContext;
-	const extensionId = page.__extensionId;
+async function rejectMetaMask(context, extensionId) {
 	const extensionPopup = await findExtensionPopup$2(context, extensionId);
 	await extensionPopup.getByTestId("confirm-footer-cancel").or(extensionPopup.getByTestId("page-container-footer-cancel")).or(extensionPopup.getByRole("button", { name: /Reject|Cancel/i })).first().click();
 	await extensionPopup.close();
 }
-async function unlockMetaMask(page) {
-	const context = page.__extensionContext;
-	const unlockUrl = `chrome-extension://${page.__extensionId}/home.html#/onboarding/unlock`;
-	const unlockPage = await context.newPage();
-	await unlockPage.goto(unlockUrl);
-	await unlockPage.waitForLoadState("domcontentloaded");
-	await unlockPage.getByTestId("unlock-password").fill(METAMASK_PASSWORD);
-	await unlockPage.getByTestId("unlock-submit").click();
-	await unlockPage.close();
+async function unlockMetaMask(context, extensionId) {
+	const sidePanelUrl = `chrome-extension://${extensionId}/sidepanel.html`;
+	const extensionUrlPrefix = `chrome-extension://${extensionId}/`;
+	let unlockPage;
+	const deadline = Date.now() + 1e4;
+	while (Date.now() < deadline) {
+		const extensionPages = context.pages().filter((p) => p.url().startsWith(extensionUrlPrefix));
+		unlockPage = extensionPages.find((p) => p.url().includes("unlock"));
+		if (unlockPage) break;
+		if (extensionPages.length > 0) break;
+		await new Promise((resolve) => setTimeout(resolve, 100));
+	}
+	if (unlockPage) {
+		await unlockPage.bringToFront();
+		await unlockPage.waitForLoadState("domcontentloaded");
+		await unlockPage.getByTestId("unlock-password").fill(DEFAULT_TEST_PASSWORD);
+		await unlockPage.getByTestId("unlock-submit").click();
+		await unlockPage.getByTestId("onboarding-complete-done").click({ timeout: 3e3 }).catch(() => {});
+		await unlockPage.goto(sidePanelUrl);
+		await unlockPage.waitForLoadState("domcontentloaded");
+		return;
+	}
+	if (context.pages().find((p) => p.url().startsWith(sidePanelUrl))) return;
+	const sidePanel = await context.newPage();
+	await sidePanel.goto(sidePanelUrl);
+	await sidePanel.waitForLoadState("domcontentloaded");
 }
-async function importSeedPhrase(page, { seedPhrase }) {
-	const context = page.__extensionContext;
-	const extensionId = page.__extensionId;
+async function importSeedPhrase(context, extensionId, { seedPhrase }) {
 	const extensionPage = await findOnboardingPage$1(context, extensionId);
 	try {
 		await completeOnboarding$1(extensionPage, seedPhrase);
@@ -126,19 +134,17 @@ async function importSeedPhrase(page, { seedPhrase }) {
 /* c8 ignore stop */
 
 //#endregion
-//#region src/wallets/polkadot-js.ts
-const VERSION$1 = "0.62.6";
-const POLKADOT_JS_CONFIG = {
-	downloadUrl: `https://github.com/polkadot-js/extension/releases/download/v${VERSION$1}/master-chrome-build.zip`,
-	extensionName: `polkadot-extension-${VERSION$1}`
-};
+//#region src/utils/find-extension-popup.ts
+/**
+* Poll the BrowserContext for a page whose URL points at the given Chrome
+* extension. Returns once the page is reachable and DOM-loaded.
+*/
 /* c8 ignore start */
-async function findExtensionPopup$1(context, extensionId) {
-	const maxAttempts = 10;
-	const retryDelay = 500;
+async function findExtensionPopup$1(context, extensionId, options = {}) {
+	const { maxAttempts = 10, retryDelay = 500, viewport } = options;
 	for (let attempt = 0; attempt < maxAttempts; attempt++) {
-		const pages = context.pages();
-		for (const p of pages) if (p.url().includes(`chrome-extension://${extensionId}/`)) {
+		for (const p of context.pages()) if (p.url().includes(`chrome-extension://${extensionId}/`)) {
+			if (viewport) await p.setViewportSize(viewport);
 			await p.waitForLoadState("domcontentloaded");
 			return p;
 		}
@@ -147,25 +153,29 @@ async function findExtensionPopup$1(context, extensionId) {
 	throw new Error(`Extension popup not found for ID: ${extensionId}`);
 }
 /* c8 ignore stop */
+
+//#endregion
+//#region src/wallets/polkadot-js.ts
+const VERSION$1 = "0.62.6";
+const POLKADOT_JS_CONFIG = {
+	downloadUrl: `https://github.com/polkadot-js/extension/releases/download/v${VERSION$1}/master-chrome-build.zip`,
+	extensionName: `polkadot-extension-${VERSION$1}`
+};
 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`);
+	if ((await fs.promises.readdir(extensionDir).catch(() => [])).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`);
 	return extensionDir;
 }
 /* c8 ignore start */
-async function importPolkadotJSAccount(page, { seed, name = "Test Account", password = "h3llop0lkadot!" }) {
-	const context = page.__extensionContext;
-	const extensionPopupUrl = `chrome-extension://${page.__extensionId}/index.html`;
+async function importPolkadotJSAccount(context, extensionId, { seed, name = "Test Account", password = DEFAULT_TEST_PASSWORD }) {
+	const extensionPopupUrl = `chrome-extension://${extensionId}/index.html`;
 	const extensionPage = await context.newPage();
 	try {
 		await extensionPage.goto(extensionPopupUrl);
 		const understoodButton = extensionPage.getByRole("button", { name: "Understood, let me continue" });
-		if (await understoodButton.count() > 0) {
-			await understoodButton.click();
-			await extensionPage.waitForTimeout(100);
-		}
-		if (await extensionPage.getByRole("button", { name: "I Understand" }).isVisible()) await extensionPage.getByRole("button", { name: "I Understand" }).click();
+		if (await understoodButton.count() > 0) await understoodButton.click();
+		await extensionPage.getByRole("button", { name: "I Understand" }).click({ timeout: 1e3 }).catch(() => {});
 		await extensionPage.goto(`${extensionPopupUrl}#/account/import-seed`);
 		await extensionPage.locator("textarea").fill(seed);
 		await extensionPage.locator("button:has-text(\"Next\")").click();
@@ -177,25 +187,19 @@ async function importPolkadotJSAccount(page, { seed, name = "Test Account", pass
 		await extensionPage.close();
 	}
 }
-async function authorizePolkadotJS(page) {
-	const context = page.__extensionContext;
-	const extensionId = page.__extensionId;
+async function authorizePolkadotJS(context, extensionId) {
 	const extensionPopup = await findExtensionPopup$1(context, extensionId);
 	if (await extensionPopup.getByRole("button", { name: "I Understand" }).isVisible()) await extensionPopup.getByRole("button", { name: "I Understand" }).click();
 	if (!await extensionPopup.getByText("Select all").locator("..").locator("input[type=\"checkbox\"]").isChecked().catch(() => false)) await extensionPopup.getByText("Select all").click();
 	await extensionPopup.getByRole("button", { name: /Connect \d+ account\(s\)/ }).click();
 }
-async function approvePolkadotJSTx(page, options = {}) {
-	const { password = "h3llop0lkadot!" } = options;
-	const context = page.__extensionContext;
-	const extensionId = page.__extensionId;
+async function approvePolkadotJSTx(context, extensionId, options = {}) {
+	const { password = DEFAULT_TEST_PASSWORD } = options;
 	const extensionPopup = await findExtensionPopup$1(context, extensionId);
 	await extensionPopup.getByRole("textbox").fill(password);
 	await extensionPopup.getByRole("button", { name: "Sign the transaction" }).click();
 }
-async function rejectPolkadotJSTx(page) {
-	const context = page.__extensionContext;
-	const extensionId = page.__extensionId;
+async function rejectPolkadotJSTx(context, extensionId) {
 	await (await findExtensionPopup$1(context, extensionId)).getByRole("link", { name: "Cancel" }).click();
 }
 /* c8 ignore stop */
@@ -208,28 +212,17 @@ const TALISMAN_CONFIG = {
 	extensionName: `talisman-extension-${VERSION}`
 };
 /* c8 ignore start */
-async function findExtensionPopup(context, extensionId) {
-	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}`);
+function findExtensionPopup(context, extensionId) {
+	return findExtensionPopup$1(context, extensionId, { viewport: {
+		width: 400,
+		height: 600
+	} });
 }
 /* c8 ignore stop */
 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`);
+	if ((await fs.promises.readdir(extensionDir).catch(() => [])).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`);
 	return extensionDir;
 }
 /* c8 ignore start */
@@ -260,9 +253,7 @@ async function completeOnboarding(extensionPage, password) {
 	await extensionPage.getByRole("link", { name: "Security & Privacy" }).click();
 	await extensionPage.getByTestId("component-toggle-button").first().click();
 }
-async function importPolkadotMnemonic(page, { seed, name = "Test Account", password = "h3llop0lkadot!" }) {
-	const context = page.__extensionContext;
-	const extensionId = page.__extensionId;
+async function importPolkadotMnemonic(context, extensionId, { seed, name = "Test Account", password = DEFAULT_TEST_PASSWORD }) {
 	const extensionPage = await findOnboardingPage(context, extensionId);
 	try {
 		await completeOnboarding(extensionPage, password);
@@ -281,9 +272,7 @@ async function importPolkadotMnemonic(page, { seed, name = "Test Account", passw
 		throw error;
 	}
 }
-async function importEthPrivateKey(page, { seed, name = "Test Account", password = "h3llop0lkadot!" }) {
-	const context = page.__extensionContext;
-	const extensionId = page.__extensionId;
+async function importEthPrivateKey(context, extensionId, { seed, name = "Test Account", password = DEFAULT_TEST_PASSWORD }) {
 	const extensionPage = await findOnboardingPage(context, extensionId);
 	try {
 		await completeOnboarding(extensionPage, password);
@@ -303,9 +292,7 @@ async function importEthPrivateKey(page, { seed, name = "Test Account", password
 		throw error;
 	}
 }
-async function authorizeTalisman(page, options = {}) {
-	const context = page.__extensionContext;
-	const extensionId = page.__extensionId;
+async function authorizeTalisman(context, extensionId, options = {}) {
 	const { accountName = "Test Account" } = options;
 	const extensionPopup = await findExtensionPopup(context, extensionId);
 	await extensionPopup.waitForLoadState("domcontentloaded");
@@ -318,16 +305,12 @@ async function authorizeTalisman(page, options = {}) {
 		await (await findExtensionPopup(context, extensionId)).getByRole("button", { name: "Approve" }).click();
 	} catch {}
 }
-async function approveTalismanTx(page) {
-	const context = page.__extensionContext;
-	const extensionId = page.__extensionId;
+async function approveTalismanTx(context, extensionId) {
 	const extensionPopup = await findExtensionPopup(context, extensionId);
 	if (await extensionPopup.getByRole("button", { name: "Yes" }).isVisible()) await extensionPopup.getByRole("button", { name: "Yes" }).click();
 	await extensionPopup.getByRole("button", { name: "Approve" }).click();
 }
-async function rejectTalismanTx(page) {
-	const context = page.__extensionContext;
-	const extensionId = page.__extensionId;
+async function rejectTalismanTx(context, extensionId) {
 	const extensionPopup = await findExtensionPopup(context, extensionId);
 	await extensionPopup.getByTestId("connection-reject-button").or(extensionPopup.getByRole("button", { name: "Cancel" })).or(extensionPopup.getByRole("button", { name: "Reject" })).click();
 }
@@ -336,30 +319,14 @@ async function rejectTalismanTx(page) {
 //#endregion
 //#region src/context-playwright/wallet-factory.ts
 /* c8 ignore start */
-function createExtendedPage(page, context, extensionId) {
-	const extPage = page;
-	extPage.__extensionContext = context;
-	extPage.__extensionId = extensionId;
-	return extPage;
-}
-/* c8 ignore stop */
-/* c8 ignore start */
 function createPolkadotJsWallet(extensionId, context) {
 	return {
 		extensionId,
 		type: "polkadot-js",
-		importMnemonic: async (options) => {
-			await importPolkadotJSAccount(createExtendedPage(context.pages()[0] || await context.newPage(), context, extensionId), options);
-		},
-		authorize: async () => {
-			await authorizePolkadotJS(createExtendedPage(context.pages()[0] || await context.newPage(), context, extensionId));
-		},
-		approveTx: async (options = {}) => {
-			await approvePolkadotJSTx(createExtendedPage(context.pages()[0] || await context.newPage(), context, extensionId), options);
-		},
-		rejectTx: async () => {
-			await rejectPolkadotJSTx(createExtendedPage(context.pages()[0] || await context.newPage(), context, extensionId));
-		}
+		importMnemonic: (options) => importPolkadotJSAccount(context, extensionId, options),
+		authorize: () => authorizePolkadotJS(context, extensionId),
+		approveTx: (options = {}) => approvePolkadotJSTx(context, extensionId, options),
+		rejectTx: () => rejectPolkadotJSTx(context, extensionId)
 	};
 }
 /* c8 ignore stop */
@@ -369,29 +336,23 @@ function createTalismanWallet(extensionId, context) {
 	return {
 		extensionId,
 		type: "talisman",
-		importPolkadotMnemonic: async (options) => {
-			const extPage = createExtendedPage(context.pages()[0] || await context.newPage(), context, extensionId);
+		importPolkadotMnemonic: (options) => {
 			importedAccountName = options.name || "Test Account";
-			await importPolkadotMnemonic(extPage, options);
+			return importPolkadotMnemonic(context, extensionId, options);
 		},
-		importEthPrivateKey: async (options) => {
-			const extPage = createExtendedPage(context.pages()[0] || await context.newPage(), context, extensionId);
+		importEthPrivateKey: (options) => {
 			importedAccountName = options.name || "Test Account";
-			await importEthPrivateKey(extPage, {
+			return importEthPrivateKey(context, extensionId, {
 				seed: options.privateKey,
 				name: options.name,
 				password: options.password
 			});
 		},
-		authorize: async (options = {}) => {
-			await authorizeTalisman(createExtendedPage(context.pages()[0] || await context.newPage(), context, extensionId), { accountName: options.accountName || importedAccountName });
-		},
-		approveTx: async () => {
-			await approveTalismanTx(createExtendedPage(context.pages()[0] || await context.newPage(), context, extensionId));
+		authorize: (options = {}) => {
+			return authorizeTalisman(context, extensionId, { accountName: options.accountName || importedAccountName });
 		},
-		rejectTx: async () => {
-			await rejectTalismanTx(createExtendedPage(context.pages()[0] || await context.newPage(), context, extensionId));
-		}
+		approveTx: () => approveTalismanTx(context, extensionId),
+		rejectTx: () => rejectTalismanTx(context, extensionId)
 	};
 }
 /* c8 ignore stop */
@@ -400,21 +361,10 @@ function createMetaMaskWallet(extensionId, context) {
 	return {
 		extensionId,
 		type: "metamask",
-		importSeedPhrase: async (options) => {
-			await importSeedPhrase(createExtendedPage(context.pages()[0] || await context.newPage(), context, extensionId), { seedPhrase: options.seedPhrase });
-		},
-		unlock: async () => {
-			await unlockMetaMask(createExtendedPage(context.pages()[0] || await context.newPage(), context, extensionId));
-		},
-		authorize: async () => {
-			await authorizeMetaMask(createExtendedPage(context.pages()[0] || await context.newPage(), context, extensionId));
-		},
-		reject: async () => {
-			await rejectMetaMask(createExtendedPage(context.pages()[0] || await context.newPage(), context, extensionId));
-		},
-		confirm: async () => {
-			await confirmMetaMask(createExtendedPage(context.pages()[0] || await context.newPage(), context, extensionId));
-		}
+		importSeedPhrase: (options) => importSeedPhrase(context, extensionId, options),
+		unlock: () => unlockMetaMask(context, extensionId),
+		approve: () => approveMetaMask(context, extensionId),
+		reject: () => rejectMetaMask(context, extensionId)
 	};
 }
 /* c8 ignore stop */
@@ -438,12 +388,22 @@ async function getExtensionPathForWallet(config) {
 function createWalletTest(options = {}) {
 	const { headless = false, slowMo = 150 } = options;
 	const walletConfigs = options.wallets && options.wallets.length > 0 ? options.wallets : [{ type: "polkadot-js" }];
-	const isMultiWallet = walletConfigs.length > 1;
 	/* c8 ignore start */
 	return test$1.extend({
-		walletContext: [async ({}, use) => {
+		walletContext: [async ({}, use, workerInfo) => {
 			const extensionPathsString = (await Promise.all(walletConfigs.map((config) => getExtensionPathForWallet(config)))).join(",");
-			const context = await chromium.launchPersistentContext("", {
+			const userDataDirOption = options.userDataDir;
+			const userDataDir = typeof userDataDirOption === "function" ? await userDataDirOption({ workerIndex: workerInfo.workerIndex }) : userDataDirOption ?? "";
+			if (options.cloneUserDataDirFrom && userDataDir) {
+				const sourceAbs = resolve(options.cloneUserDataDirFrom);
+				if (sourceAbs === resolve(userDataDir)) throw new Error(`cloneUserDataDirFrom and userDataDir must be different paths; both resolved to "${sourceAbs}"`);
+				await rm(userDataDir, {
+					recursive: true,
+					force: true
+				});
+				await cp(options.cloneUserDataDirFrom, userDataDir, { recursive: true });
+			}
+			const context = await chromium.launchPersistentContext(userDataDir, {
 				headless,
 				channel: "chromium",
 				args: [`--load-extension=${extensionPathsString}`, `--disable-extensions-except=${extensionPathsString}`],
@@ -454,21 +414,21 @@ function createWalletTest(options = {}) {
 		}, { scope: "worker" }],
 		walletExtensionIds: [async ({ walletContext }, use) => {
 			const extensionIds = /* @__PURE__ */ new Map();
-			if (walletContext.serviceWorkers().length === 0) await walletContext.waitForEvent("serviceworker");
-			if (isMultiWallet) await new Promise((resolve) => setTimeout(resolve, 1e3));
+			const expected = walletConfigs.length;
+			const deadline = Date.now() + 1e4;
+			while (walletContext.serviceWorkers().length < expected) {
+				if (Date.now() > deadline) break;
+				await Promise.race([walletContext.waitForEvent("serviceworker", { timeout: 2e3 }).catch(() => {}), new Promise((resolve) => setTimeout(resolve, 200))]);
+			}
 			const allServiceWorkers = walletContext.serviceWorkers();
 			for (let i = 0; i < walletConfigs.length && i < allServiceWorkers.length; i++) {
 				const extensionId = allServiceWorkers[i].url().split("/")[2];
-				const walletType = walletConfigs[i].type;
-				extensionIds.set(walletType, extensionId);
+				extensionIds.set(walletConfigs[i].type, extensionId);
 			}
 			await use(extensionIds);
 		}, { scope: "worker" }],
-		page: async ({ walletContext, walletExtensionIds }, use) => {
-			const extendedPage = walletContext.pages()[0] || await walletContext.newPage();
-			extendedPage.__extensionContext = walletContext;
-			extendedPage.__walletExtensionIds = walletExtensionIds;
-			await use(extendedPage);
+		page: async ({ walletContext }, use) => {
+			await use(walletContext.pages()[0] || await walletContext.newPage());
 		},
 		wallets: async ({ walletContext, walletExtensionIds }, use) => {
 			const walletMap = {};

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@avalix/chroma",
   "type": "module",
-  "version": "0.0.16",
+  "version": "1.0.0",
   "description": "End-to-end testing library for Polkadot wallet interactions",
   "author": "Avalix Labs",
   "license": "MIT",
@@ -51,16 +51,15 @@
     "test:unit:coverage": "vitest run --coverage"
   },
   "peerDependencies": {
-    "@playwright/test": "^1.57.0"
+    "@playwright/test": "^1.59.1"
   },
   "dependencies": {
     "adm-zip": "^0.5.16"
   },
   "devDependencies": {
     "@antfu/eslint-config": "^6.5.1",
-    "@playwright/test": "^1.57.0",
+    "@playwright/test": "^1.59.1",
     "@types/node": "^24.10.2",
-    "@types/unzipper": "^0.10.11",
     "@vitest/coverage-v8": "4.0.18",
     "eslint": "^9.39.1",
     "tsdown": "^0.20.1",

package/scripts/download-extensions.ts

@@ -18,11 +18,8 @@ async function getVersion(): Promise<string> {
 
 async function clearChromaDir(): Promise<void> {
   const chromaDir = path.resolve(process.cwd(), '.chroma')
-
-  if (fs.existsSync(chromaDir)) {
-    console.log('🗑️ Clearing existing .chroma directory...')
-    await fs.promises.rm(chromaDir, { recursive: true, force: true })
-  }
+  console.log('🗑️ Clearing existing .chroma directory...')
+  await fs.promises.rm(chromaDir, { recursive: true, force: true })
 }
 
 async function main() {
@@ -34,23 +31,21 @@ async function main() {
     // Clear existing .chroma directory
     await clearChromaDir()
 
-    // Download Polkadot-JS extension
-    await downloadAndExtractExtension({
-      downloadUrl: POLKADOT_JS_CONFIG.downloadUrl,
-      extensionName: POLKADOT_JS_CONFIG.extensionName,
-    })
-
-    // Download Talisman extension
-    await downloadAndExtractExtension({
-      downloadUrl: TALISMAN_CONFIG.downloadUrl,
-      extensionName: TALISMAN_CONFIG.extensionName,
-    })
-
-    // Download MetaMask extension
-    await downloadAndExtractExtension({
-      downloadUrl: METAMASK_CONFIG.downloadUrl,
-      extensionName: METAMASK_CONFIG.extensionName,
-    })
+    // Download all extensions in parallel — they're independent network ops
+    await Promise.all([
+      downloadAndExtractExtension({
+        downloadUrl: POLKADOT_JS_CONFIG.downloadUrl,
+        extensionName: POLKADOT_JS_CONFIG.extensionName,
+      }),
+      downloadAndExtractExtension({
+        downloadUrl: TALISMAN_CONFIG.downloadUrl,
+        extensionName: TALISMAN_CONFIG.extensionName,
+      }),
+      downloadAndExtractExtension({
+        downloadUrl: METAMASK_CONFIG.downloadUrl,
+        extensionName: METAMASK_CONFIG.extensionName,
+      }),
+    ])
 
     console.log('\n✅ All extensions downloaded successfully!')
     console.log('You can now run your Playwright tests.')

package/src/context-playwright/index.test.ts

@@ -1,8 +1,5 @@
 import { beforeEach, describe, expect, it, vi } from 'vitest'
-import { getMetaMaskExtensionPath } from '../wallets/metamask.js'
-import { getPolkadotJSExtensionPath } from '../wallets/polkadot-js.js'
-import { getTalismanExtensionPath } from '../wallets/talisman.js'
-import { createWalletTest, getExtensionPathForWallet } from './index.js'
+import { createWalletTest } from './index.js'
 
 // Mock @playwright/test
 vi.mock('@playwright/test', () => {
@@ -118,44 +115,37 @@ describe('context-playwright/index', () => {
       expect(result).toBeDefined()
     })
 
-    it('should handle wallets with downloadUrl option', () => {
+    it('should accept userDataDir as a string', () => {
       const result = createWalletTest({
-        wallets: [{
-          type: 'polkadot-js',
-          downloadUrl: 'https://custom-url.com/extension.zip',
-        }],
+        userDataDir: '.cache/wallet-setup',
       })
 
       expect(result).toBeDefined()
     })
-  })
 
-  describe('getExtensionPathForWallet', () => {
-    it('should return polkadot-js extension path', async () => {
-      const result = await getExtensionPathForWallet({ type: 'polkadot-js' })
+    it('should accept userDataDir as a per-worker function', () => {
+      const result = createWalletTest({
+        userDataDir: ({ workerIndex }) => `.cache/wallet-w${workerIndex}`,
+      })
 
-      expect(result).toBe('/mock/path/polkadot-extension')
-      expect(getPolkadotJSExtensionPath).toHaveBeenCalled()
+      expect(result).toBeDefined()
     })
 
-    it('should return talisman extension path', async () => {
-      const result = await getExtensionPathForWallet({ type: 'talisman' })
+    it('should accept userDataDir as an async function', () => {
+      const result = createWalletTest({
+        userDataDir: async ({ workerIndex }) => `.cache/wallet-w${workerIndex}`,
+      })
 
-      expect(result).toBe('/mock/path/talisman-extension')
-      expect(getTalismanExtensionPath).toHaveBeenCalled()
+      expect(result).toBeDefined()
     })
 
-    it('should return metamask extension path', async () => {
-      const result = await getExtensionPathForWallet({ type: 'metamask' })
-
-      expect(result).toBe('/mock/path/metamask-extension')
-      expect(getMetaMaskExtensionPath).toHaveBeenCalled()
-    })
+    it('should accept cloneUserDataDirFrom alongside userDataDir', () => {
+      const result = createWalletTest({
+        userDataDir: ({ workerIndex }) => `.cache/wallet-w${workerIndex}`,
+        cloneUserDataDirFrom: '.cache/wallet-setup',
+      })
 
-    it('should throw error for unsupported wallet type', async () => {
-      await expect(
-        getExtensionPathForWallet({ type: 'unsupported' as any }),
-      ).rejects.toThrow('Unsupported wallet type: unsupported')
+      expect(result).toBeDefined()
     })
   })
 })