Carlos Florêncio
61 lines
1.8 KiB
TypeScript
61 lines
1.8 KiB
TypeScript
import path from "node:path";
|
|
import type { Page } from "@playwright/test";
|
|
import type { SeedData } from "../../fixtures/test-base";
|
|
import type { ApiClient } from "../../helpers/api-client";
|
|
import type { BackendContext } from "../../fixtures/backend";
|
|
import { GitHelper, makeGitEnv } from "../../helpers/git-helper";
|
|
import { SessionPage } from "../../pages/session-page";
|
|
|
|
export const LARGE_FILE_TREE_FOLDER = "large-file-tree";
|
|
export const LARGE_FILE_TREE_COUNT = 600;
|
|
|
|
export function largeFileTreePath(index: number): string {
|
|
return `${LARGE_FILE_TREE_FOLDER}/entry-${index.toString().padStart(4, "0")}.txt`;
|
|
}
|
|
|
|
export function seedLargeFileTree(backend: BackendContext): void {
|
|
const git = new GitHelper(
|
|
path.join(backend.tmpDir, "repos", "e2e-repo"),
|
|
makeGitEnv(backend.tmpDir),
|
|
);
|
|
for (let index = 0; index < LARGE_FILE_TREE_COUNT; index += 1) {
|
|
git.createFile(largeFileTreePath(index), `large tree entry ${index}\n`);
|
|
}
|
|
git.stageAll();
|
|
if (git.exec("git status --short").trim()) {
|
|
git.commit("seed large file tree");
|
|
}
|
|
}
|
|
|
|
export async function setupLargeFileTreeTask({
|
|
testPage,
|
|
apiClient,
|
|
seedData,
|
|
backend,
|
|
title,
|
|
}: {
|
|
testPage: Page;
|
|
apiClient: ApiClient;
|
|
seedData: SeedData;
|
|
backend: BackendContext;
|
|
title: string;
|
|
}): Promise<SessionPage> {
|
|
seedLargeFileTree(backend);
|
|
const task = await apiClient.createTaskWithAgent(
|
|
seedData.workspaceId,
|
|
title,
|
|
seedData.agentProfileId,
|
|
{
|
|
description: "/e2e:simple-message",
|
|
workflow_id: seedData.workflowId,
|
|
workflow_step_id: seedData.startStepId,
|
|
repository_ids: [seedData.repositoryId],
|
|
},
|
|
);
|
|
await testPage.goto(`/t/${task.id}`);
|
|
const session = new SessionPage(testPage);
|
|
await session.waitForLoad();
|
|
await session.waitForChatIdle({ timeout: 45_000 });
|
|
return session;
|
|
}
|