소스 검색

Apply OpenSAML Test Fix to Remaining Test

There was one test that wasn't converted and now it is.

Issue gh-15395
Josh Cummings 11 달 전
부모
커밋
c9d083717f

+ 6 - 13
saml2/saml2-service-provider/src/opensaml4Test/java/org/springframework/security/saml2/provider/service/registration/OpenSaml4AssertingPartyMetadataRepositoryTests.java

@@ -261,13 +261,12 @@ public class OpenSaml4AssertingPartyMetadataRepositoryTests {
 		TestOpenSamlObjects.signed(descriptor, TestSaml2X509Credentials.assertingPartySigningCredential(),
 				descriptor.getEntityID());
 		String serialized = serialize(descriptor);
-		try (MockWebServer server = new MockWebServer()) {
-			enqueue(server, serialized, 3);
-			AssertingPartyMetadataRepository parties = OpenSaml4AssertingPartyMetadataRepository
-				.withTrustedMetadataLocation(server.url("/").toString())
-				.build();
-			assertThat(parties.findByEntityId(registration.getAssertingPartyDetails().getEntityId())).isNotNull();
-		}
+		String endpoint = "/" + UUID.randomUUID().toString();
+		dispatcher.addResponse(endpoint, serialized);
+		AssertingPartyMetadataRepository parties = OpenSaml4AssertingPartyMetadataRepository
+			.withTrustedMetadataLocation(web.url(endpoint).toString())
+			.build();
+		assertThat(parties.findByEntityId(registration.getAssertingPartyDetails().getEntityId())).isNotNull();
 	}
 
 	@Test
@@ -362,12 +361,6 @@ public class OpenSaml4AssertingPartyMetadataRepositoryTests {
 		}
 	}
 
-	private static void enqueue(MockWebServer web, String body, int times) {
-		for (int i = 0; i < times; i++) {
-			web.enqueue(new MockResponse().setBody(body).setResponseCode(200));
-		}
-	}
-
 	private static final class MetadataDispatcher extends Dispatcher {
 
 		private final MockResponse head = new MockResponse();

+ 6 - 13
saml2/saml2-service-provider/src/opensaml5Test/java/org/springframework/security/saml2/provider/service/registration/OpenSaml5AssertingPartyMetadataRepositoryTests.java

@@ -261,13 +261,12 @@ public class OpenSaml5AssertingPartyMetadataRepositoryTests {
 		TestOpenSamlObjects.signed(descriptor, TestSaml2X509Credentials.assertingPartySigningCredential(),
 				descriptor.getEntityID());
 		String serialized = serialize(descriptor);
-		try (MockWebServer server = new MockWebServer()) {
-			enqueue(server, serialized, 3);
-			AssertingPartyMetadataRepository parties = OpenSaml5AssertingPartyMetadataRepository
-				.withTrustedMetadataLocation(server.url("/").toString())
-				.build();
-			assertThat(parties.findByEntityId(registration.getAssertingPartyDetails().getEntityId())).isNotNull();
-		}
+		String endpoint = "/" + UUID.randomUUID().toString();
+		dispatcher.addResponse(endpoint, serialized);
+		AssertingPartyMetadataRepository parties = OpenSaml5AssertingPartyMetadataRepository
+			.withTrustedMetadataLocation(web.url(endpoint).toString())
+			.build();
+		assertThat(parties.findByEntityId(registration.getAssertingPartyDetails().getEntityId())).isNotNull();
 	}
 
 	@Test
@@ -362,12 +361,6 @@ public class OpenSaml5AssertingPartyMetadataRepositoryTests {
 		}
 	}
 
-	private static void enqueue(MockWebServer web, String body, int times) {
-		for (int i = 0; i < times; i++) {
-			web.enqueue(new MockResponse().setBody(body).setResponseCode(200));
-		}
-	}
-
 	private static final class MetadataDispatcher extends Dispatcher {
 
 		private final MockResponse head = new MockResponse();