From 4b076d737e5dc001cbe6c3fcb8b0587724cb0b2b Mon Sep 17 00:00:00 2001 From: Marcus Da Coregio Date: Mon, 7 Nov 2022 08:13:50 -0300 Subject: [PATCH] Fix test failing on CI --- .../java/example/CustomUrlsApplicationITests.java | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/servlet/spring-boot/java/saml2/custom-urls/src/integTest/java/example/CustomUrlsApplicationITests.java b/servlet/spring-boot/java/saml2/custom-urls/src/integTest/java/example/CustomUrlsApplicationITests.java index 4661be3..a87ad3c 100644 --- a/servlet/spring-boot/java/saml2/custom-urls/src/integTest/java/example/CustomUrlsApplicationITests.java +++ b/servlet/spring-boot/java/saml2/custom-urls/src/integTest/java/example/CustomUrlsApplicationITests.java @@ -16,6 +16,9 @@ package example; +import java.util.ArrayList; +import java.util.List; + import com.gargoylesoftware.htmlunit.ElementNotFoundException; import com.gargoylesoftware.htmlunit.WebClient; import com.gargoylesoftware.htmlunit.html.HtmlElement; @@ -66,7 +69,10 @@ public class CustomUrlsApplicationITests { HtmlElement rpLogoutButton = home.getHtmlElementById("rp_logout_button"); HtmlPage loginPage = rpLogoutButton.click(); this.webClient.waitForBackgroundJavaScript(10000); - assertThat(loginPage.getUrl().getFile()).isEqualTo("/login?logout"); + List urls = new ArrayList<>(); + urls.add(loginPage.getUrl().getFile()); + urls.add(((HtmlPage) this.webClient.getCurrentWindow().getEnclosedPage()).getUrl().getFile()); + assertThat(urls).contains("/login?logout"); } @Test