diff --git a/test/nbrowser/homeUtil.ts b/test/nbrowser/homeUtil.ts index 2d0369c9..7b1bbcb1 100644 --- a/test/nbrowser/homeUtil.ts +++ b/test/nbrowser/homeUtil.ts @@ -253,10 +253,11 @@ export class HomeUtil { } /** - * Returns whether we are currently on the Cognito login page. + * Returns whether we are currently on the Cognito or test login page. */ public async isOnLoginPage() { - return /^https:\/\/gristlogin/.test(await this.driver.getCurrentUrl()); + const url = await this.driver.getCurrentUrl(); + return /^https:\/\/gristlogin/.test(url) || await this.isOnTestLoginPage(); } /**