|
|
@@ -0,0 +1,391 @@
|
|
|
+import { base64Decode } from "@opencode-ai/util/encode"
|
|
|
+import fs from "node:fs/promises"
|
|
|
+import path from "node:path"
|
|
|
+import type { Page } from "@playwright/test"
|
|
|
+
|
|
|
+import { test, expect } from "../fixtures"
|
|
|
+
|
|
|
+test.describe.configure({ mode: "serial" })
|
|
|
+import {
|
|
|
+ cleanupTestProject,
|
|
|
+ clickMenuItem,
|
|
|
+ confirmDialog,
|
|
|
+ createTestProject,
|
|
|
+ openSidebar,
|
|
|
+ openWorkspaceMenu,
|
|
|
+ seedProjects,
|
|
|
+ setWorkspacesEnabled,
|
|
|
+} from "../actions"
|
|
|
+import { inlineInputSelector, projectSwitchSelector, workspaceItemSelector } from "../selectors"
|
|
|
+import { dirSlug } from "../utils"
|
|
|
+
|
|
|
+function slugFromUrl(url: string) {
|
|
|
+ return /\/([^/]+)\/session(?:\/|$)/.exec(url)?.[1] ?? ""
|
|
|
+}
|
|
|
+
|
|
|
+async function setupWorkspaceTest(page: Page, directory: string, gotoSession: () => Promise<void>) {
|
|
|
+ const project = await createTestProject()
|
|
|
+ const rootSlug = dirSlug(project)
|
|
|
+ await seedProjects(page, { directory, extra: [project] })
|
|
|
+
|
|
|
+ await gotoSession()
|
|
|
+ await openSidebar(page)
|
|
|
+
|
|
|
+ const target = page.locator(projectSwitchSelector(rootSlug)).first()
|
|
|
+ await expect(target).toBeVisible()
|
|
|
+ await target.click()
|
|
|
+ await expect(page).toHaveURL(new RegExp(`/${rootSlug}/session`))
|
|
|
+
|
|
|
+ await openSidebar(page)
|
|
|
+ await setWorkspacesEnabled(page, rootSlug, true)
|
|
|
+
|
|
|
+ await page.getByRole("button", { name: "New workspace" }).first().click()
|
|
|
+ await expect
|
|
|
+ .poll(
|
|
|
+ () => {
|
|
|
+ const slug = slugFromUrl(page.url())
|
|
|
+ return slug.length > 0 && slug !== rootSlug
|
|
|
+ },
|
|
|
+ { timeout: 45_000 },
|
|
|
+ )
|
|
|
+ .toBe(true)
|
|
|
+
|
|
|
+ const slug = slugFromUrl(page.url())
|
|
|
+ const dir = base64Decode(slug)
|
|
|
+
|
|
|
+ await openSidebar(page)
|
|
|
+
|
|
|
+ await expect
|
|
|
+ .poll(
|
|
|
+ async () => {
|
|
|
+ const item = page.locator(workspaceItemSelector(slug)).first()
|
|
|
+ try {
|
|
|
+ await item.hover({ timeout: 500 })
|
|
|
+ return true
|
|
|
+ } catch {
|
|
|
+ return false
|
|
|
+ }
|
|
|
+ },
|
|
|
+ { timeout: 60_000 },
|
|
|
+ )
|
|
|
+ .toBe(true)
|
|
|
+
|
|
|
+ return { project, rootSlug, slug, directory: dir }
|
|
|
+}
|
|
|
+
|
|
|
+test("can enable and disable workspaces from project menu", async ({ page, directory, gotoSession }) => {
|
|
|
+ await page.setViewportSize({ width: 1400, height: 800 })
|
|
|
+
|
|
|
+ const project = await createTestProject()
|
|
|
+ const slug = dirSlug(project)
|
|
|
+ await seedProjects(page, { directory, extra: [project] })
|
|
|
+
|
|
|
+ try {
|
|
|
+ await gotoSession()
|
|
|
+ await openSidebar(page)
|
|
|
+
|
|
|
+ const target = page.locator(projectSwitchSelector(slug)).first()
|
|
|
+ await expect(target).toBeVisible()
|
|
|
+ await target.click()
|
|
|
+ await expect(page).toHaveURL(new RegExp(`/${slug}/session`))
|
|
|
+
|
|
|
+ await openSidebar(page)
|
|
|
+
|
|
|
+ await expect(page.getByRole("button", { name: "New session" }).first()).toBeVisible()
|
|
|
+ await expect(page.getByRole("button", { name: "New workspace" })).toHaveCount(0)
|
|
|
+
|
|
|
+ await setWorkspacesEnabled(page, slug, true)
|
|
|
+ await expect(page.getByRole("button", { name: "New workspace" }).first()).toBeVisible()
|
|
|
+ await expect(page.locator(workspaceItemSelector(slug)).first()).toBeVisible()
|
|
|
+
|
|
|
+ await setWorkspacesEnabled(page, slug, false)
|
|
|
+ await expect(page.getByRole("button", { name: "New session" }).first()).toBeVisible()
|
|
|
+ await expect(page.locator(workspaceItemSelector(slug))).toHaveCount(0)
|
|
|
+ } finally {
|
|
|
+ await cleanupTestProject(project)
|
|
|
+ }
|
|
|
+})
|
|
|
+
|
|
|
+test("can create a workspace", async ({ page, directory, gotoSession }) => {
|
|
|
+ await page.setViewportSize({ width: 1400, height: 800 })
|
|
|
+
|
|
|
+ const project = await createTestProject()
|
|
|
+ const slug = dirSlug(project)
|
|
|
+ await seedProjects(page, { directory, extra: [project] })
|
|
|
+
|
|
|
+ try {
|
|
|
+ await gotoSession()
|
|
|
+ await openSidebar(page)
|
|
|
+
|
|
|
+ const target = page.locator(projectSwitchSelector(slug)).first()
|
|
|
+ await expect(target).toBeVisible()
|
|
|
+ await target.click()
|
|
|
+ await expect(page).toHaveURL(new RegExp(`/${slug}/session`))
|
|
|
+
|
|
|
+ await openSidebar(page)
|
|
|
+ await setWorkspacesEnabled(page, slug, true)
|
|
|
+
|
|
|
+ await expect(page.getByRole("button", { name: "New workspace" }).first()).toBeVisible()
|
|
|
+
|
|
|
+ await page.getByRole("button", { name: "New workspace" }).first().click()
|
|
|
+
|
|
|
+ await expect
|
|
|
+ .poll(
|
|
|
+ () => {
|
|
|
+ const currentSlug = slugFromUrl(page.url())
|
|
|
+ return currentSlug.length > 0 && currentSlug !== slug
|
|
|
+ },
|
|
|
+ { timeout: 45_000 },
|
|
|
+ )
|
|
|
+ .toBe(true)
|
|
|
+
|
|
|
+ const workspaceSlug = slugFromUrl(page.url())
|
|
|
+ const workspaceDir = base64Decode(workspaceSlug)
|
|
|
+
|
|
|
+ await openSidebar(page)
|
|
|
+
|
|
|
+ await expect
|
|
|
+ .poll(
|
|
|
+ async () => {
|
|
|
+ const item = page.locator(workspaceItemSelector(workspaceSlug)).first()
|
|
|
+ try {
|
|
|
+ await item.hover({ timeout: 500 })
|
|
|
+ return true
|
|
|
+ } catch {
|
|
|
+ return false
|
|
|
+ }
|
|
|
+ },
|
|
|
+ { timeout: 60_000 },
|
|
|
+ )
|
|
|
+ .toBe(true)
|
|
|
+
|
|
|
+ await expect(page.locator(workspaceItemSelector(workspaceSlug)).first()).toBeVisible()
|
|
|
+
|
|
|
+ await cleanupTestProject(workspaceDir)
|
|
|
+ } finally {
|
|
|
+ await cleanupTestProject(project)
|
|
|
+ }
|
|
|
+})
|
|
|
+
|
|
|
+test("can rename a workspace", async ({ page, directory, gotoSession }) => {
|
|
|
+ await page.setViewportSize({ width: 1400, height: 800 })
|
|
|
+
|
|
|
+ const { project, slug } = await setupWorkspaceTest(page, directory, gotoSession)
|
|
|
+
|
|
|
+ try {
|
|
|
+ const rename = `e2e workspace ${Date.now()}`
|
|
|
+ const menu = await openWorkspaceMenu(page, slug)
|
|
|
+ await clickMenuItem(menu, /^Rename$/i, { force: true })
|
|
|
+
|
|
|
+ await expect(menu).toHaveCount(0)
|
|
|
+
|
|
|
+ const item = page.locator(workspaceItemSelector(slug)).first()
|
|
|
+ await expect(item).toBeVisible()
|
|
|
+ const input = item.locator(inlineInputSelector).first()
|
|
|
+ await expect(input).toBeVisible()
|
|
|
+ await input.fill(rename)
|
|
|
+ await input.press("Enter")
|
|
|
+ await expect(item).toContainText(rename)
|
|
|
+ } finally {
|
|
|
+ await cleanupTestProject(project)
|
|
|
+ }
|
|
|
+})
|
|
|
+
|
|
|
+test("can reset a workspace", async ({ page, directory, sdk, gotoSession }) => {
|
|
|
+ await page.setViewportSize({ width: 1400, height: 800 })
|
|
|
+
|
|
|
+ const { project, slug, directory: createdDir } = await setupWorkspaceTest(page, directory, gotoSession)
|
|
|
+
|
|
|
+ try {
|
|
|
+ const readme = path.join(createdDir, "README.md")
|
|
|
+ const extra = path.join(createdDir, `e2e_reset_${Date.now()}.txt`)
|
|
|
+ const original = await fs.readFile(readme, "utf8")
|
|
|
+ const dirty = `${original.trimEnd()}\n\nchange_${Date.now()}\n`
|
|
|
+ await fs.writeFile(readme, dirty, "utf8")
|
|
|
+ await fs.writeFile(extra, `created_${Date.now()}\n`, "utf8")
|
|
|
+
|
|
|
+ await expect
|
|
|
+ .poll(async () => {
|
|
|
+ return await fs
|
|
|
+ .stat(extra)
|
|
|
+ .then(() => true)
|
|
|
+ .catch(() => false)
|
|
|
+ })
|
|
|
+ .toBe(true)
|
|
|
+
|
|
|
+ await expect
|
|
|
+ .poll(async () => {
|
|
|
+ const files = await sdk.file
|
|
|
+ .status({ directory: createdDir })
|
|
|
+ .then((r) => r.data ?? [])
|
|
|
+ .catch(() => [])
|
|
|
+ return files.length
|
|
|
+ })
|
|
|
+ .toBeGreaterThan(0)
|
|
|
+
|
|
|
+ const menu = await openWorkspaceMenu(page, slug)
|
|
|
+ await clickMenuItem(menu, /^Reset$/i, { force: true })
|
|
|
+ await confirmDialog(page, /^Reset workspace$/i)
|
|
|
+
|
|
|
+ await expect
|
|
|
+ .poll(
|
|
|
+ async () => {
|
|
|
+ const files = await sdk.file
|
|
|
+ .status({ directory: createdDir })
|
|
|
+ .then((r) => r.data ?? [])
|
|
|
+ .catch(() => [])
|
|
|
+ return files.length
|
|
|
+ },
|
|
|
+ { timeout: 60_000 },
|
|
|
+ )
|
|
|
+ .toBe(0)
|
|
|
+
|
|
|
+ await expect.poll(() => fs.readFile(readme, "utf8"), { timeout: 60_000 }).toBe(original)
|
|
|
+
|
|
|
+ await expect
|
|
|
+ .poll(async () => {
|
|
|
+ return await fs
|
|
|
+ .stat(extra)
|
|
|
+ .then(() => true)
|
|
|
+ .catch(() => false)
|
|
|
+ })
|
|
|
+ .toBe(false)
|
|
|
+ } finally {
|
|
|
+ await cleanupTestProject(project)
|
|
|
+ }
|
|
|
+})
|
|
|
+
|
|
|
+test("can delete a workspace", async ({ page, directory, gotoSession }) => {
|
|
|
+ await page.setViewportSize({ width: 1400, height: 800 })
|
|
|
+
|
|
|
+ const { project, rootSlug, slug } = await setupWorkspaceTest(page, directory, gotoSession)
|
|
|
+
|
|
|
+ try {
|
|
|
+ const menu = await openWorkspaceMenu(page, slug)
|
|
|
+ await clickMenuItem(menu, /^Delete$/i, { force: true })
|
|
|
+ await confirmDialog(page, /^Delete workspace$/i)
|
|
|
+
|
|
|
+ await expect(page).toHaveURL(new RegExp(`/${rootSlug}/session`))
|
|
|
+ await expect(page.locator(workspaceItemSelector(slug))).toHaveCount(0)
|
|
|
+ await expect(page.locator(workspaceItemSelector(rootSlug)).first()).toBeVisible()
|
|
|
+ } finally {
|
|
|
+ await cleanupTestProject(project)
|
|
|
+ }
|
|
|
+})
|
|
|
+
|
|
|
+test("can reorder workspaces by drag and drop", async ({ page, directory, gotoSession }) => {
|
|
|
+ await page.setViewportSize({ width: 1400, height: 800 })
|
|
|
+
|
|
|
+ const project = await createTestProject()
|
|
|
+ const rootSlug = dirSlug(project)
|
|
|
+ await seedProjects(page, { directory, extra: [project] })
|
|
|
+
|
|
|
+ const workspaces = [] as { directory: string; slug: string }[]
|
|
|
+
|
|
|
+ const listSlugs = async () => {
|
|
|
+ const nodes = page.locator('[data-component="sidebar-nav-desktop"] [data-component="workspace-item"]')
|
|
|
+ const slugs = await nodes.evaluateAll((els) => {
|
|
|
+ return els.map((el) => el.getAttribute("data-workspace") ?? "").filter((x) => x.length > 0)
|
|
|
+ })
|
|
|
+ return slugs
|
|
|
+ }
|
|
|
+
|
|
|
+ const waitReady = async (slug: string) => {
|
|
|
+ await expect
|
|
|
+ .poll(
|
|
|
+ async () => {
|
|
|
+ const item = page.locator(workspaceItemSelector(slug)).first()
|
|
|
+ try {
|
|
|
+ await item.hover({ timeout: 500 })
|
|
|
+ return true
|
|
|
+ } catch {
|
|
|
+ return false
|
|
|
+ }
|
|
|
+ },
|
|
|
+ { timeout: 60_000 },
|
|
|
+ )
|
|
|
+ .toBe(true)
|
|
|
+ }
|
|
|
+
|
|
|
+ const drag = async (from: string, to: string) => {
|
|
|
+ const src = page.locator(workspaceItemSelector(from)).first()
|
|
|
+ const dst = page.locator(workspaceItemSelector(to)).first()
|
|
|
+
|
|
|
+ await src.scrollIntoViewIfNeeded()
|
|
|
+ await dst.scrollIntoViewIfNeeded()
|
|
|
+
|
|
|
+ const a = await src.boundingBox()
|
|
|
+ const b = await dst.boundingBox()
|
|
|
+ if (!a || !b) throw new Error("Failed to resolve workspace drag bounds")
|
|
|
+
|
|
|
+ await page.mouse.move(a.x + a.width / 2, a.y + a.height / 2)
|
|
|
+ await page.mouse.down()
|
|
|
+ await page.mouse.move(b.x + b.width / 2, b.y + b.height / 2, { steps: 12 })
|
|
|
+ await page.mouse.up()
|
|
|
+ }
|
|
|
+
|
|
|
+ try {
|
|
|
+ await gotoSession()
|
|
|
+ await openSidebar(page)
|
|
|
+
|
|
|
+ const target = page.locator(projectSwitchSelector(rootSlug)).first()
|
|
|
+ await expect(target).toBeVisible()
|
|
|
+ await target.click()
|
|
|
+ await expect(page).toHaveURL(new RegExp(`/${rootSlug}/session`))
|
|
|
+
|
|
|
+ await openSidebar(page)
|
|
|
+ await setWorkspacesEnabled(page, rootSlug, true)
|
|
|
+
|
|
|
+ for (const _ of [0, 1]) {
|
|
|
+ const prev = slugFromUrl(page.url())
|
|
|
+ await page.getByRole("button", { name: "New workspace" }).first().click()
|
|
|
+ await expect
|
|
|
+ .poll(
|
|
|
+ () => {
|
|
|
+ const slug = slugFromUrl(page.url())
|
|
|
+ return slug.length > 0 && slug !== rootSlug && slug !== prev
|
|
|
+ },
|
|
|
+ { timeout: 45_000 },
|
|
|
+ )
|
|
|
+ .toBe(true)
|
|
|
+
|
|
|
+ const slug = slugFromUrl(page.url())
|
|
|
+ const dir = base64Decode(slug)
|
|
|
+ workspaces.push({ slug, directory: dir })
|
|
|
+
|
|
|
+ await openSidebar(page)
|
|
|
+ }
|
|
|
+
|
|
|
+ if (workspaces.length !== 2) throw new Error("Expected two created workspaces")
|
|
|
+
|
|
|
+ const a = workspaces[0].slug
|
|
|
+ const b = workspaces[1].slug
|
|
|
+
|
|
|
+ await waitReady(a)
|
|
|
+ await waitReady(b)
|
|
|
+
|
|
|
+ const list = async () => {
|
|
|
+ const slugs = await listSlugs()
|
|
|
+ return slugs.filter((s) => s !== rootSlug && (s === a || s === b)).slice(0, 2)
|
|
|
+ }
|
|
|
+
|
|
|
+ await expect
|
|
|
+ .poll(async () => {
|
|
|
+ const slugs = await list()
|
|
|
+ return slugs.length === 2
|
|
|
+ })
|
|
|
+ .toBe(true)
|
|
|
+
|
|
|
+ const before = await list()
|
|
|
+ const from = before[1]
|
|
|
+ const to = before[0]
|
|
|
+ if (!from || !to) throw new Error("Failed to resolve initial workspace order")
|
|
|
+
|
|
|
+ await drag(from, to)
|
|
|
+
|
|
|
+ await expect.poll(async () => await list()).toEqual([from, to])
|
|
|
+ } finally {
|
|
|
+ await Promise.all(workspaces.map((w) => cleanupTestProject(w.directory)))
|
|
|
+ await cleanupTestProject(project)
|
|
|
+ }
|
|
|
+})
|