Handle upgrade screen

WordPress · Feb 1, 2021 · bd77454 · bd77454
1 parent d0deff5
commit bd77454
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/packages/e2e-test-utils/src/visit-admin-page.js b/packages/e2e-test-utils/src/visit-admin-page.js
@@ -20,6 +20,14 @@ import { getPageError } from './get-page-error';
 export async function visitAdminPage( adminPath, query ) {
 	await page.goto( createURL( join( 'wp-admin', adminPath ), query ) );
 
+	// Handle upgrade required screen
+	if ( isCurrentURL( 'wp-admin/upgrade.php' ) ) {
+		// Click update
+		await page.click( '.button.button-large.button-primary' );
+		// Click continue
+		await page.click( '.button.button-large' );
+	}
+
 	if ( isCurrentURL( 'wp-login.php' ) ) {
 		await loginUser();
 		await visitAdminPage( adminPath, query );