PMM-13329 add wait for upgrade to button (#842)

* PMM-13329 add wait for upgrade to button

* PMM-13329 add wait for upgrade to button

Author: yurkovychv

playwright-tests/pages/home-dashboard.page.ts

@@ -25,7 +25,8 @@ export default class HomeDashboardPage extends BaseDashboard {
     await expect(this.page).toHaveURL(this.PAGE_PATH);
   }
   upgradePmm = async () => {
-    await this.pmmUpgradeWidget.elements.upgradeButton.waitFor({ state: 'visible', timeout: Wait.ThreeMinutes });
+    await this.pmmUpgradeWidget.elements.upgradeButton.waitFor({ state: 'visible', timeout: Wait.TwoMinutes });
+
     const currentVersion = await this.pmmUpgradeWidget.elements.currentVersion.textContent();
 
     await this.pmmUpgradeWidget.elements.upgradeButton.click();

playwright-tests/pages/page-components/pmm-upgrade-widget.ts

@@ -9,12 +9,20 @@ export default class PmmUpgradeWidget {
   elements: any = {
     currentVersion: this.containers.upgradeContainer.getByTestId('update-installed-version'),
     availableVersion: this.containers.upgradeContainer.getByTestId('update-latest-version'),
+    checkForUpgradesButton: this.containers.upgradeContainer.getByTestId('update-last-check-button'),
     upToDate: this.containers.upgradeContainer.getByText('You are up to date'),
     lastUpgradeCheckDate: this.containers.upgradeContainer.getByTestId('update-last-check'),
     upgradeButton: this.containers.upgradeContainer.getByText('Upgrade to', { exact: false }),
   };
 
   verifyUpgradeWidget = async () => {
+    await expect(this.elements.checkForUpgradesButton).toBeEnabled({ timeout: Wait.ThreeMinutes });
+
+    await expect(async () => {
+      await this.elements.checkForUpgradesButton.click();
+      await this.elements.upgradeButton.waitFor({ state: 'visible', timeout: Wait.TwoMinutes });
+    }).toPass({ timeout: Wait.TenMinutes });
+
     await this.elements.upgradeButton.waitFor({ state: 'visible', timeout: Wait.ThreeMinutes });
     await expect(this.elements.upToDate).toBeHidden();
     await this.elements.lastUpgradeCheckDate.waitFor({ state: 'visible' });