87 lines
3.0 KiB
TypeScript
87 lines
3.0 KiB
TypeScript
import { test, expect } from "./fixtures"
|
|
import { modKey, promptSelector } from "./utils"
|
|
|
|
test("hiding a model removes it from the model picker", async ({ page, gotoSession }) => {
|
|
await gotoSession()
|
|
|
|
await page.locator(promptSelector).click()
|
|
await page.keyboard.type("/model")
|
|
|
|
const command = page.locator('[data-slash-id="model.choose"]')
|
|
await expect(command).toBeVisible()
|
|
await command.hover()
|
|
await page.keyboard.press("Enter")
|
|
|
|
const picker = page.getByRole("dialog")
|
|
await expect(picker).toBeVisible()
|
|
|
|
const target = picker.locator('[data-slot="list-item"]').first()
|
|
await expect(target).toBeVisible()
|
|
|
|
const key = await target.getAttribute("data-key")
|
|
if (!key) throw new Error("Failed to resolve model key from list item")
|
|
|
|
const name = (await target.locator("span").first().innerText()).trim()
|
|
if (!name) throw new Error("Failed to resolve model name from list item")
|
|
|
|
await page.keyboard.press("Escape")
|
|
await expect(picker).toHaveCount(0)
|
|
|
|
const settings = page.getByRole("dialog")
|
|
|
|
await page.keyboard.press(`${modKey}+Comma`).catch(() => undefined)
|
|
const opened = await settings
|
|
.waitFor({ state: "visible", timeout: 3000 })
|
|
.then(() => true)
|
|
.catch(() => false)
|
|
|
|
if (!opened) {
|
|
await page.getByRole("button", { name: "Settings" }).first().click()
|
|
await expect(settings).toBeVisible()
|
|
}
|
|
|
|
await settings.getByRole("tab", { name: "Models" }).click()
|
|
const search = settings.getByPlaceholder("Search models")
|
|
await expect(search).toBeVisible()
|
|
await search.fill(name)
|
|
|
|
const toggle = settings.locator('[data-component="switch"]').filter({ hasText: name }).first()
|
|
const input = toggle.locator('[data-slot="switch-input"]')
|
|
await expect(toggle).toBeVisible()
|
|
await expect(input).toHaveAttribute("aria-checked", "true")
|
|
await toggle.locator('[data-slot="switch-control"]').click()
|
|
await expect(input).toHaveAttribute("aria-checked", "false")
|
|
|
|
await page.keyboard.press("Escape")
|
|
const closed = await settings
|
|
.waitFor({ state: "detached", timeout: 1500 })
|
|
.then(() => true)
|
|
.catch(() => false)
|
|
if (!closed) {
|
|
await page.keyboard.press("Escape")
|
|
const closedSecond = await settings
|
|
.waitFor({ state: "detached", timeout: 1500 })
|
|
.then(() => true)
|
|
.catch(() => false)
|
|
if (!closedSecond) {
|
|
await page.locator('[data-component="dialog-overlay"]').click({ position: { x: 5, y: 5 } })
|
|
await expect(settings).toHaveCount(0)
|
|
}
|
|
}
|
|
|
|
await page.locator(promptSelector).click()
|
|
await page.keyboard.type("/model")
|
|
await expect(command).toBeVisible()
|
|
await command.hover()
|
|
await page.keyboard.press("Enter")
|
|
|
|
const pickerAgain = page.getByRole("dialog")
|
|
await expect(pickerAgain).toBeVisible()
|
|
await expect(pickerAgain.locator('[data-slot="list-item"]').first()).toBeVisible()
|
|
|
|
await expect(pickerAgain.locator(`[data-slot="list-item"][data-key="${key}"]`)).toHaveCount(0)
|
|
|
|
await page.keyboard.press("Escape")
|
|
await expect(pickerAgain).toHaveCount(0)
|
|
})
|