/* * Copyright 2002-2018 the original author or authors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.springframework.mock.web.test.server; import java.time.Clock; import java.time.Duration; import java.time.Instant; import java.util.Map; import reactor.core.publisher.Mono; import org.springframework.lang.Nullable; import org.springframework.util.Assert; import org.springframework.web.server.WebSession; import org.springframework.web.server.session.InMemoryWebSessionStore; /** * Implementation of {@code WebSession} that delegates to a session instance * obtained via {@link InMemoryWebSessionStore}. * * <p>This is intended for use with the * {@link MockServerWebExchange.Builder#session(WebSession) session(WebSession)} * method of the {@code MockServerWebExchange} builder, eliminating the need * to use {@code WebSessionManager} or {@code WebSessionStore} altogether. * * @author Rossen Stoyanchev * @since 5.1 */ public class MockWebSession implements WebSession { private final WebSession delegate; public MockWebSession() { this(null); } public MockWebSession(@Nullable Clock clock) { InMemoryWebSessionStore sessionStore = new InMemoryWebSessionStore(); if (clock != null) { sessionStore.setClock(clock); } WebSession session = sessionStore.createWebSession().block(); Assert.state(session != null, "WebSession must not be null"); this.delegate = session; } @Override public String getId() { return this.delegate.getId(); } @Override public Map<String, Object> getAttributes() { return this.delegate.getAttributes(); } @Override public void start() { this.delegate.start(); } @Override public boolean isStarted() { return this.delegate.isStarted(); } @Override public Mono<Void> changeSessionId() { return this.delegate.changeSessionId(); } @Override public Mono<Void> invalidate() { return this.delegate.invalidate(); } @Override public Mono<Void> save() { return this.delegate.save(); } @Override public boolean isExpired() { return this.delegate.isExpired(); } @Override public Instant getCreationTime() { return this.delegate.getCreationTime(); } @Override public Instant getLastAccessTime() { return this.delegate.getLastAccessTime(); } @Override public void setMaxIdleTime(Duration maxIdleTime) { this.delegate.setMaxIdleTime(maxIdleTime); } @Override public Duration getMaxIdleTime() { return this.delegate.getMaxIdleTime(); } }