chore(#8244): Fix upgrade e2e test when pushing a beta (#8250)

Cherry-picked from 536e72f
medic · May 17, 2023 · 618f0e1 · 618f0e1
1 parent 5aa9464
commit 618f0e1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/page-objects/upgrade/upgrade.wdio.page.js b/tests/page-objects/upgrade/upgrade.wdio.page.js
@@ -6,7 +6,11 @@ const deploymentInProgress = () => $('legend*=Deployment in progress');
 const deploymentComplete = () => $('div*=Deployment complete');
 
 const getInstallButton = async (branch, tag) => {
-  const element = tag ? await $(`span=${tag}`) : await $(`span*=${branch} (`);
+  let selector = `span*=${branch} (`;
+  if (tag) {
+    selector = tag.includes('beta') ? `span*=${tag}` : `span=${tag}`;
+  }
+  const element = await $(selector);
   const parent = await (await element.parentElement()).parentElement();
   return await parent.$('.btn-primary');
 };