@Override public boolean supportsParameter(MethodParameter param) { return checkAnnotatedParamNoReactiveWrapper(param, CookieValue.class, (annot, type) -> true); }
@Test public void supportsParameter() { assertTrue(this.resolver.supportsParameter(this.cookieParameter)); assertTrue(this.resolver.supportsParameter(this.cookieStringParameter)); }
result.add(new CookieValueMethodArgumentResolver(beanFactory, reactiveRegistry)); result.add(new ExpressionValueMethodArgumentResolver(beanFactory, reactiveRegistry)); result.add(new SessionAttributeMethodArgumentResolver(beanFactory, reactiveRegistry));
@Test public void resolveCookieDefaultValue() { MockServerWebExchange exchange = MockServerWebExchange.from(MockServerHttpRequest.get("/")); Object result = this.resolver.resolveArgument(this.cookieStringParameter, this.bindingContext, exchange).block(); assertTrue(result instanceof String); assertEquals("bar", result); }
@Before @SuppressWarnings("resource") public void setup() throws Exception { AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext(); context.refresh(); ReactiveAdapterRegistry adapterRegistry = ReactiveAdapterRegistry.getSharedInstance(); this.resolver = new CookieValueMethodArgumentResolver(context.getBeanFactory(), adapterRegistry); this.bindingContext = new BindingContext(); Method method = ReflectionUtils.findMethod(getClass(), "params", (Class<?>[]) null); this.cookieParameter = new SynthesizingMethodParameter(method, 0); this.cookieStringParameter = new SynthesizingMethodParameter(method, 1); this.stringParameter = new SynthesizingMethodParameter(method, 2); this.cookieMonoParameter = new SynthesizingMethodParameter(method, 3); }
@Test public void resolveCookieArgument() { HttpCookie expected = new HttpCookie("name", "foo"); MockServerWebExchange exchange = MockServerWebExchange.from(MockServerHttpRequest.get("/").cookie(expected)); Mono<Object> mono = this.resolver.resolveArgument( this.cookieParameter, this.bindingContext, exchange); assertEquals(expected, mono.block()); }
protected List<SyncHandlerMethodArgumentResolver> getDefaultInitBinderArgumentResolvers() { List<SyncHandlerMethodArgumentResolver> resolvers = new ArrayList<>(); // Annotation-based argument resolution resolvers.add(new RequestParamMethodArgumentResolver(getBeanFactory(), false)); resolvers.add(new RequestParamMapMethodArgumentResolver()); resolvers.add(new PathVariableMethodArgumentResolver(getBeanFactory())); resolvers.add(new PathVariableMapMethodArgumentResolver()); resolvers.add(new RequestHeaderMethodArgumentResolver(getBeanFactory())); resolvers.add(new RequestHeaderMapMethodArgumentResolver()); resolvers.add(new CookieValueMethodArgumentResolver(getBeanFactory())); resolvers.add(new ExpressionValueMethodArgumentResolver(getBeanFactory())); resolvers.add(new RequestAttributeMethodArgumentResolver(getBeanFactory())); // Type-based argument resolution resolvers.add(new ModelArgumentResolver()); resolvers.add(new ServerWebExchangeArgumentResolver()); // Custom resolvers if (getCustomInitBinderArgumentResolvers() != null) { resolvers.addAll(getCustomInitBinderArgumentResolvers()); } // Catch-all resolvers.add(new RequestParamMethodArgumentResolver(getBeanFactory(), true)); return resolvers; }
@Test public void resolveCookieStringArgument() { HttpCookie cookie = new HttpCookie("name", "foo"); MockServerWebExchange exchange = MockServerWebExchange.from(MockServerHttpRequest.get("/").cookie(cookie)); Mono<Object> mono = this.resolver.resolveArgument( this.cookieStringParameter, this.bindingContext, exchange); assertEquals("Invalid result", cookie.getValue(), mono.block()); }
@Test public void doesNotSupportParameter() { assertFalse(this.resolver.supportsParameter(this.stringParameter)); try { this.resolver.supportsParameter(this.cookieMonoParameter); fail(); } catch (IllegalStateException ex) { assertTrue("Unexpected error message:\n" + ex.getMessage(), ex.getMessage().startsWith( "CookieValueMethodArgumentResolver doesn't support reactive type wrapper")); } }
resolvers.add(new RequestHeaderMethodArgumentResolver(getBeanFactory())); resolvers.add(new RequestHeaderMapMethodArgumentResolver()); resolvers.add(new CookieValueMethodArgumentResolver(getBeanFactory())); resolvers.add(new ExpressionValueMethodArgumentResolver(getBeanFactory())); resolvers.add(new SessionAttributeMethodArgumentResolver(getBeanFactory()));
@Test public void notFound() { MockServerWebExchange exchange = MockServerWebExchange.from(MockServerHttpRequest.get("/")); Mono<Object> mono = resolver.resolveArgument(this.cookieParameter, this.bindingContext, exchange); StepVerifier.create(mono) .expectNextCount(0) .expectError(ServerWebInputException.class) .verify(); }