try grabbing text and comparing expected error

uc-cdis · Jun 27, 2024 · c10be88 · c10be88
1 parent f80e1ea
commit c10be88
Showing 1 changed file with 2 additions and 9 deletions.
diff --git a/services/apis/fence/fenceTasks.js b/services/apis/fence/fenceTasks.js
@@ -515,15 +515,8 @@ module.exports = {
         }
       // }
     } else {
-      try {
-        I.seeTextEquals('Bad Request', 'h2');
-      } catch (err) {
-        try {
-          I.seeTextEquals('Unauthorized', 'h2');
-        } catch (err) {
-          console.log('Expected to see Bad Request or Unauthorized on the page');
-        }
-      }
+      let errorMessage = await I.grabTextFrom('//div[@class="error-page__status-code-text"]/h2');
+      expect(errorMessage).to.be.oneOf(["Bad Request", "Unauthorized"]);
     }
     // I.saveScreenshot('consent_auth_code_flow.png');
     const urlStr = await I.grabCurrentUrl();