iter 2
This commit is contained in:
@@ -0,0 +1,246 @@
|
||||
import { expect, test } from "@playwright/test"
|
||||
|
||||
const USER_ID = "00000000-0000-0000-0000-000000000001"
|
||||
const PROJECT_ID = "65df675b-013b-4b1f-ab2d-075dadbcd0d9"
|
||||
const CAPTION_PRESET_ID = "00000000-0000-0000-0000-000000000010"
|
||||
const TRANSCRIPTION_ARTIFACT_ID =
|
||||
"00000000-0000-0000-0000-000000000020"
|
||||
const TRANSCRIPTION_ID = "00000000-0000-0000-0000-000000000030"
|
||||
const CAPTION_JOB_ID = "00000000-0000-0000-0000-000000000040"
|
||||
const PRIMARY_FILE_KEY = "projects/test/video.mp4"
|
||||
|
||||
const DEFAULT_USER = {
|
||||
id: USER_ID,
|
||||
username: "testuser",
|
||||
email: "test@example.com",
|
||||
first_name: "Test",
|
||||
last_name: "User",
|
||||
phone_number: null,
|
||||
avatar: null,
|
||||
email_verified: true,
|
||||
phone_verified: false,
|
||||
is_active: true,
|
||||
is_staff: false,
|
||||
is_superuser: false,
|
||||
date_joined: "2025-01-01T00:00:00Z",
|
||||
}
|
||||
|
||||
test.describe("Caption Settings Step", () => {
|
||||
test("should recover a missing transcription artifact from project data", async ({
|
||||
page,
|
||||
}) => {
|
||||
let project: Record<string, unknown> = {
|
||||
id: PROJECT_ID,
|
||||
owner_id: USER_ID,
|
||||
name: "Тестовый проект",
|
||||
description: null,
|
||||
language: "auto",
|
||||
folder: null,
|
||||
status: "DRAFT",
|
||||
workspace_state: {
|
||||
wizard: {
|
||||
current_step: "caption-settings",
|
||||
completed_steps: [
|
||||
"upload",
|
||||
"verify",
|
||||
"silence-settings",
|
||||
"processing",
|
||||
"fragments",
|
||||
"transcription-settings",
|
||||
"transcription-processing",
|
||||
"subtitle-revision",
|
||||
],
|
||||
primary_file_key: PRIMARY_FILE_KEY,
|
||||
video_url: "http://localhost:9000/projects/test/video.mp4",
|
||||
silence_settings: {
|
||||
min_silence_duration_ms: 200,
|
||||
silence_threshold_db: 16,
|
||||
padding_ms: 100,
|
||||
},
|
||||
active_job_id: null,
|
||||
active_job_type: null,
|
||||
silence_job_id: null,
|
||||
transcription_artifact_id: null,
|
||||
caption_preset_id: CAPTION_PRESET_ID,
|
||||
caption_style_config: null,
|
||||
captioned_video_path: null,
|
||||
},
|
||||
},
|
||||
is_active: true,
|
||||
created_at: "2025-06-01T00:00:00Z",
|
||||
updated_at: "2025-06-01T00:00:00Z",
|
||||
}
|
||||
let savedWizardState: Record<string, unknown> | null = null
|
||||
let generateRequestBody: Record<string, unknown> | null = null
|
||||
let generateRequestCount = 0
|
||||
|
||||
await page.context().addCookies([
|
||||
{
|
||||
name: "access_token",
|
||||
value: "fake-access-jwt",
|
||||
domain: "localhost",
|
||||
path: "/",
|
||||
},
|
||||
{
|
||||
name: "refresh_token",
|
||||
value: "fake-refresh-jwt",
|
||||
domain: "localhost",
|
||||
path: "/",
|
||||
},
|
||||
])
|
||||
|
||||
await page.route("**/api/users/me/", async (route) => {
|
||||
await route.fulfill({
|
||||
status: 200,
|
||||
contentType: "application/json",
|
||||
body: JSON.stringify(DEFAULT_USER),
|
||||
})
|
||||
})
|
||||
|
||||
await page.route(`**/api/projects/${PROJECT_ID}/`, async (route) => {
|
||||
if (route.request().method() === "GET") {
|
||||
await route.fulfill({
|
||||
status: 200,
|
||||
contentType: "application/json",
|
||||
body: JSON.stringify(project),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
if (route.request().method() === "PATCH") {
|
||||
const body = route.request().postDataJSON() as {
|
||||
workspace_state?: { wizard?: Record<string, unknown> }
|
||||
}
|
||||
|
||||
savedWizardState = body.workspace_state?.wizard ?? null
|
||||
project = {
|
||||
...project,
|
||||
workspace_state: body.workspace_state ?? project.workspace_state,
|
||||
}
|
||||
|
||||
await route.fulfill({
|
||||
status: 200,
|
||||
contentType: "application/json",
|
||||
body: JSON.stringify(project),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
await route.fallback()
|
||||
})
|
||||
|
||||
await page.route("**/api/media/artifacts/", async (route) => {
|
||||
await route.fulfill({
|
||||
status: 200,
|
||||
contentType: "application/json",
|
||||
body: JSON.stringify([
|
||||
{
|
||||
id: TRANSCRIPTION_ARTIFACT_ID,
|
||||
project_id: PROJECT_ID,
|
||||
file_id: null,
|
||||
media_file_id: null,
|
||||
artifact_type: "TRANSCRIPTION_JSON",
|
||||
is_deleted: false,
|
||||
is_active: true,
|
||||
created_at: "2025-06-01T00:00:00Z",
|
||||
updated_at: "2025-06-01T00:00:00Z",
|
||||
},
|
||||
]),
|
||||
})
|
||||
})
|
||||
|
||||
await page.route(
|
||||
`**/api/transcribe/transcriptions/by-artifact/${TRANSCRIPTION_ARTIFACT_ID}/`,
|
||||
async (route) => {
|
||||
await route.fulfill({
|
||||
status: 200,
|
||||
contentType: "application/json",
|
||||
body: JSON.stringify({
|
||||
id: TRANSCRIPTION_ID,
|
||||
artifact_id: TRANSCRIPTION_ARTIFACT_ID,
|
||||
}),
|
||||
})
|
||||
},
|
||||
)
|
||||
|
||||
await page.route("**/api/captions/presets/", async (route) => {
|
||||
await route.fulfill({
|
||||
status: 200,
|
||||
contentType: "application/json",
|
||||
body: JSON.stringify([
|
||||
{
|
||||
id: CAPTION_PRESET_ID,
|
||||
user_id: null,
|
||||
name: "Системный пресет",
|
||||
description: null,
|
||||
is_system: true,
|
||||
style_config: {},
|
||||
preview_url: null,
|
||||
created_at: "2025-06-01T00:00:00Z",
|
||||
updated_at: "2025-06-01T00:00:00Z",
|
||||
},
|
||||
]),
|
||||
})
|
||||
})
|
||||
|
||||
await page.route("**/api/tasks/captions-generate/", async (route) => {
|
||||
generateRequestCount += 1
|
||||
generateRequestBody = route.request().postDataJSON() as Record<
|
||||
string,
|
||||
unknown
|
||||
>
|
||||
|
||||
await route.fulfill({
|
||||
status: 200,
|
||||
contentType: "application/json",
|
||||
body: JSON.stringify({ job_id: CAPTION_JOB_ID }),
|
||||
})
|
||||
})
|
||||
|
||||
await page.route("**/api/tasks/status/**", async (route) => {
|
||||
await route.fulfill({
|
||||
status: 200,
|
||||
contentType: "application/json",
|
||||
body: JSON.stringify({
|
||||
status: "RUNNING",
|
||||
progress_pct: 0,
|
||||
output_data: null,
|
||||
}),
|
||||
})
|
||||
})
|
||||
|
||||
await page.goto(`/projects/${PROJECT_ID}`)
|
||||
|
||||
const captionStep = page.locator("[data-testid='CaptionSettingsStep']")
|
||||
const generateButton = captionStep.getByRole("button", {
|
||||
name: "Генерировать",
|
||||
})
|
||||
|
||||
await expect(captionStep).toBeVisible()
|
||||
await expect(captionStep.getByText("Системный пресет")).toBeVisible()
|
||||
await expect(generateButton).toBeEnabled()
|
||||
|
||||
await expect
|
||||
.poll(() => savedWizardState?.transcription_artifact_id ?? null)
|
||||
.toBe(TRANSCRIPTION_ARTIFACT_ID)
|
||||
|
||||
await generateButton.click()
|
||||
|
||||
expect(generateRequestBody).toMatchObject({
|
||||
video_s3_path: PRIMARY_FILE_KEY,
|
||||
transcription_id: TRANSCRIPTION_ID,
|
||||
project_id: PROJECT_ID,
|
||||
preset_id: CAPTION_PRESET_ID,
|
||||
})
|
||||
expect(generateRequestCount).toBe(1)
|
||||
|
||||
await expect
|
||||
.poll(() => savedWizardState?.current_step ?? null)
|
||||
.toBe("caption-processing")
|
||||
await expect
|
||||
.poll(() => savedWizardState?.active_job_id ?? null)
|
||||
.toBe(CAPTION_JOB_ID)
|
||||
|
||||
await expect(page.locator("[data-testid='ProcessingStep']")).toBeVisible()
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user