|
|
@@ -0,0 +1,86 @@
|
|
|
+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)
|
|
|
+})
|