org.springframework.web.server.session.DefaultWebSessionManager Java Examples

@Test // gh-22754
public void subscribeWithoutDemand() throws Exception {
	ZeroDemandResponse response = new ZeroDemandResponse();
	ServerWebExchange exchange = new DefaultServerWebExchange(
			MockServerHttpRequest.get("/path").build(), response,
			new DefaultWebSessionManager(), ServerCodecConfigurer.create(),
			new AcceptHeaderLocaleContextResolver());

	Map<String, Object> model = new HashMap<>();
	model.put("title", "Layout example");
	model.put("body", "This is the body");
	String viewUrl = "org/springframework/web/reactive/result/view/script/nashorn/template.html";
	ScriptTemplateView view = createViewWithUrl(viewUrl, ScriptTemplatingConfiguration.class);
	view.render(model, null, exchange).subscribe();

	response.cancelWrite();
	response.checkForLeaks();
}
 

@Test // gh-22754
public void subscribeWithoutDemand() {
	ZeroDemandResponse response = new ZeroDemandResponse();
	ServerWebExchange exchange = new DefaultServerWebExchange(
			MockServerHttpRequest.get("/path").build(), response,
			new DefaultWebSessionManager(), ServerCodecConfigurer.create(),
			new AcceptHeaderLocaleContextResolver());

	FreeMarkerView view = new FreeMarkerView();
	view.setConfiguration(this.freeMarkerConfig);
	view.setUrl("test.ftl");

	ModelMap model = new ExtendedModelMap();
	model.addAttribute("hello", "hi FreeMarker");
	view.render(model, null, exchange).subscribe();

	response.cancelWrite();
	response.checkForLeaks();
}
 

@Before
public void setup() {
	MockServerHttpRequest request = MockServerHttpRequest.post("/path").build();
	MockServerHttpResponse response = new MockServerHttpResponse();
	DefaultWebSessionManager sessionManager = new DefaultWebSessionManager();
	exchange = new DefaultServerWebExchange(request, response, sessionManager);
	pattern = "/pattern";
	path = "/path";

	matcher = new PathMatcherServerWebExchangeMatcher(pattern);
	matcher.setPathMatcher(pathMatcher);
}
 

/**
 * Configure a {@link WebSessionManager} using a provided
 * {@link ReactiveSessionRepository}.
 * @param repository a bean that implements {@link ReactiveSessionRepository}.
 * @return a configured {@link WebSessionManager} registered with a preconfigured
 * name.
 */
@Bean(WebHttpHandlerBuilder.WEB_SESSION_MANAGER_BEAN_NAME)
public WebSessionManager webSessionManager(ReactiveSessionRepository<? extends Session> repository) {
	SpringSessionWebSessionStore<? extends Session> sessionStore = new SpringSessionWebSessionStore<>(repository);
	DefaultWebSessionManager manager = new DefaultWebSessionManager();
	manager.setSessionStore(sessionStore);

	if (this.webSessionIdResolver != null) {
		manager.setSessionIdResolver(this.webSessionIdResolver);
	}

	return manager;
}
 

@Test
void defaultSessionIdResolverShouldBeCookieBased() {

	this.context = new AnnotationConfigApplicationContext();
	this.context.register(GoodConfig.class);
	this.context.refresh();

	DefaultWebSessionManager manager = this.context.getBean(DefaultWebSessionManager.class);
	assertThat(manager.getSessionIdResolver().getClass()).isAssignableFrom(CookieWebSessionIdResolver.class);
}
 

@Test
void providedSessionIdResolverShouldBePickedUpAutomatically() {

	this.context = new AnnotationConfigApplicationContext();
	this.context.register(OverrideSessionIdResolver.class);
	this.context.refresh();

	DefaultWebSessionManager manager = this.context.getBean(DefaultWebSessionManager.class);
	assertThat(manager.getSessionIdResolver().getClass()).isAssignableFrom(HeaderWebSessionIdResolver.class);
}
 

private DefaultServerWebExchange createExchange(MockServerHttpRequest request) {
	return new DefaultServerWebExchange(request, new MockServerHttpResponse(),
			new DefaultWebSessionManager(), ServerCodecConfigurer.create(),
			new AcceptHeaderLocaleContextResolver());
}
 

/**
 * Build the {@code MockServerWebExchange} instance.
 */
