Skip to content

Commit 5b1710f

Browse files
committed
Polish
Co-authored-by: ghaege <ghaege@qaepps.de> Issue gh-12538
1 parent fa9c7fb commit 5b1710f

File tree

1 file changed

+27
-15
lines changed

1 file changed

+27
-15
lines changed

saml2/saml2-service-provider/src/test/java/org/springframework/security/saml2/provider/service/web/authentication/logout/OpenSaml4LogoutRequestResolverTests.java

Lines changed: 27 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616

1717
package org.springframework.security.saml2.provider.service.web.authentication.logout;
1818

19-
import jakarta.servlet.http.HttpServletRequest;
19+
import org.junit.jupiter.api.BeforeEach;
2020
import org.junit.jupiter.api.Test;
2121

2222
import org.springframework.mock.web.MockHttpServletRequest;
@@ -38,29 +38,41 @@
3838
*/
3939
public class OpenSaml4LogoutRequestResolverTests {
4040

41-
RelyingPartyRegistrationResolver relyingPartyRegistrationResolver = mock(RelyingPartyRegistrationResolver.class);
41+
RelyingPartyRegistration registration;
42+
43+
RelyingPartyRegistrationResolver registrationResolver;
44+
45+
OpenSaml4LogoutRequestResolver logoutRequestResolver;
46+
47+
@BeforeEach
48+
public void setup() {
49+
this.registration = TestRelyingPartyRegistrations.full().build();
50+
this.registrationResolver = mock(RelyingPartyRegistrationResolver.class);
51+
this.logoutRequestResolver = new OpenSaml4LogoutRequestResolver(this.registrationResolver);
52+
}
4253

4354
@Test
4455
public void resolveWhenCustomParametersConsumerThenUses() {
45-
OpenSaml4LogoutRequestResolver logoutRequestResolver = new OpenSaml4LogoutRequestResolver(
46-
this.relyingPartyRegistrationResolver);
47-
logoutRequestResolver.setParametersConsumer((parameters) -> parameters.getLogoutRequest().setID("myid"));
48-
HttpServletRequest request = new MockHttpServletRequest();
49-
RelyingPartyRegistration registration = TestRelyingPartyRegistrations.relyingPartyRegistration()
50-
.assertingPartyDetails((party) -> party.singleLogoutServiceLocation("https://ap.example.com/logout"))
51-
.build();
52-
Authentication authentication = new TestingAuthenticationToken("user", "password");
53-
given(this.relyingPartyRegistrationResolver.resolve(any(), any())).willReturn(registration);
54-
Saml2LogoutRequest logoutRequest = logoutRequestResolver.resolve(request, authentication);
56+
this.logoutRequestResolver.setParametersConsumer((parameters) -> parameters.getLogoutRequest().setID("myid"));
57+
given(this.registrationResolver.resolve(any(), any())).willReturn(this.registration);
58+
59+
Saml2LogoutRequest logoutRequest = this.logoutRequestResolver.resolve(givenRequest(), givenAuthentication());
60+
5561
assertThat(logoutRequest.getId()).isEqualTo("myid");
5662
}
5763

5864
@Test
5965
public void setParametersConsumerWhenNullThenIllegalArgument() {
60-
OpenSaml4LogoutRequestResolver logoutRequestResolver = new OpenSaml4LogoutRequestResolver(
61-
this.relyingPartyRegistrationResolver);
6266
assertThatExceptionOfType(IllegalArgumentException.class)
63-
.isThrownBy(() -> logoutRequestResolver.setParametersConsumer(null));
67+
.isThrownBy(() -> this.logoutRequestResolver.setParametersConsumer(null));
68+
}
69+
70+
private static Authentication givenAuthentication() {
71+
return new TestingAuthenticationToken("user", "password");
72+
}
73+
74+
private MockHttpServletRequest givenRequest() {
75+
return new MockHttpServletRequest();
6476
}
6577

6678
}

0 commit comments

Comments
 (0)