org.springframework.mock.web.server.MockWebSession Maven / Gradle / Ivy
Show all versions of spring-test Show documentation
/*
* 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
*
* https://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.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}.
*
* 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 getAttributes() {
return this.delegate.getAttributes();
}
@Override
public void start() {
this.delegate.start();
}
@Override
public boolean isStarted() {
return this.delegate.isStarted();
}
@Override
public Mono changeSessionId() {
return this.delegate.changeSessionId();
}
@Override
public Mono invalidate() {
return this.delegate.invalidate();
}
@Override
public Mono 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();
}
}