Updated tests for framework changes

Author: spencergibb

spring-cloud-gateway-server/src/test/java/org/springframework/cloud/gateway/filter/factory/TokenRelayGatewayFilterFactoryTests.java

@@ -121,7 +121,11 @@ public void whenPrincipalExistsAuthorizationHeaderAdded() {
 		GatewayFilter filter = new TokenRelayGatewayFilterFactory(objectProvider).apply();
 		filter.filter(exchange, filterChain).block(TIMEOUT);
 
-		assertThat(request.getHeaders()).containsEntry(HttpHeaders.AUTHORIZATION,
+		ArgumentCaptor<ServerWebExchange> sweCaptor = ArgumentCaptor.forClass(ServerWebExchange.class);
+		verify(filterChain).filter(sweCaptor.capture());
+		ServerWebExchange mutatedServerWebExchange = sweCaptor.getValue();
+
+		assertThat(mutatedServerWebExchange.getRequest().getHeaders()).containsEntry(HttpHeaders.AUTHORIZATION,
 				Collections.singletonList("Bearer mytoken"));
 
 		ArgumentCaptor<OAuth2AuthorizeRequest> authorizeRequestCaptor = ArgumentCaptor
@@ -161,7 +165,11 @@ public void whenClientRegistrationIdConfiguredAuthorizationHeaderAdded() {
 		GatewayFilter filter = new TokenRelayGatewayFilterFactory(objectProvider).apply(config);
 		filter.filter(exchange, filterChain).block(TIMEOUT);
 
-		assertThat(request.getHeaders()).containsEntry(HttpHeaders.AUTHORIZATION,
+		ArgumentCaptor<ServerWebExchange> sweCaptor = ArgumentCaptor.forClass(ServerWebExchange.class);
+		verify(filterChain).filter(sweCaptor.capture());
+		ServerWebExchange mutatedServerWebExchange = sweCaptor.getValue();
+
+		assertThat(mutatedServerWebExchange.getRequest().getHeaders()).containsEntry(HttpHeaders.AUTHORIZATION,
 				Collections.singletonList("Bearer mytoken"));
 
 		ArgumentCaptor<OAuth2AuthorizeRequest> authorizeRequestCaptor = ArgumentCaptor
@@ -212,7 +220,10 @@ public void whenPrincipalIsNotOAuth2AuthenticationTokenAndClientRegistrationIdCo
 		GatewayFilter filter = new TokenRelayGatewayFilterFactory(objectProvider).apply(config);
 		filter.filter(exchange, filterChain).block(TIMEOUT);
 
-		assertThat(request.getHeaders()).containsEntry(HttpHeaders.AUTHORIZATION,
+		ArgumentCaptor<ServerWebExchange> sweCaptor = ArgumentCaptor.forClass(ServerWebExchange.class);
+		verify(filterChain).filter(sweCaptor.capture());
+		ServerWebExchange mutatedServerWebExchange = sweCaptor.getValue();
+		assertThat(mutatedServerWebExchange.getRequest().getHeaders()).containsEntry(HttpHeaders.AUTHORIZATION,
 				Collections.singletonList("Bearer mytoken"));
 
 		ArgumentCaptor<OAuth2AuthorizeRequest> authorizeRequestCaptor = ArgumentCaptor

spring-cloud-gateway-server/src/test/java/org/springframework/cloud/gateway/filter/factory/rewrite/ModifyRequestBodyGatewayFilterFactoryTests.java

@@ -86,10 +86,10 @@ public void modifyRequestBodyToLarge() {
 			.body(BodyInserters.fromValue("request"))
 			.exchange()
 			.expectStatus()
-			.isEqualTo(HttpStatus.INTERNAL_SERVER_ERROR)
+			.isEqualTo(HttpStatus.PAYLOAD_TOO_LARGE)
 			.expectBody()
-			.jsonPath("message")
-			.isEqualTo("Exceeded limit on max bytes to buffer : 13");
+			.jsonPath("error")
+			.isEqualTo("Payload Too Large");
 	}
 
 	@Test

spring-cloud-gateway-server/src/test/java/org/springframework/cloud/gateway/filter/factory/rewrite/ModifyResponseBodyGatewayFilterFactoryTests.java

@@ -87,18 +87,18 @@ public void testModificationOfContentType() {
 	}
 
 	@Test
-	public void modifyResponeBodyToLarge() {
+	public void modifyResponseBodyToLarge() {
 		testClient.post()
 			.uri("/post")
 			.header("Host", "www.modifyresponsebodyjavatoolarge.org")
 			.header(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_VALUE)
 			.body(BodyInserters.fromValue(toLarge))
 			.exchange()
 			.expectStatus()
-			.isEqualTo(HttpStatus.INTERNAL_SERVER_ERROR)
+			.isEqualTo(HttpStatus.PAYLOAD_TOO_LARGE)
 			.expectBody()
-			.jsonPath("message")
-			.isEqualTo("Exceeded limit on max bytes to buffer : 40");
+			.jsonPath("error")
+			.isEqualTo("Payload Too Large");
 	}
 
 	@EnableAutoConfiguration