diff --git a/airflow-core/src/airflow/ui/tests/e2e/components/HITLReviewDrawer.ts b/airflow-core/src/airflow/ui/tests/e2e/components/HITLReviewDrawer.ts index 09990c6cd07e6..6cae368b33d09 100644 --- a/airflow-core/src/airflow/ui/tests/e2e/components/HITLReviewDrawer.ts +++ b/airflow-core/src/airflow/ui/tests/e2e/components/HITLReviewDrawer.ts @@ -25,8 +25,8 @@ export class HITLReviewDrawer { this.root = page.getByRole("dialog", { name: "Required Actions" }); } - public async expectOpenWith(dagId: string): Promise { + public async expectOpenWith(expectedText: string): Promise { await expect(this.root).toBeVisible(); - await expect(this.root).toContainText(dagId); + await expect(this.root).toContainText(expectedText); } } diff --git a/airflow-core/src/airflow/ui/tests/e2e/components/HITLReviewModal.ts b/airflow-core/src/airflow/ui/tests/e2e/components/HITLReviewModal.ts index b9975762c0e54..c86adeb3932bf 100644 --- a/airflow-core/src/airflow/ui/tests/e2e/components/HITLReviewModal.ts +++ b/airflow-core/src/airflow/ui/tests/e2e/components/HITLReviewModal.ts @@ -25,8 +25,8 @@ export class HITLReviewModal { this.root = page.getByRole("dialog", { name: "Required Actions" }); } - public async expectOpenWith(dagId: string): Promise { + public async expectOpenWith(expectedText: string): Promise { await expect(this.root).toBeVisible(); - await expect(this.root).toContainText(dagId, { timeout: 60_000 }); + await expect(this.root).toContainText(expectedText); } } diff --git a/airflow-core/src/airflow/ui/tests/e2e/utils/api/hitl.ts b/airflow-core/src/airflow/ui/tests/e2e/utils/api/hitl.ts index 159dd0b05c780..619fb1179ad19 100644 --- a/airflow-core/src/airflow/ui/tests/e2e/utils/api/hitl.ts +++ b/airflow-core/src/airflow/ui/tests/e2e/utils/api/hitl.ts @@ -200,7 +200,7 @@ export async function setupPendingHITLFlowViaAPI(source: RequestLike, dagId: str const response = await request.patch(`${baseUrl}/api/v2/dags/${dagId}`, { data: { is_paused: false } }); if (!response.ok()) { - throw new Error(`HITL response failed (${response.status()})`); + throw new Error(`Failed to unpause Dag ${dagId} (${response.status()})`); } const { dagRunId } = await apiTriggerDagRun(request, dagId);