opencode/packages/app/e2e/session/session.spec.ts

import { test, expect } from "../fixtures"
import {
  openSidebar,
  openSessionMoreMenu,
  clickMenuItem,
  confirmDialog,
  openSharePopover,
  withSession,
} from "../actions"
import { sessionItemSelector, inlineInputSelector } from "../selectors"

const shareDisabled = process.env.OPENCODE_DISABLE_SHARE === "true" || process.env.OPENCODE_DISABLE_SHARE === "1"

type Sdk = Parameters<typeof withSession>[0]

async function seedMessage(sdk: Sdk, sessionID: string) {
  await sdk.session.promptAsync({
    sessionID,
    noReply: true,
    parts: [{ type: "text", text: "e2e seed" }],
  })

  await expect
    .poll(
      async () => {
        const messages = await sdk.session.messages({ sessionID, limit: 1 }).then((r) => r.data ?? [])
        return messages.length
      },
      { timeout: 30_000 },
    )
    .toBeGreaterThan(0)
}

test("session can be renamed via header menu", async ({ page, project }) => {
  const stamp = Date.now()
  const originalTitle = `e2e rename test ${stamp}`
  const renamedTitle = `e2e renamed ${stamp}`

  await project.open()
  await withSession(project.sdk, originalTitle, async (session) => {
    project.trackSession(session.id)
    await seedMessage(project.sdk, session.id)
    await project.gotoSession(session.id)
    await expect(page.getByRole("heading", { level: 1 }).first()).toHaveText(originalTitle)

    const menu = await openSessionMoreMenu(page, session.id)
    await clickMenuItem(menu, /rename/i)

    const input = page.locator(".scroll-view__viewport").locator(inlineInputSelector).first()
    await expect(input).toBeVisible()
    await expect(input).toBeFocused()
    await input.fill(renamedTitle)
    await expect(input).toHaveValue(renamedTitle)
    await input.press("Enter")

    await expect
      .poll(
        async () => {
          const data = await project.sdk.session.get({ sessionID: session.id }).then((r) => r.data)
          return data?.title
        },
        { timeout: 30_000 },
      )
      .toBe(renamedTitle)

    await expect(page.getByRole("heading", { level: 1 }).first()).toHaveText(renamedTitle)
  })
})

test("session can be archived via header menu", async ({ page, project }) => {
  const stamp = Date.now()
  const title = `e2e archive test ${stamp}`

  await project.open()
  await withSession(project.sdk, title, async (session) => {
    project.trackSession(session.id)
    await seedMessage(project.sdk, session.id)
    await project.gotoSession(session.id)
    const menu = await openSessionMoreMenu(page, session.id)
    await clickMenuItem(menu, /archive/i)

    await expect
      .poll(
        async () => {
          const data = await project.sdk.session.get({ sessionID: session.id }).then((r) => r.data)
          return data?.time?.archived
        },
        { timeout: 30_000 },
      )
      .not.toBeUndefined()

    await openSidebar(page)
    await expect(page.locator(sessionItemSelector(session.id))).toHaveCount(0)
  })
})

test("session can be deleted via header menu", async ({ page, project }) => {
  const stamp = Date.now()
  const title = `e2e delete test ${stamp}`

  await project.open()
  await withSession(project.sdk, title, async (session) => {
    project.trackSession(session.id)
    await seedMessage(project.sdk, session.id)
    await project.gotoSession(session.id)
    const menu = await openSessionMoreMenu(page, session.id)
    await clickMenuItem(menu, /delete/i)
    await confirmDialog(page, /delete/i)

    await expect
      .poll(
        async () => {
          const data = await project.sdk.session
            .get({ sessionID: session.id })
            .then((r) => r.data)
            .catch(() => undefined)
          return data?.id
        },
        { timeout: 30_000 },
      )
      .toBeUndefined()

    await openSidebar(page)
    await expect(page.locator(sessionItemSelector(session.id))).toHaveCount(0)
  })
})

test("session can be shared and unshared via header button", async ({ page, project }) => {
  test.skip(shareDisabled, "Share is disabled in this environment (OPENCODE_DISABLE_SHARE).")

  const stamp = Date.now()
  const title = `e2e share test ${stamp}`

  await project.open()
  await withSession(project.sdk, title, async (session) => {
    project.trackSession(session.id)
    await project.gotoSession(session.id)
    await project.prompt(`share seed ${stamp}`)

    const shared = await openSharePopover(page)
    const publish = shared.popoverBody.getByRole("button", { name: "Publish" }).first()
    await expect(publish).toBeVisible({ timeout: 30_000 })
    await publish.click()

    await expect(shared.popoverBody.getByRole("button", { name: "Unpublish" }).first()).toBeVisible({
      timeout: 30_000,
    })

    await expect
      .poll(
        async () => {
          const data = await project.sdk.session.get({ sessionID: session.id }).then((r) => r.data)
          return data?.share?.url || undefined
        },
        { timeout: 30_000 },
      )
      .not.toBeUndefined()

    const unpublish = shared.popoverBody.getByRole("button", { name: "Unpublish" }).first()
    await expect(unpublish).toBeVisible({ timeout: 30_000 })
    await unpublish.click()

    await expect(shared.popoverBody.getByRole("button", { name: "Publish" }).first()).toBeVisible({
      timeout: 30_000,
    })

    await expect
      .poll(
        async () => {
          const data = await project.sdk.session.get({ sessionID: session.id }).then((r) => r.data)
          return data?.share?.url || undefined
        },
        { timeout: 30_000 },
      )
      .toBeUndefined()

    const unshared = await openSharePopover(page)
    await expect(unshared.popoverBody.getByRole("button", { name: "Publish" }).first()).toBeVisible({
      timeout: 30_000,
    })
  })
})