public MockServerWebExchange build() {
	return new MockServerWebExchange(this.request,
			this.sessionManager != null ? this.sessionManager : new DefaultWebSessionManager());
}
 

/**
 * Build the {@code MockServerWebExchange} instance.
 */
public MockServerWebExchange build() {
	return new MockServerWebExchange(this.request,
			this.sessionManager != null ? this.sessionManager : new DefaultWebSessionManager());
}
 

AbstractMockServerSpec() {
	// Default instance to be re-used across requests, unless one is configured explicitly
	this.sessionManager = new DefaultWebSessionManager();
}
 

private DefaultServerWebExchange createExchange(MockServerHttpRequest request) {
	return new DefaultServerWebExchange(request, new MockServerHttpResponse(),
			new DefaultWebSessionManager(), ServerCodecConfigurer.create(),
			new AcceptHeaderLocaleContextResolver());
}
 

/**
 * Build the {@code MockServerWebExchange} instance.
 */
public MockServerWebExchange build() {
	return new MockServerWebExchange(this.request,
			this.sessionManager != null ? this.sessionManager : new DefaultWebSessionManager());
}
 

/**
 * Build the {@code MockServerWebExchange} instance.
 */
public MockServerWebExchange build() {
	return new MockServerWebExchange(this.request,
			this.sessionManager != null ? this.sessionManager : new DefaultWebSessionManager());
}
 

AbstractMockServerSpec() {
	// Default instance to be re-used across requests, unless one is configured explicitly
	this.sessionManager = new DefaultWebSessionManager();
}
 

@Before
public void beforeMethod() {
    resetTracing();

    spanRecorder = new SpanRecorder();
    Tracer.getInstance().addSpanLifecycleListener(spanRecorder);

    initialSpanNameFromStrategy = new AtomicReference<>("span-name-from-strategy-" + UUID.randomUUID().toString());
    strategyInitialSpanNameMethodCalled = new AtomicBoolean(false);
    strategyRequestTaggingMethodCalled = new AtomicBoolean(false);
    strategyResponseTaggingAndFinalSpanNameMethodCalled = new AtomicBoolean(false);
    strategyInitialSpanNameArgs = new AtomicReference<>(null);
    strategyRequestTaggingArgs = new AtomicReference<>(null);
    strategyResponseTaggingArgs = new AtomicReference<>(null);
    tagAndNamingStrategy = new ArgCapturingHttpTagAndSpanNamingStrategy<>(
        initialSpanNameFromStrategy, strategyInitialSpanNameMethodCalled, strategyRequestTaggingMethodCalled,
        strategyResponseTaggingAndFinalSpanNameMethodCalled, strategyInitialSpanNameArgs,
        strategyRequestTaggingArgs, strategyResponseTaggingArgs
    );
    //noinspection unchecked
    tagAndNamingAdapterMock = mock(HttpTagAndSpanNamingAdapter.class);

    userIdHeaderKeys = Arrays.asList("user-id", "alt-user-id");

    filterSpy = spy(
        WingtipsSpringWebfluxWebFilter
            .newBuilder()
            .withTagAndNamingStrategy(tagAndNamingStrategy)
            .withTagAndNamingAdapter(tagAndNamingAdapterMock)
            .withUserIdHeaderKeys(userIdHeaderKeys)
            .build()
    );

    requestMock = mock(ServerHttpRequest.class);
    requestHeadersMock = mock(HttpHeaders.class);
    responseMock = mock(ServerHttpResponse.class);
    responseHeadersMock = mock(HttpHeaders.class);

    doReturn("someRequestId").when(requestMock).getId();

    doReturn(requestHeadersMock).when(requestMock).getHeaders();
    doReturn(responseHeadersMock).when(responseMock).getHeaders();

    exchange = new DefaultServerWebExchange(
        requestMock, responseMock, new DefaultWebSessionManager(), new DefaultServerCodecConfigurer(),
        new FixedLocaleContextResolver(Locale.US)
    );
    
    chainMock = mock(WebFilterChain.class);
    expectedResultMonoDuration = Duration.ofMillis(50);
    webFilterChainResult = Mono.delay(expectedResultMonoDuration).flatMap(l -> Mono.empty());

    doReturn(webFilterChainResult).when(chainMock).filter(any(ServerWebExchange.class));
}