Polish
This commit is contained in:
@@ -132,27 +132,32 @@ public class JerseyWebEndpointIntegrationTests extends
|
||||
protected void doFilterInternal(HttpServletRequest request,
|
||||
HttpServletResponse response, FilterChain filterChain)
|
||||
throws ServletException, IOException {
|
||||
filterChain.doFilter(new HttpServletRequestWrapper(request) {
|
||||
|
||||
@Override
|
||||
public Principal getUserPrincipal() {
|
||||
|
||||
return new Principal() {
|
||||
|
||||
@Override
|
||||
public String getName() {
|
||||
return "Alice";
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
}, response);
|
||||
filterChain.doFilter(new MockPrincipalWrapper(request), response);
|
||||
}
|
||||
|
||||
};
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
private static class MockPrincipalWrapper extends HttpServletRequestWrapper {
|
||||
|
||||
public MockPrincipalWrapper(HttpServletRequest request) {
|
||||
super(request);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Principal getUserPrincipal() {
|
||||
return new MockPrincipal();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
private static class MockPrincipal implements Principal {
|
||||
|
||||
@Override
|
||||
public String getName() {
|
||||
return "Alice";
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -152,21 +152,8 @@ public class WebFluxEndpointIntegrationTests
|
||||
@Override
|
||||
public Mono<Void> filter(ServerWebExchange exchange,
|
||||
WebFilterChain chain) {
|
||||
return chain.filter(new ServerWebExchangeDecorator(exchange) {
|
||||
|
||||
@Override
|
||||
public Mono<Principal> getPrincipal() {
|
||||
return Mono.just(new Principal() {
|
||||
|
||||
@Override
|
||||
public String getName() {
|
||||
return "Alice";
|
||||
}
|
||||
|
||||
});
|
||||
}
|
||||
|
||||
});
|
||||
return chain.filter(
|
||||
new MockPrincipalServerWebExchangeDecorator(exchange));
|
||||
}
|
||||
|
||||
};
|
||||
@@ -174,4 +161,27 @@ public class WebFluxEndpointIntegrationTests
|
||||
|
||||
}
|
||||
|
||||
private static class MockPrincipalServerWebExchangeDecorator
|
||||
extends ServerWebExchangeDecorator {
|
||||
|
||||
protected MockPrincipalServerWebExchangeDecorator(ServerWebExchange delegate) {
|
||||
super(delegate);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Mono<Principal> getPrincipal() {
|
||||
return Mono.just(new MockPrincipal());
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
private static class MockPrincipal implements Principal {
|
||||
|
||||
@Override
|
||||
public String getName() {
|
||||
return "Alice";
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -146,27 +146,32 @@ public class MvcWebEndpointIntegrationTests extends
|
||||
protected void doFilterInternal(HttpServletRequest request,
|
||||
HttpServletResponse response, FilterChain filterChain)
|
||||
throws ServletException, IOException {
|
||||
filterChain.doFilter(new HttpServletRequestWrapper(request) {
|
||||
|
||||
@Override
|
||||
public Principal getUserPrincipal() {
|
||||
|
||||
return new Principal() {
|
||||
|
||||
@Override
|
||||
public String getName() {
|
||||
return "Alice";
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
}, response);
|
||||
filterChain.doFilter(new MockPrincipalWrapper(request), response);
|
||||
}
|
||||
|
||||
};
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
private static class MockPrincipalWrapper extends HttpServletRequestWrapper {
|
||||
|
||||
public MockPrincipalWrapper(HttpServletRequest request) {
|
||||
super(request);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Principal getUserPrincipal() {
|
||||
return new MockPrincipal();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
private static class MockPrincipal implements Principal {
|
||||
|
||||
@Override
|
||||
public String getName() {
|
||||
return "Alice";
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user