scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-25 01:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

test(ui): preserve cleanup on failed readback

Browse source
This commit is contained in:
Yuneng Jiang 2026-09-17 23:15:31 -07:00
parent b4c3adc37d
commit 035271b510
No known key found for this signature in database
2 changed files with 42 additions and 44 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
tests/e2e/ui/tests/prompts/addPrompt.spec.ts
View file

@ -17,32 +17,32 @@ test.describe("Prompt upload form", () => {
await navigateToPage(page, DashboardPage.Prompts);
await page.getByRole("button", { name: "Upload .prompt File" }).click();
await expect(
page.getByRole("dialog", { name: "Add New Prompt" }),
).toBeVisible();
await page.getByLabel("Prompt ID").fill(promptId);
await page.locator('input[type="file"]').setInputFiles({
name: "e2e.prompt",
mimeType: "text/plain",
buffer: Buffer.from(
`model: fake-openai-gpt-4\ntemplate: "${promptContent}"\n`,
),
});
await expect(page.getByText("Selected: e2e.prompt")).toBeVisible();
await page.getByRole("button", { name: "Create Prompt" }).click();
await expect
.poll(async () => {
const response = await page.request.get(
`/prompts/${encodeURIComponent(promptId)}/info?environment=development`,
{
headers: { Authorization: `Bearer ${masterKey()}` },
},
);
return response.ok();
})
.toBe(true);
try {
await expect(
page.getByRole("dialog", { name: "Add New Prompt" }),
).toBeVisible();
await page.getByLabel("Prompt ID").fill(promptId);
await page.locator('input[type="file"]').setInputFiles({
name: "e2e.prompt",
mimeType: "text/plain",
buffer: Buffer.from(
`model: fake-openai-gpt-4\ntemplate: "${promptContent}"\n`,
),
});
await expect(page.getByText("Selected: e2e.prompt")).toBeVisible();
await page.getByRole("button", { name: "Create Prompt" }).click();
await expect
.poll(async () => {
const response = await page.request.get(
`/prompts/${encodeURIComponent(promptId)}/info?environment=development`,
{
headers: { Authorization: `Bearer ${masterKey()}` },
},
);
return response.ok();
})
.toBe(true);
await expect
.poll(async () => {
const response = await page.request.get(
@ -60,13 +60,12 @@ test.describe("Prompt upload form", () => {
.toContain(promptContent);
await expect(page.getByText(promptId, { exact: true })).toBeVisible();
} finally {
const deleteResponse = await page.request.delete(
await page.request.delete(
`/prompts/${encodeURIComponent(promptId)}?environment=development`,
{
headers: { Authorization: `Bearer ${masterKey()}` },
},
);
expect(deleteResponse.ok()).toBe(true);
}
});
});

33
tests/e2e/ui/tests/tagManagement/tagManagement.spec.ts
View file

@ -17,22 +17,22 @@ test.describe("Tag management", () => {
await navigateToPage(page, DashboardPage.TagManagement);
await page.getByRole("button", { name: "+ Create New Tag" }).click();
await expect(
page.getByRole("dialog", { name: "Create New Tag" }),
).toBeVisible();
await page.getByLabel("Tag Name").fill(tagName);
await page.getByLabel("Description").fill(description);
await page.getByRole("button", { name: "Create Tag" }).click();
await expect
.poll(async () => {
const response = await readBack<
Record<string, Record<string, unknown>>
>(page, "/tag/list");
return Object.values(response).some((tag) => tag.name === tagName);
})
.toBe(true);
try {
await expect(
page.getByRole("dialog", { name: "Create New Tag" }),
).toBeVisible();
await page.getByLabel("Tag Name").fill(tagName);
await page.getByLabel("Description").fill(description);
await page.getByRole("button", { name: "Create Tag" }).click();
await expect
.poll(async () => {
const response = await readBack<
Record<string, Record<string, unknown>>
>(page, "/tag/list");
return Object.values(response).some((tag) => tag.name === tagName);
})
.toBe(true);
await expect(page.getByText(tagName, { exact: true })).toBeVisible();
await page.getByText(tagName, { exact: true }).click();
@ -64,14 +64,13 @@ test.describe("Tag management", () => {
})
.toBe(updatedDescription);
} finally {
const deleteResponse = await page.request.post("/tag/delete", {
await page.request.post("/tag/delete", {
headers: {
Authorization: `Bearer ${masterKey()}`,
"Content-Type": "application/json",
},
data: { name: tagName },
});
expect(deleteResponse.ok()).toBe(true);
}
});
});