vishakha1411 commented on code in PR #60738: URL: https://github.com/apache/airflow/pull/60738#discussion_r2794724317
########## airflow-core/src/airflow/ui/tests/e2e/pages/ConnectionsPage.ts: ########## @@ -0,0 +1,622 @@ +/*! + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ +import { expect, type Locator, type Page } from "@playwright/test"; +import { BasePage } from "tests/e2e/pages/BasePage"; + +type ConnectionDetails = { + conn_type: string; + connection_id: string; + description?: string; + extra?: string; + host?: string; + login?: string; + password?: string; + port?: number | string; + schema?: string; +}; + +export class ConnectionsPage extends BasePage { + // Page URLs + public static get connectionsListUrl(): string { + return "/connections"; + } + + public readonly addButton: Locator; + public readonly confirmDeleteButton: Locator; + public readonly connectionForm: Locator; + public readonly connectionIdHeader: Locator; + public readonly connectionIdInput: Locator; + // Core page elements + public readonly connectionsTable: Locator; + public readonly connectionTypeHeader: Locator; + public readonly connectionTypeSelect: Locator; + public readonly descriptionInput: Locator; + public readonly emptyState: Locator; + public readonly hostHeader: Locator; + public readonly hostInput: Locator; + public readonly loginInput: Locator; + + // Pagination elements + public readonly paginationNextButton: Locator; + public readonly paginationPrevButton: Locator; + public readonly passwordInput: Locator; + + public readonly portInput: Locator; + public readonly rowsPerPageSelect: Locator; + public readonly saveButton: Locator; + + public readonly schemaInput: Locator; + public readonly searchInput: Locator; + public readonly successAlert: Locator; + // Sorting and filtering + public readonly tableHeader: Locator; + public readonly testConnectionButton: Locator; + + public constructor(page: Page) { + super(page); + // Table elements (Chakra UI DataTable) + this.connectionsTable = page.locator('[role="grid"], table'); + this.emptyState = page.locator("text=/No connection found!/i"); + + // Action buttons + this.addButton = page.getByRole("button", { name: "Add Connection" }); + this.testConnectionButton = page.locator('button:has-text("Test")'); + this.saveButton = page.getByRole("button", { name: /^save$/i }); + + // Form inputs (Chakra UI inputs) + this.connectionForm = page.locator('[data-scope="dialog"][data-part="content"]'); + this.connectionIdInput = page.locator('input[name="connection_id"]').first(); + this.connectionTypeSelect = page.getByRole("combobox").first(); + this.hostInput = page.locator('input[name="host"]').first(); + this.portInput = page.locator('input[name="port"]').first(); + this.loginInput = page.locator('input[name="login"]').first(); + this.passwordInput = page.locator('input[name="password"], input[type="password"]').first(); + this.schemaInput = page.locator('input[name="schema"]').first(); + // Try multiple possible selectors + this.descriptionInput = page.locator('[name="description"]').first(); + + // Alerts + this.successAlert = page.locator('[data-scope="toast"][data-part="root"]'); + + // Delete confirmation dialog + this.confirmDeleteButton = page.locator('button:has-text("Delete")').first(); + + // Pagination + this.paginationNextButton = page.locator('[data-testid="next"]'); + this.paginationPrevButton = page.locator('[data-testid="prev"]'); + this.rowsPerPageSelect = page.locator("select"); + + // Sorting and filtering + this.tableHeader = page.locator('[role="columnheader"]').first(); + this.connectionIdHeader = page.locator('[role="columnheader"]:has-text("Connection ID")').first(); + this.connectionTypeHeader = page.locator('[role="columnheader"]:has-text("Connection Type")').first(); + this.hostHeader = page.locator('[role="columnheader"]:has-text("Host")').first(); + this.searchInput = page.locator('input[placeholder*="Search"], input[placeholder*="search"]').first(); + } + + // Click the Add button to create a new connection + public async clickAddButton(): Promise<void> { + await expect(this.addButton).toBeVisible({ timeout: 5000 }); + await expect(this.addButton).toBeEnabled({ timeout: 5000 }); + await this.addButton.click(); + // Wait for form to load + await expect(this.connectionForm).toBeVisible({ timeout: 10_000 }); + } + + // Click edit button for a specific connection + public async clickEditButton(connectionId: string): Promise<void> { + const row = await this.findConnectionRow(connectionId); + + if (!row) { + throw new Error(`Connection ${connectionId} not found`); + } + + const editButton = row.getByRole("button", { name: "Edit Connection" }); + + await expect(editButton).toBeVisible({ timeout: 5000 }); + await expect(editButton).toBeEnabled({ timeout: 5000 }); + await editButton.click(); + await expect(this.connectionForm).toBeVisible({ timeout: 10_000 }); + } + + public async clickNextPage(): Promise<void> { + const isEnabled = await this.paginationNextButton.isEnabled({ timeout: 3000 }).catch(() => false); + + if (!isEnabled) { + return; + } + + const initialIds = await this.getConnectionIds(); + + // Set up API listener before action + const responsePromise = this.page + .waitForResponse((resp) => resp.url().includes("/connections") && resp.status() === 200, { + timeout: 10_000, + }) + .catch(() => { + /* API might be cached */ + }); + + await expect(this.paginationNextButton).toBeVisible({ timeout: 5000 }); + await this.paginationNextButton.click(); + + // Wait for API response + await responsePromise; + + // Wait for UI to actually change + await expect + .poll(() => this.getConnectionIds(), { + message: "List did not update after clicking next page", + timeout: 10_000, + }) + .not.toEqual(initialIds); + } + + public async clickPrevPage(): Promise<void> { + const isEnabled = await this.paginationPrevButton.isEnabled({ timeout: 3000 }).catch(() => false); + + if (!isEnabled) { + return; + } + + const initialIds = await this.getConnectionIds(); + + // Set up API listener before action + const responsePromise = this.page + .waitForResponse((resp) => resp.url().includes("/connections") && resp.status() === 200, { + timeout: 10_000, + }) + .catch(() => { + /* API might be cached */ + }); + + await expect(this.paginationPrevButton).toBeVisible({ timeout: 5000 }); + await this.paginationPrevButton.click(); + + // Wait for API response + await responsePromise; + + // Wait for UI to actually change + await expect + .poll(() => this.getConnectionIds(), { + message: "List did not update after clicking prev page", + timeout: 10_000, + }) + .not.toEqual(initialIds); + } + + // Check if a connection exists in the current view + public async connectionExists(connectionId: string): Promise<boolean> { + const emptyState = await this.page + .locator("text=No connection found!") + .isVisible({ timeout: 1000 }) + .catch(() => false); + + if (emptyState) { + return false; + } + const row = await this.findConnectionRow(connectionId); + const visible = row !== null; + + return visible; + } + + // Create a new connection with full workflow + public async createConnection(details: ConnectionDetails): Promise<void> { + await this.clickAddButton(); + await this.fillConnectionForm(details); + await this.saveConnection(); + await this.waitForConnectionsListLoad(); + } + + // Delete a connection by connection ID + public async deleteConnection(connectionId: string): Promise<void> { + // await this.navigate(); + const row = await this.findConnectionRow(connectionId); + + if (!row) { + throw new Error(`Connection ${connectionId} not found`); + } + + // Find delete button in the row + await this.page.evaluate(() => { + const backdrops = document.querySelectorAll<HTMLElement>('[data-scope="dialog"][data-part="backdrop"]'); + + backdrops.forEach((backdrop) => { + const { state } = backdrop.dataset; + + if (state === "closed") { + backdrop.remove(); + } + }); + }); + const deleteButton = row.getByRole("button", { name: "Delete Connection" }); + + await expect(deleteButton).toBeVisible({ timeout: 10_000 }); + await expect(deleteButton).toBeEnabled({ timeout: 5000 }); + await deleteButton.click(); + + await expect(this.confirmDeleteButton).toBeVisible({ timeout: 10_000 }); + await expect(this.confirmDeleteButton).toBeEnabled({ timeout: 5000 }); + await this.confirmDeleteButton.click(); + + await expect(this.emptyState).toBeVisible({ timeout: 5000 }); + } + + // Edit a connection by connection ID + public async editConnection(connectionId: string, updates: Partial<ConnectionDetails>): Promise<void> { + const row = await this.findConnectionRow(connectionId); + + if (!row) { + throw new Error(`Connection ${connectionId} not found`); + } + + await this.clickEditButton(connectionId); + + // Wait for form to load + await expect(this.connectionIdInput).toBeVisible({ timeout: 10_000 }); + + // Fill the fields that need updating + await this.fillConnectionForm(updates); + await this.saveConnection(); + } + + // Fill connection form with details + public async fillConnectionForm(details: Partial<ConnectionDetails>): Promise<void> { + if (details.connection_id !== undefined && details.connection_id !== "") { + await this.connectionIdInput.fill(details.connection_id); + } + + if (details.conn_type !== undefined && details.conn_type !== "") { + // Click the select field to open the dropdown + const selectCombobox = this.page.getByRole("combobox").first(); + + await expect(selectCombobox).toBeEnabled({ timeout: 10_000 }); + + await selectCombobox.click({ timeout: 3000 }).catch(() => { Review Comment: On initial load, this combobox takes time to load and shows loading state for about 10-20 seconds as observed from playwright test reports. <img width="1875" height="587" alt="Image" src="https://github.com/user-attachments/assets/93ec420d-6c56-4329-949e-99cb95874954"; /> So to give it ample time to load I added extra timeouts. But yes, The .catch() was unnecessary defensive code. I'll remove it. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
