|
| 1 | +import { expect, test, type Page } from "@playwright/test" |
| 2 | +import { base64Encode } from "@opencode-ai/core/util/encode" |
| 3 | +import { mockOpenCodeServer } from "../utils/mock-server" |
| 4 | + |
| 5 | +const directory = "C:/OpenCode/PromptThinkingLevelRegression" |
| 6 | +const projectID = "proj_prompt_thinking_level_regression" |
| 7 | +const sessionID = "ses_prompt_thinking_level_regression" |
| 8 | + |
| 9 | +test("shows the V2 thinking level control while relevant", async ({ page }) => { |
| 10 | + await mockOpenCodeServer(page, { |
| 11 | + directory, |
| 12 | + project: { |
| 13 | + id: projectID, |
| 14 | + worktree: directory, |
| 15 | + vcs: "git", |
| 16 | + name: "prompt-thinking-level-regression", |
| 17 | + time: { created: 1700000000000, updated: 1700000000000 }, |
| 18 | + sandboxes: [], |
| 19 | + }, |
| 20 | + provider: { |
| 21 | + all: [ |
| 22 | + { |
| 23 | + id: "opencode", |
| 24 | + name: "OpenCode", |
| 25 | + models: { |
| 26 | + "thinking-model": { |
| 27 | + id: "thinking-model", |
| 28 | + name: "Thinking Model", |
| 29 | + limit: { context: 200_000 }, |
| 30 | + variants: { high: {} }, |
| 31 | + }, |
| 32 | + }, |
| 33 | + }, |
| 34 | + ], |
| 35 | + connected: ["opencode"], |
| 36 | + default: { providerID: "opencode", modelID: "thinking-model" }, |
| 37 | + }, |
| 38 | + sessions: [ |
| 39 | + { |
| 40 | + id: sessionID, |
| 41 | + slug: "prompt-thinking-level-regression", |
| 42 | + projectID, |
| 43 | + directory, |
| 44 | + title: "Prompt thinking level regression", |
| 45 | + version: "dev", |
| 46 | + time: { created: 1700000000000, updated: 1700000000000 }, |
| 47 | + }, |
| 48 | + ], |
| 49 | + pageMessages: () => ({ items: [] }), |
| 50 | + }) |
| 51 | + await page.addInitScript(() => { |
| 52 | + localStorage.setItem("settings.v3", JSON.stringify({ general: { newLayoutDesigns: true } })) |
| 53 | + }) |
| 54 | + |
| 55 | + await page.goto(`/${base64Encode(directory)}/session/${sessionID}`) |
| 56 | + const composer = page.locator('[data-component="session-composer"]') |
| 57 | + const input = composer.locator('[data-component="prompt-input"]') |
| 58 | + const control = composer.locator('[data-component="prompt-variant-control"]') |
| 59 | + await expect(composer).toBeVisible() |
| 60 | + |
| 61 | + await idleComposer(page) |
| 62 | + await expect(control).toBeHidden() |
| 63 | + |
| 64 | + await composer.hover() |
| 65 | + await expect(control).toBeVisible() |
| 66 | + |
| 67 | + await control.locator('[data-action="prompt-model-variant"]').click() |
| 68 | + const high = page.getByRole("option", { name: "high" }) |
| 69 | + await expect(high).toBeVisible() |
| 70 | + await page.mouse.move(0, 0) |
| 71 | + await expect(control).toBeVisible() |
| 72 | + await expect(high).toBeVisible() |
| 73 | + await high.click() |
| 74 | + |
| 75 | + await idleComposer(page) |
| 76 | + await input.focus() |
| 77 | + await expect(control).toBeVisible() |
| 78 | + |
| 79 | + await idleComposer(page) |
| 80 | + await expect(control).toBeVisible() |
| 81 | +}) |
| 82 | + |
| 83 | +async function idleComposer(page: Page) { |
| 84 | + await page.mouse.move(0, 0) |
| 85 | + await page.evaluate(() => (document.activeElement as HTMLElement | null)?.blur()) |
| 86 | +} |
0 commit comments