Augment main E2E test to verify prompt creation and open in ZDS (#194)

* Augment main E2E test to verify prompt creation and open in ZDS

Towards #184. Extend the current primary E2E test to not just load the
dashboard after logging in, but also to create a prompt, redirect to its
page, wait for it to successfully generate, and open it in a Zoo Design
Studio temporary workspace.

* Fix up test to account for creating a new tab

* Fix up test to pass more reliably, remove extra cond

* fmt

Author: franknoirot

tests/e2e.playwright.ts

@@ -1,14 +1,43 @@
 import { ITEMS_PER_PAGE, PAGES_AHEAD_TO_FETCH } from '../src/lib/consts'
 import { expect, test } from '@playwright/test'
 
-test('Redirects to the dashboard from home when logged-in', async ({ page }) => {
+test('User can generate from prompt and open in Zoo Design Studio', async ({ page }) => {
+	test.setTimeout(120_000)
 	// Go to the home page
 	await page.goto('http://localhost:3000')
 
-	// Assert that we are now on the dashboard
-	await page.waitForURL('**/dashboard', { waitUntil: 'domcontentloaded' })
-	await expect(page.locator('h1')).toHaveText('Text-to-CAD')
-	await expect(page.locator('textarea')).toBeFocused()
+	// locators and constants
+	const prompt = `A shepherd's hook bolt`
+	const promptInput = page.locator('textarea')
+	const submitButton = page.locator('button[type="submit"]')
+	const downloadLink = page.getByRole('link', { name: 'download' })
+	const openInDesignStudioLink = page.getByRole('link', { name: 'open in zoo design studio' })
+
+	await test.step('Setup', async () => {
+		// Assert that we are now on the dashboard
+		await page.waitForURL('**/dashboard', { waitUntil: 'networkidle' })
+		await expect(page.locator('h1')).toHaveText('Text-to-CAD')
+		await expect(promptInput).toBeFocused()
+	})
+
+	await test.step('Submit prompt and view successful generation', async () => {
+		await promptInput.fill(prompt)
+		await submitButton.click()
+
+		await page.waitForURL('**/view/*')
+		await expect(page.getByRole('heading', { name: prompt })).toBeVisible()
+
+		// Sometimes generation can take a while
+		await expect(downloadLink).toBeEnabled({ timeout: 90_000 })
+	})
+
+	await test.step('Open the result in Zoo Design Studio temporary workspace', async () => {
+		const newTabPromise = page.waitForEvent('popup')
+		await openInDesignStudioLink.click()
+		const newTab = await newTabPromise
+		await newTab.waitForURL('https://app.zoo.dev/?ask-open-desktop=true&create-file=true*')
+		await newTab.getByRole('button', { name: 'continue to web' }).click()
+	})
 })
 
 test('Prompt input is visible and usable on mobile', async ({ page }) => {