spring RedirectViewTests 源码
spring RedirectViewTests 代码
文件路径：/spring-webmvc/src/test/java/org/springframework/web/servlet/view/RedirectViewTests.java
/*
* Copyright 2002-2019 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.web.servlet.view;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.springframework.beans.testfixture.beans.TestBean;
import org.springframework.http.HttpStatus;
import org.springframework.ui.ModelMap;
import org.springframework.web.context.ContextLoader;
import org.springframework.web.context.support.StaticWebApplicationContext;
import org.springframework.web.servlet.DispatcherServlet;
import org.springframework.web.servlet.FlashMap;
import org.springframework.web.servlet.View;
import org.springframework.web.servlet.support.RequestDataValueProcessor;
import org.springframework.web.servlet.support.RequestDataValueProcessorWrapper;
import org.springframework.web.servlet.support.SessionFlashMapManager;
import org.springframework.web.testfixture.servlet.MockHttpServletRequest;
import org.springframework.web.testfixture.servlet.MockHttpServletResponse;
import org.springframework.web.testfixture.servlet.MockServletContext;
import org.springframework.web.util.WebUtils;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
import static org.mockito.BDDMockito.given;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
/**
* Tests for redirect view, and query string construction.
* Doesn't test URL encoding, although it does check that it's called.
*
* @author Rod Johnson
* @author Juergen Hoeller
* @author Sam Brannen
* @author Arjen Poutsma
* @author Rossen Stoyanchev
* @since 27.05.2003
*/
public class RedirectViewTests {
private MockHttpServletRequest request;
private MockHttpServletResponse response;
@BeforeEach
public void setUp() throws Exception {
this.request = new MockHttpServletRequest();
this.request.setContextPath("/context");
this.request.setCharacterEncoding(WebUtils.DEFAULT_CHARACTER_ENCODING);
this.request.setAttribute(DispatcherServlet.OUTPUT_FLASH_MAP_ATTRIBUTE, new FlashMap());
this.request.setAttribute(DispatcherServlet.FLASH_MAP_MANAGER_ATTRIBUTE, new SessionFlashMapManager());
this.response = new MockHttpServletResponse();
}
@Test
public void noUrlSet() throws Exception {
RedirectView rv = new RedirectView();
assertThatIllegalArgumentException().isThrownBy(
rv::afterPropertiesSet);
}
@Test
public void http11() throws Exception {
RedirectView rv = new RedirectView();
rv.setUrl("https://url.somewhere.com");
rv.setHttp10Compatible(false);
rv.render(new HashMap<>(), request, response);
assertThat(response.getStatus()).isEqualTo(303);
assertThat(response.getHeader("Location")).isEqualTo("https://url.somewhere.com");
}
@Test
public void explicitStatusCodeHttp11() throws Exception {
RedirectView rv = new RedirectView();
rv.setUrl("https://url.somewhere.com");
rv.setHttp10Compatible(false);
rv.setStatusCode(HttpStatus.MOVED_PERMANENTLY);
rv.render(new HashMap<>(), request, response);
assertThat(response.getStatus()).isEqualTo(301);
assertThat(response.getHeader("Location")).isEqualTo("https://url.somewhere.com");
}
@Test
public void explicitStatusCodeHttp10() throws Exception {
RedirectView rv = new RedirectView();
rv.setUrl("https://url.somewhere.com");
rv.setStatusCode(HttpStatus.MOVED_PERMANENTLY);
rv.render(new HashMap<>(), request, response);
assertThat(response.getStatus()).isEqualTo(301);
assertThat(response.getHeader("Location")).isEqualTo("https://url.somewhere.com");
}
@Test
public void attributeStatusCodeHttp10() throws Exception {
RedirectView rv = new RedirectView();
rv.setUrl("https://url.somewhere.com");
request.setAttribute(View.RESPONSE_STATUS_ATTRIBUTE, HttpStatus.CREATED);
rv.render(new HashMap<>(), request, response);
assertThat(response.getStatus()).isEqualTo(201);
assertThat(response.getHeader("Location")).isEqualTo("https://url.somewhere.com");
}
@Test
public void attributeStatusCodeHttp11() throws Exception {
RedirectView rv = new RedirectView();
rv.setUrl("https://url.somewhere.com");
rv.setHttp10Compatible(false);
request.setAttribute(View.RESPONSE_STATUS_ATTRIBUTE, HttpStatus.CREATED);
rv.render(new HashMap<>(), request, response);
assertThat(response.getStatus()).isEqualTo(201);
assertThat(response.getHeader("Location")).isEqualTo("https://url.somewhere.com");
}
@Test
public void flashMap() throws Exception {
RedirectView rv = new RedirectView();
rv.setUrl("https://url.somewhere.com/path");
rv.setHttp10Compatible(false);
FlashMap flashMap = new FlashMap();
flashMap.put("successMessage", "yay!");
request.setAttribute(DispatcherServlet.OUTPUT_FLASH_MAP_ATTRIBUTE, flashMap);
ModelMap model = new ModelMap("id", "1");
rv.render(model, request, response);
assertThat(response.getStatus()).isEqualTo(303);
assertThat(response.getHeader("Location")).isEqualTo("https://url.somewhere.com/path?id=1");
assertThat(flashMap.getTargetRequestPath()).isEqualTo("/path");
assertThat(flashMap.getTargetRequestParams().toSingleValueMap()).isEqualTo(model);
}
@Test
public void updateTargetUrl() throws Exception {
StaticWebApplicationContext wac = new StaticWebApplicationContext();
wac.registerSingleton("requestDataValueProcessor", RequestDataValueProcessorWrapper.class);
wac.setServletContext(new MockServletContext());
wac.refresh();
RequestDataValueProcessor mockProcessor = mock(RequestDataValueProcessor.class);
wac.getBean(RequestDataValueProcessorWrapper.class).setRequestDataValueProcessor(mockProcessor);
RedirectView rv = new RedirectView();
rv.setApplicationContext(wac); // Init RedirectView with WebAppCxt
rv.setUrl("/path");
request.setAttribute(DispatcherServlet.WEB_APPLICATION_CONTEXT_ATTRIBUTE, wac);
given(mockProcessor.processUrl(request, "/path")).willReturn("/path?key=123");
rv.render(new ModelMap(), request, response);
verify(mockProcessor).processUrl(request, "/path");
}
@Test
public void updateTargetUrlWithContextLoader() throws Exception {
StaticWebApplicationContext wac = new StaticWebApplicationContext();
wac.registerSingleton("requestDataValueProcessor", RequestDataValueProcessorWrapper.class);
MockServletContext servletContext = new MockServletContext();
ContextLoader contextLoader = new ContextLoader(wac);
contextLoader.initWebApplicationContext(servletContext);
try {
RequestDataValueProcessor mockProcessor = mock(RequestDataValueProcessor.class);
wac.getBean(RequestDataValueProcessorWrapper.class).setRequestDataValueProcessor(mockProcessor);
RedirectView rv = new RedirectView();
rv.setUrl("/path");
given(mockProcessor.processUrl(request, "/path")).willReturn("/path?key=123");
rv.render(new ModelMap(), request, response);
verify(mockProcessor).processUrl(request, "/path");
}
finally {
contextLoader.closeWebApplicationContext(servletContext);
}
}
@Test // SPR-13693
public void remoteHost() throws Exception {
RedirectView rv = new RedirectView();
assertThat(rv.isRemoteHost("https://url.somewhere.com")).isFalse();
assertThat(rv.isRemoteHost("/path")).isFalse();
assertThat(rv.isRemoteHost("http://somewhereelse.example")).isFalse();
rv.setHosts(new String[] {"url.somewhere.com"});
assertThat(rv.isRemoteHost("https://url.somewhere.com")).isFalse();
assertThat(rv.isRemoteHost("/path")).isFalse();
assertThat(rv.isRemoteHost("http://somewhereelse.example")).isTrue();
}
@Test // SPR-16752
public void contextRelativeWithValidatedContextPath() throws Exception {
String url = "/myUrl";
this.request.setContextPath("//context");
this.response = new MockHttpServletResponse();
doTest(new HashMap<>(), url, true, "/context" + url);
this.request.setContextPath("///context");
this.response = new MockHttpServletResponse();
doTest(new HashMap<>(), url, true, "/context" + url);
}
@Test
public void emptyMap() throws Exception {
String url = "/myUrl";
doTest(new HashMap<>(), url, false, url);
}
@Test
public void emptyMapWithContextRelative() throws Exception {
String url = "/myUrl";
doTest(new HashMap<>(), url, true, "/context" + url);
}
@Test
public void singleParam() throws Exception {
String url = "https://url.somewhere.com";
String key = "foo";
String val = "bar";
Map<String, String> model = new HashMap<>();
model.put(key, val);
String expectedUrlForEncoding = url + "?" + key + "=" + val;
doTest(model, url, false, expectedUrlForEncoding);
}
@Test
public void singleParamWithoutExposingModelAttributes() throws Exception {
String url = "https://url.somewhere.com";
Map<String, String> model = Collections.singletonMap("foo", "bar");
TestRedirectView rv = new TestRedirectView(url, false, model);
rv.setExposeModelAttributes(false);
rv.render(model, request, response);
assertThat(this.response.getRedirectedUrl()).isEqualTo(url);
}
@Test
public void paramWithAnchor() throws Exception {
String url = "https://url.somewhere.com/test.htm#myAnchor";
String key = "foo";
String val = "bar";
Map<String, String> model = new HashMap<>();
model.put(key, val);
String expectedUrlForEncoding = "https://url.somewhere.com/test.htm" + "?" + key + "=" + val + "#myAnchor";
doTest(model, url, false, expectedUrlForEncoding);
}
@Test
public void contextRelativeQueryParam() throws Exception {
String url = "/test.html?id=1";
doTest(new HashMap<>(), url, true, "/context" + url);
}
@Test
public void twoParams() throws Exception {
String url = "https://url.somewhere.com";
String key = "foo";
String val = "bar";
String key2 = "thisIsKey2";
String val2 = "andThisIsVal2";
Map<String, String> model = new HashMap<>();
model.put(key, val);
model.put(key2, val2);
try {
String expectedUrlForEncoding = url + "?" + key + "=" + val + "&" + key2 + "=" + val2;
doTest(model, url, false, expectedUrlForEncoding);
}
catch (AssertionError err) {
// OK, so it's the other order... probably on Sun JDK 1.6 or IBM JDK 1.5
String expectedUrlForEncoding = url + "?" + key2 + "=" + val2 + "&" + key + "=" + val;
doTest(model, url, false, expectedUrlForEncoding);
}
}
@Test
public void arrayParam() throws Exception {
String url = "https://url.somewhere.com";
String key = "foo";
String[] val = new String[] {"bar", "baz"};
Map<String, String[]> model = new HashMap<>();
model.put(key, val);
try {
String expectedUrlForEncoding = url + "?" + key + "=" + val[0] + "&" + key + "=" + val[1];
doTest(model, url, false, expectedUrlForEncoding);
}
catch (AssertionError err) {
// OK, so it's the other order... probably on Sun JDK 1.6 or IBM JDK 1.5
String expectedUrlForEncoding = url + "?" + key + "=" + val[1] + "&" + key + "=" + val[0];
doTest(model, url, false, expectedUrlForEncoding);
}
}
@Test
public void collectionParam() throws Exception {
String url = "https://url.somewhere.com";
String key = "foo";
List<String> val = new ArrayList<>();
val.add("bar");
val.add("baz");
Map<String, List<String>> model = new HashMap<>();
model.put(key, val);
try {
String expectedUrlForEncoding = url + "?" + key + "=" + val.get(0) + "&" + key + "=" + val.get(1);
doTest(model, url, false, expectedUrlForEncoding);
}
catch (AssertionError err) {
// OK, so it's the other order... probably on Sun JDK 1.6 or IBM JDK 1.5
String expectedUrlForEncoding = url + "?" + key + "=" + val.get(1) + "&" + key + "=" + val.get(0);
doTest(model, url, false, expectedUrlForEncoding);
}
}
@Test
public void objectConversion() throws Exception {
String url = "https://url.somewhere.com";
String key = "foo";
String val = "bar";
String key2 = "int2";
Object val2 = 611;
String key3 = "tb";
Object val3 = new TestBean();
Map<String, Object> model = new LinkedHashMap<>();
model.put(key, val);
model.put(key2, val2);
model.put(key3, val3);
String expectedUrlForEncoding = url + "?" + key + "=" + val + "&" + key2 + "=" + val2;
doTest(model, url, false, expectedUrlForEncoding);
}
@Test
public void propagateQueryParams() throws Exception {
RedirectView rv = new RedirectView();
rv.setPropagateQueryParams(true);
rv.setUrl("https://url.somewhere.com?foo=bar#bazz");
request.setQueryString("a=b&c=d");
rv.render(new HashMap<>(), request, response);
assertThat(response.getStatus()).isEqualTo(302);
assertThat(response.getHeader("Location")).isEqualTo("https://url.somewhere.com?foo=bar&a=b&c=d#bazz");
}
private void doTest(Map<String, ?> map, String url, boolean contextRelative, String expectedUrl)
throws Exception {
TestRedirectView rv = new TestRedirectView(url, contextRelative, map);
rv.render(map, request, response);
assertThat(rv.queryPropertiesCalled).as("queryProperties() should have been called.").isTrue();
assertThat(this.response.getRedirectedUrl()).isEqualTo(expectedUrl);
}
private static class TestRedirectView extends RedirectView {
private Map<String, ?> expectedModel;
private boolean queryPropertiesCalled = false;
public TestRedirectView(String url, boolean contextRelative, Map<String, ?> expectedModel) {
super(url, contextRelative);
this.expectedModel = expectedModel;
}
/**
* Test whether this callback method is called with correct args
*/
@Override
protected Map<String, Object> queryProperties(Map<String, Object> model) {
assertThat(this.expectedModel.equals(model)).as("Map and model must be equal.").isTrue();
this.queryPropertiesCalled = true;
return super.queryProperties(model);
}
}
}
相关信息
相关文章
spring ContentNegotiatingViewResolverTests 源码
spring DefaultRequestToViewNameTranslatorTests 源码
spring DummyMacroRequestContext 源码
spring InternalResourceViewTests 源码
spring RedirectViewUriTemplateTests 源码
spring ResourceBundleViewResolverNoCacheTests 源码
0
赞
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
7、 openharmony
-
9、 golang