Fix failing case

Holo-Host · Jun 30, 2024 · a16b612 · a16b612
1 parent c42e24a
commit a16b612
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/helpers.ts b/tests/helpers.ts
@@ -27,9 +27,8 @@ export const clickButtonAndWaitForNewPage = async (
 	button: Locator<Element>,
 	downloadPath: string
 ): Promise<Page> => {
-	await button?.click();
+	await button.click();
 	const newPage = await waitForNewPage(browser);
-	await newPage.waitForNavigation();
 
 	const session = await newPage.createCDPSession();
 	await session.send('Page.setDownloadBehavior', {
@@ -57,7 +56,7 @@ export const fileExists = (filePath: string): Promise<boolean> => {
 
 const findElementByText = (context: Page, selector: string) => async (text: string) => {
 	const locator = context.locator(`${selector}::-p-text(${text})`);
-	await locator.wait();
+	await locator.waitHandle();
 	return locator;
 };