SpringCloud/spring-cloud-gateway
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Polish gh-2721 updates test to verify attributes are passed

Browse Source
This commit is contained in:
sgibb
2024-03-08 16:11:23 -05:00
parent b9f7bcb002
commit 8c6cd4690a
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
spring-cloud-gateway-server/src/test/java/org/springframework/cloud/gateway/filter/ReactiveLoadBalancerClientFilterTests.java
View File

@@ -312,6 +312,7 @@ class ReactiveLoadBalancerClientFilterTests {
URI lbUri = URI.create("lb://service1?a=b");
ServerWebExchange serverWebExchange = mock(ServerWebExchange.class);
when(serverWebExchange.getAttribute(GATEWAY_REQUEST_URL_ATTR)).thenReturn(lbUri);
when(serverWebExchange.getAttributes()).thenReturn(new HashMap<>(Map.of("myattr", "myattrval")));
when(serverWebExchange.getRequiredAttribute(GATEWAY_ORIGINAL_REQUEST_URL_ATTR))
.thenReturn(new LinkedHashSet<>());
when(serverWebExchange.getRequest()).thenReturn(request);
@@ -323,9 +324,12 @@ class ReactiveLoadBalancerClientFilterTests {
filter.filter(serverWebExchange, chain);
verify(loadBalancer).choose(argThat((Request passedRequest) -> ((RequestDataContext) passedRequest.getContext())
.getClientRequest().getUrl().equals(request.getURI())
&& ((RequestDataContext) passedRequest.getContext()).getHint().equals(hint)));
verify(loadBalancer).choose(argThat((Request passedRequest) -> {
RequestDataContext context = (RequestDataContext) passedRequest.getContext();
return context.getClientRequest().getUrl().equals(request.getURI())
&& "myattrval".equals(context.getClientRequest().getAttributes().get("myattr"))
&& context.getHint().equals(hint);
}));
}