mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-12 01:02:39 +02:00
test(e2e): cover manual file upload journey
This commit is contained in:
parent
0846857b52
commit
03ce8c1b81
1 changed files with 95 additions and 0 deletions
95
surfsense_web/tests/documents/file-upload/journey.spec.ts
Normal file
95
surfsense_web/tests/documents/file-upload/journey.spec.ts
Normal file
|
|
@ -0,0 +1,95 @@
|
||||||
|
import path from "node:path";
|
||||||
|
import type { APIRequestContext, Page } from "@playwright/test";
|
||||||
|
import { expect, test } from "../../fixtures";
|
||||||
|
import { getEditorContent, listDocuments } from "../../helpers/api/documents";
|
||||||
|
import { CANARY_TOKENS } from "../../helpers/canary";
|
||||||
|
import { waitForDocumentByTitle } from "../../helpers/waits/indexing";
|
||||||
|
|
||||||
|
type UploadFixture = {
|
||||||
|
path: string;
|
||||||
|
name: string;
|
||||||
|
canary: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
type SearchSpace = {
|
||||||
|
id: number;
|
||||||
|
};
|
||||||
|
|
||||||
|
const MD_FILE: UploadFixture = {
|
||||||
|
path: path.join(__dirname, "fixtures", "canary.md"),
|
||||||
|
name: "canary.md",
|
||||||
|
canary: CANARY_TOKENS.manualUploadMdCanary,
|
||||||
|
};
|
||||||
|
|
||||||
|
const PDF_FILE: UploadFixture = {
|
||||||
|
path: path.join(__dirname, "fixtures", "canary.pdf"),
|
||||||
|
name: "canary.pdf",
|
||||||
|
canary: CANARY_TOKENS.manualUploadPdfCanary,
|
||||||
|
};
|
||||||
|
|
||||||
|
async function uploadAndAssert({
|
||||||
|
page,
|
||||||
|
request,
|
||||||
|
apiToken,
|
||||||
|
searchSpace,
|
||||||
|
file,
|
||||||
|
}: {
|
||||||
|
page: Page;
|
||||||
|
request: APIRequestContext;
|
||||||
|
apiToken: string;
|
||||||
|
searchSpace: SearchSpace;
|
||||||
|
file: UploadFixture;
|
||||||
|
}) {
|
||||||
|
await page.goto(`/dashboard/${searchSpace.id}/new-chat`, {
|
||||||
|
waitUntil: "domcontentloaded",
|
||||||
|
});
|
||||||
|
|
||||||
|
await page.getByRole("button", { name: "Upload" }).click();
|
||||||
|
const dialog = page.getByRole("dialog", { name: "Upload Documents" });
|
||||||
|
await expect(dialog).toBeVisible();
|
||||||
|
|
||||||
|
await dialog.locator('input[type="file"]').first().setInputFiles(file.path);
|
||||||
|
await dialog.getByRole("button", { name: /Upload 1 file/i }).click();
|
||||||
|
|
||||||
|
await expect(page.getByText(/Upload Task Initiated/i)).toBeVisible({
|
||||||
|
timeout: 15_000,
|
||||||
|
});
|
||||||
|
|
||||||
|
await waitForDocumentByTitle(request, apiToken, searchSpace.id, file.name, {
|
||||||
|
timeoutMs: 200_000,
|
||||||
|
});
|
||||||
|
|
||||||
|
const docs = await listDocuments(request, apiToken, searchSpace.id);
|
||||||
|
const uploaded = docs.find((d) => d.title === file.name);
|
||||||
|
expect(uploaded, `${file.name} must exist after indexing`).toBeDefined();
|
||||||
|
if (!uploaded) throw new Error("unreachable: uploaded asserted defined above");
|
||||||
|
expect(uploaded.document_type).toBe("FILE");
|
||||||
|
|
||||||
|
const editor = await getEditorContent(request, apiToken, searchSpace.id, uploaded.id);
|
||||||
|
expect(editor.source_markdown).toContain(file.canary);
|
||||||
|
expect(editor.chunk_count).toBeGreaterThan(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
test.describe("Manual file upload journey", () => {
|
||||||
|
test("user uploads a markdown file (PLAINTEXT branch)", async ({
|
||||||
|
page,
|
||||||
|
request,
|
||||||
|
apiToken,
|
||||||
|
searchSpace,
|
||||||
|
}) => {
|
||||||
|
test.setTimeout(180_000);
|
||||||
|
|
||||||
|
await uploadAndAssert({ page, request, apiToken, searchSpace, file: MD_FILE });
|
||||||
|
});
|
||||||
|
|
||||||
|
test("user uploads a PDF (DOCUMENT branch via real Docling)", async ({
|
||||||
|
page,
|
||||||
|
request,
|
||||||
|
apiToken,
|
||||||
|
searchSpace,
|
||||||
|
}) => {
|
||||||
|
test.setTimeout(240_000); // Docling cold-start can take 30-60s on first invocation.
|
||||||
|
|
||||||
|
await uploadAndAssert({ page, request, apiToken, searchSpace, file: PDF_FILE });
|
||||||
|
});
|
||||||
|
});
|
||||||
Loading…
Add table
Add a link
Reference in a new issue