update click for edit button

woocommerce · Oct 10, 2023 · b6b67c4 · b6b67c4
1 parent a15f117
commit b6b67c4
Showing 1 changed file with 10 additions and 22 deletions.
diff --git a/tests/e2e/tests/checkout/checkout.page.ts b/tests/e2e/tests/checkout/checkout.page.ts
@@ -126,34 +126,22 @@ export class CheckoutPage {
 	}
 
 	async editBillingDetails() {
-		const editButton = await this.page
-			.locator(
-				'.wc-block-checkout__billing-fields .wc-block-components-address-card__edit'
-			)
-			.isVisible();
+		const editButton = this.page.locator(
+			'.wc-block-checkout__billing-fields .wc-block-components-address-card__edit'
+		);
 
-		if ( editButton ) {
-			await this.page
-				.locator(
-					'.wc-block-checkout__billing-fields .wc-block-components-address-card__edit'
-				)
-				.click();
+		if ( await editButton.isVisible() ) {
+			await editButton.click();
 		}
 	}
 
 	async editShippingDetails() {
-		const editButton = await this.page
-			.locator(
-				'.wc-block-checkout__shipping-fields .wc-block-components-address-card__edit'
-			)
-			.isVisible();
+		const editButton = this.page.locator(
+			'.wc-block-checkout__shipping-fields .wc-block-components-address-card__edit'
+		);
 
-		if ( editButton ) {
-			await this.page
-				.locator(
-					'.wc-block-checkout__shipping-fields .wc-block-components-address-card__edit'
-				)
-				.click();
+		if ( await editButton.isVisible() ) {
+			await editButton.click();
 		}
 	}