MockHttpServletRequest.java 27.9 KB
Newer Older
1
/*
2
 * Copyright 2002-2014 the original author or authors.
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
 *
 * 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;

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
21
import java.io.IOException;
22 23 24 25 26 27 28 29 30 31 32
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.io.UnsupportedEncodingException;
import java.security.Principal;
import java.util.Collection;
import java.util.Collections;
import java.util.Date;
import java.util.Enumeration;
import java.util.HashSet;
import java.util.LinkedHashMap;
33
import java.util.LinkedHashSet;
34 35 36 37 38
import java.util.LinkedList;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Set;
39

40 41
import javax.servlet.AsyncContext;
import javax.servlet.DispatcherType;
42 43
import javax.servlet.RequestDispatcher;
import javax.servlet.ServletContext;
44
import javax.servlet.ServletException;
45
import javax.servlet.ServletInputStream;
46 47
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
48 49
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
50
import javax.servlet.http.HttpServletResponse;
51
import javax.servlet.http.HttpSession;
52
import javax.servlet.http.Part;
53 54

import org.springframework.util.Assert;
55
import org.springframework.util.LinkedCaseInsensitiveMap;
56
import org.springframework.util.StringUtils;
57 58

/**
59 60
 * Mock implementation of the {@link javax.servlet.http.HttpServletRequest} interface.
 *
61
 * <p>As of Spring 4.0, this set of mocks is designed on a Servlet 3.0 baseline.
62
 *
63 64 65 66
 * @author Juergen Hoeller
 * @author Rod Johnson
 * @author Rick Evans
 * @author Mark Fisher
67
 * @author Chris Beams
68
 * @author Sam Brannen
69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102
 * @since 1.0.2
 */
public class MockHttpServletRequest implements HttpServletRequest {

	/**
	 * The default protocol: 'http'.
	 */
	public static final String DEFAULT_PROTOCOL = "http";

	/**
	 * The default server address: '127.0.0.1'.
	 */
	public static final String DEFAULT_SERVER_ADDR = "127.0.0.1";

	/**
	 * The default server name: 'localhost'.
	 */
	public static final String DEFAULT_SERVER_NAME = "localhost";

	/**
	 * The default server port: '80'.
	 */
	public static final int DEFAULT_SERVER_PORT = 80;

	/**
	 * The default remote address: '127.0.0.1'.
	 */
	public static final String DEFAULT_REMOTE_ADDR = "127.0.0.1";

	/**
	 * The default remote host: 'localhost'.
	 */
	public static final String DEFAULT_REMOTE_HOST = "localhost";

103 104
	private static final String CONTENT_TYPE_HEADER = "Content-Type";

105
	private static final String CHARSET_PREFIX = "charset=";
106

107 108 109
	private static final ServletInputStream EMPTY_SERVLET_INPUT_STREAM =
			new DelegatingServletInputStream(new ByteArrayInputStream(new byte[0]));

110

111 112
	private boolean active = true;

113

114
	// ---------------------------------------------------------------------
115
	// ServletRequest properties
116
	// ---------------------------------------------------------------------
117

J
Juergen Hoeller 已提交
118
	private final Map<String, Object> attributes = new LinkedHashMap<String, Object>();
119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154

	private String characterEncoding;

	private byte[] content;

	private String contentType;

	private final Map<String, String[]> parameters = new LinkedHashMap<String, String[]>(16);

	private String protocol = DEFAULT_PROTOCOL;

	private String scheme = DEFAULT_PROTOCOL;

	private String serverName = DEFAULT_SERVER_NAME;

	private int serverPort = DEFAULT_SERVER_PORT;

	private String remoteAddr = DEFAULT_REMOTE_ADDR;

	private String remoteHost = DEFAULT_REMOTE_HOST;

	/** List of locales in descending order */
	private final List<Locale> locales = new LinkedList<Locale>();

	private boolean secure = false;

	private final ServletContext servletContext;

	private int remotePort = DEFAULT_SERVER_PORT;

	private String localName = DEFAULT_SERVER_NAME;

	private String localAddr = DEFAULT_SERVER_ADDR;

	private int localPort = DEFAULT_SERVER_PORT;

155 156 157 158 159 160 161 162
	private boolean asyncStarted = false;

	private boolean asyncSupported = false;

	private MockAsyncContext asyncContext;

	private DispatcherType dispatcherType = DispatcherType.REQUEST;

163

164
	// ---------------------------------------------------------------------
165
	// HttpServletRequest properties
166
	// ---------------------------------------------------------------------
167 168 169 170 171

	private String authType;

	private Cookie[] cookies;

172
	private final Map<String, HeaderValueHolder> headers = new LinkedCaseInsensitiveMap<HeaderValueHolder>();
173 174 175 176 177 178 179 180 181 182 183 184 185 186 187

	private String method;

	private String pathInfo;

	private String contextPath = "";

	private String queryString;

	private String remoteUser;

	private final Set<String> userRoles = new HashSet<String>();

	private Principal userPrincipal;

188 189
	private String requestedSessionId;

190 191 192 193 194 195 196 197 198 199 200 201
	private String requestURI;

	private String servletPath = "";

	private HttpSession session;

	private boolean requestedSessionIdValid = true;

	private boolean requestedSessionIdFromCookie = true;

	private boolean requestedSessionIdFromURL = false;

202 203
	private final Map<String, Part> parts = new LinkedHashMap<String, Part>();

204

205
	// ---------------------------------------------------------------------
206
	// Constructors
207
	// ---------------------------------------------------------------------
208 209

	/**
210
	 * Create a new {@code MockHttpServletRequest} with a default
211
	 * {@link MockServletContext}.
212
	 * @see #MockHttpServletRequest(ServletContext, String, String)
213 214 215 216 217 218
	 */
	public MockHttpServletRequest() {
		this(null, "", "");
	}

	/**
219
	 * Create a new {@code MockHttpServletRequest} with a default
220
	 * {@link MockServletContext}.
221 222
	 * @param method the request method (may be {@code null})
	 * @param requestURI the request URI (may be {@code null})
223 224
	 * @see #setMethod
	 * @see #setRequestURI
225
	 * @see #MockHttpServletRequest(ServletContext, String, String)
226 227 228 229 230 231
	 */
	public MockHttpServletRequest(String method, String requestURI) {
		this(null, method, requestURI);
	}

	/**
232
	 * Create a new {@code MockHttpServletRequest} with the supplied {@link ServletContext}.
233 234
	 * @param servletContext the ServletContext that the request runs in
	 * (may be {@code null} to use a default {@link MockServletContext})
235
	 * @see #MockHttpServletRequest(ServletContext, String, String)
236 237 238 239 240 241
	 */
	public MockHttpServletRequest(ServletContext servletContext) {
		this(servletContext, "", "");
	}

	/**
242 243 244
	 * Create a new {@code MockHttpServletRequest} with the supplied {@link ServletContext},
	 * {@code method}, and {@code requestURI}.
	 * <p>The preferred locale will be set to {@link Locale#ENGLISH}.
245
	 * @param servletContext the ServletContext that the request runs in (may be
246 247 248
	 * {@code null} to use a default {@link MockServletContext})
	 * @param method the request method (may be {@code null})
	 * @param requestURI the request URI (may be {@code null})
249 250
	 * @see #setMethod
	 * @see #setRequestURI
251
	 * @see #setPreferredLocales
252 253 254 255
	 * @see MockServletContext
	 */
	public MockHttpServletRequest(ServletContext servletContext, String method, String requestURI) {
		this.servletContext = (servletContext != null ? servletContext : new MockServletContext());
256 257
		this.method = method;
		this.requestURI = requestURI;
258 259 260
		this.locales.add(Locale.ENGLISH);
	}

261

262
	// ---------------------------------------------------------------------
263
	// Lifecycle methods
264
	// ---------------------------------------------------------------------
265 266

	/**
267 268
	 * Return the ServletContext that this request is associated with. (Not
	 * available in the standard HttpServletRequest interface for some reason.)
269
	 */
270
	@Override
271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306
	public ServletContext getServletContext() {
		return this.servletContext;
	}

	/**
	 * Return whether this request is still active (that is, not completed yet).
	 */
	public boolean isActive() {
		return this.active;
	}

	/**
	 * Mark this request as completed, keeping its state.
	 */
	public void close() {
		this.active = false;
	}

	/**
	 * Invalidate this request, clearing its state.
	 */
	public void invalidate() {
		close();
		clearAttributes();
	}

	/**
	 * Check whether this request is still active (that is, not completed yet),
	 * throwing an IllegalStateException if not active anymore.
	 */
	protected void checkActive() throws IllegalStateException {
		if (!this.active) {
			throw new IllegalStateException("Request is not active anymore");
		}
	}

307

308
	// ---------------------------------------------------------------------
309
	// ServletRequest interface
310
	// ---------------------------------------------------------------------
311

312
	@Override
313 314 315 316 317
	public Object getAttribute(String name) {
		checkActive();
		return this.attributes.get(name);
	}

318
	@Override
319 320
	public Enumeration<String> getAttributeNames() {
		checkActive();
321
		return Collections.enumeration(new LinkedHashSet<String>(this.attributes.keySet()));
322 323
	}

324
	@Override
325 326 327 328
	public String getCharacterEncoding() {
		return this.characterEncoding;
	}

329
	@Override
330 331
	public void setCharacterEncoding(String characterEncoding) {
		this.characterEncoding = characterEncoding;
332
		updateContentTypeHeader();
333
	}
334

335 336 337
	private void updateContentTypeHeader() {
		if (this.contentType != null) {
			StringBuilder sb = new StringBuilder(this.contentType);
J
Juergen Hoeller 已提交
338
			if (!this.contentType.toLowerCase().contains(CHARSET_PREFIX) && this.characterEncoding != null) {
339 340 341
				sb.append(";").append(CHARSET_PREFIX).append(this.characterEncoding);
			}
			doAddHeaderValue(CONTENT_TYPE_HEADER, sb.toString(), true);
342
		}
343 344 345 346 347 348
	}

	public void setContent(byte[] content) {
		this.content = content;
	}

349
	@Override
350 351 352 353
	public int getContentLength() {
		return (this.content != null ? this.content.length : -1);
	}

354 355 356 357
	public long getContentLengthLong() {
		return getContentLength();
	}

358 359
	public void setContentType(String contentType) {
		this.contentType = contentType;
360 361 362 363 364 365
		if (contentType != null) {
			int charsetIndex = contentType.toLowerCase().indexOf(CHARSET_PREFIX);
			if (charsetIndex != -1) {
				String encoding = contentType.substring(charsetIndex + CHARSET_PREFIX.length());
				this.characterEncoding = encoding;
			}
366
			updateContentTypeHeader();
367
		}
368 369
	}

370
	@Override
371 372 373 374
	public String getContentType() {
		return this.contentType;
	}

375
	@Override
376 377 378 379 380
	public ServletInputStream getInputStream() {
		if (this.content != null) {
			return new DelegatingServletInputStream(new ByteArrayInputStream(this.content));
		}
		else {
381
			return EMPTY_SERVLET_INPUT_STREAM;
382 383 384 385 386
		}
	}

	/**
	 * Set a single value for the specified HTTP parameter.
387
	 * <p>If there are already one or more values registered for the given
388 389 390
	 * parameter name, they will be replaced.
	 */
	public void setParameter(String name, String value) {
391
		setParameter(name, new String[] {value});
392 393 394 395
	}

	/**
	 * Set an array of values for the specified HTTP parameter.
396
	 * <p>If there are already one or more values registered for the given
397 398 399 400 401 402 403 404
	 * parameter name, they will be replaced.
	 */
	public void setParameter(String name, String[] values) {
		Assert.notNull(name, "Parameter name must not be null");
		this.parameters.put(name, values);
	}

	/**
405
	 * Sets all provided parameters <strong>replacing</strong> any existing
406 407
	 * values for the provided parameter names. To add without replacing
	 * existing values, use {@link #addParameters(java.util.Map)}.
408
	 */
409
	@SuppressWarnings("rawtypes")
410 411 412
	public void setParameters(Map params) {
		Assert.notNull(params, "Parameter map must not be null");
		for (Object key : params.keySet()) {
413 414
			Assert.isInstanceOf(String.class, key,
					"Parameter map key must be of type [" + String.class.getName() + "]");
415 416 417 418 419 420 421 422
			Object value = params.get(key);
			if (value instanceof String) {
				this.setParameter((String) key, (String) value);
			}
			else if (value instanceof String[]) {
				this.setParameter((String) key, (String[]) value);
			}
			else {
423 424
				throw new IllegalArgumentException(
						"Parameter map value must be single value " + " or array of type [" + String.class.getName() + "]");
425 426 427 428 429 430
			}
		}
	}

	/**
	 * Add a single value for the specified HTTP parameter.
431
	 * <p>If there are already one or more values registered for the given
432 433 434
	 * parameter name, the given value will be added to the end of the list.
	 */
	public void addParameter(String name, String value) {
435
		addParameter(name, new String[] {value});
436 437 438 439
	}

	/**
	 * Add an array of values for the specified HTTP parameter.
440
	 * <p>If there are already one or more values registered for the given
441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457
	 * parameter name, the given values will be added to the end of the list.
	 */
	public void addParameter(String name, String[] values) {
		Assert.notNull(name, "Parameter name must not be null");
		String[] oldArr = this.parameters.get(name);
		if (oldArr != null) {
			String[] newArr = new String[oldArr.length + values.length];
			System.arraycopy(oldArr, 0, newArr, 0, oldArr.length);
			System.arraycopy(values, 0, newArr, oldArr.length, values.length);
			this.parameters.put(name, newArr);
		}
		else {
			this.parameters.put(name, values);
		}
	}

	/**
458
	 * Adds all provided parameters <strong>without</strong> replacing any
459
	 * existing values. To replace existing values, use
460 461
	 * {@link #setParameters(java.util.Map)}.
	 */
462
	@SuppressWarnings("rawtypes")
463 464 465
	public void addParameters(Map params) {
		Assert.notNull(params, "Parameter map must not be null");
		for (Object key : params.keySet()) {
466 467
			Assert.isInstanceOf(String.class, key,
					"Parameter map key must be of type [" + String.class.getName() + "]");
468 469 470 471 472 473 474 475
			Object value = params.get(key);
			if (value instanceof String) {
				this.addParameter((String) key, (String) value);
			}
			else if (value instanceof String[]) {
				this.addParameter((String) key, (String[]) value);
			}
			else {
476 477
				throw new IllegalArgumentException("Parameter map value must be single value " +
						" or array of type [" + String.class.getName() + "]");
478 479 480 481 482
			}
		}
	}

	/**
483
	 * Remove already registered values for the specified HTTP parameter, if any.
484 485 486 487 488 489 490 491 492 493 494 495 496
	 */
	public void removeParameter(String name) {
		Assert.notNull(name, "Parameter name must not be null");
		this.parameters.remove(name);
	}

	/**
	 * Removes all existing parameters.
	 */
	public void removeAllParameters() {
		this.parameters.clear();
	}

497
	@Override
498
	public String getParameter(String name) {
499
		String[] arr = (name != null ? this.parameters.get(name) : null);
500 501 502
		return (arr != null && arr.length > 0 ? arr[0] : null);
	}

503
	@Override
504 505 506 507
	public Enumeration<String> getParameterNames() {
		return Collections.enumeration(this.parameters.keySet());
	}

508
	@Override
509
	public String[] getParameterValues(String name) {
510
		return (name != null ? this.parameters.get(name) : null);
511 512
	}

513
	@Override
514 515 516 517 518 519 520 521
	public Map<String, String[]> getParameterMap() {
		return Collections.unmodifiableMap(this.parameters);
	}

	public void setProtocol(String protocol) {
		this.protocol = protocol;
	}

522
	@Override
523 524 525 526 527 528 529 530
	public String getProtocol() {
		return this.protocol;
	}

	public void setScheme(String scheme) {
		this.scheme = scheme;
	}

531
	@Override
532 533 534 535 536 537 538 539
	public String getScheme() {
		return this.scheme;
	}

	public void setServerName(String serverName) {
		this.serverName = serverName;
	}

540
	@Override
541 542 543 544 545 546 547 548
	public String getServerName() {
		return this.serverName;
	}

	public void setServerPort(int serverPort) {
		this.serverPort = serverPort;
	}

549
	@Override
550 551 552 553
	public int getServerPort() {
		return this.serverPort;
	}

554
	@Override
555 556 557
	public BufferedReader getReader() throws UnsupportedEncodingException {
		if (this.content != null) {
			InputStream sourceStream = new ByteArrayInputStream(this.content);
558 559
			Reader sourceReader = (this.characterEncoding != null) ?
					new InputStreamReader(sourceStream, this.characterEncoding) : new InputStreamReader(sourceStream);
560 561 562 563 564 565 566 567 568 569 570
			return new BufferedReader(sourceReader);
		}
		else {
			return null;
		}
	}

	public void setRemoteAddr(String remoteAddr) {
		this.remoteAddr = remoteAddr;
	}

571
	@Override
572 573 574 575 576 577 578 579
	public String getRemoteAddr() {
		return this.remoteAddr;
	}

	public void setRemoteHost(String remoteHost) {
		this.remoteHost = remoteHost;
	}

580
	@Override
581 582 583 584
	public String getRemoteHost() {
		return this.remoteHost;
	}

585
	@Override
586 587 588 589 590 591 592 593 594 595 596
	public void setAttribute(String name, Object value) {
		checkActive();
		Assert.notNull(name, "Attribute name must not be null");
		if (value != null) {
			this.attributes.put(name, value);
		}
		else {
			this.attributes.remove(name);
		}
	}

597
	@Override
598 599 600 601 602 603 604 605 606 607 608 609 610 611 612
	public void removeAttribute(String name) {
		checkActive();
		Assert.notNull(name, "Attribute name must not be null");
		this.attributes.remove(name);
	}

	/**
	 * Clear all of this request's attributes.
	 */
	public void clearAttributes() {
		this.attributes.clear();
	}

	/**
	 * Add a new preferred locale, before any existing locales.
613
	 * @see #setPreferredLocales
614 615 616 617 618 619
	 */
	public void addPreferredLocale(Locale locale) {
		Assert.notNull(locale, "Locale must not be null");
		this.locales.add(0, locale);
	}

620 621 622 623 624 625 626
	/**
	 * Set the list of preferred locales, in descending order, effectively replacing
	 * any existing locales.
	 * @see #addPreferredLocale
	 * @since 3.2
	 */
	public void setPreferredLocales(List<Locale> locales) {
627
		Assert.notEmpty(locales, "Locale list must not be empty");
628 629 630 631
		this.locales.clear();
		this.locales.addAll(locales);
	}

632
	@Override
633 634 635 636
	public Locale getLocale() {
		return this.locales.get(0);
	}

637
	@Override
638 639 640 641 642 643 644 645
	public Enumeration<Locale> getLocales() {
		return Collections.enumeration(this.locales);
	}

	public void setSecure(boolean secure) {
		this.secure = secure;
	}

646
	@Override
647 648 649 650
	public boolean isSecure() {
		return this.secure;
	}

651
	@Override
652 653 654 655
	public RequestDispatcher getRequestDispatcher(String path) {
		return new MockRequestDispatcher(path);
	}

656
	@Override
P
Phillip Webb 已提交
657
	@Deprecated
658 659 660 661 662 663 664 665
	public String getRealPath(String path) {
		return this.servletContext.getRealPath(path);
	}

	public void setRemotePort(int remotePort) {
		this.remotePort = remotePort;
	}

666
	@Override
667 668 669 670 671 672 673 674
	public int getRemotePort() {
		return this.remotePort;
	}

	public void setLocalName(String localName) {
		this.localName = localName;
	}

675
	@Override
676 677 678 679 680 681 682 683
	public String getLocalName() {
		return this.localName;
	}

	public void setLocalAddr(String localAddr) {
		this.localAddr = localAddr;
	}

684
	@Override
685 686 687 688 689 690 691 692
	public String getLocalAddr() {
		return this.localAddr;
	}

	public void setLocalPort(int localPort) {
		this.localPort = localPort;
	}

693
	@Override
694 695 696 697
	public int getLocalPort() {
		return this.localPort;
	}

698
	@Override
699 700 701 702
	public AsyncContext startAsync() {
		return startAsync(this, null);
	}

703
	@Override
704 705 706 707 708 709 710 711 712 713 714 715 716
	public AsyncContext startAsync(ServletRequest request, ServletResponse response) {
		if (!this.asyncSupported) {
			throw new IllegalStateException("Async not supported");
		}
		this.asyncStarted = true;
		this.asyncContext = new MockAsyncContext(request, response);
		return this.asyncContext;
	}

	public void setAsyncStarted(boolean asyncStarted) {
		this.asyncStarted = asyncStarted;
	}

717
	@Override
718 719 720 721 722 723 724 725
	public boolean isAsyncStarted() {
		return this.asyncStarted;
	}

	public void setAsyncSupported(boolean asyncSupported) {
		this.asyncSupported = asyncSupported;
	}

726
	@Override
727 728 729 730 731 732 733 734
	public boolean isAsyncSupported() {
		return this.asyncSupported;
	}

	public void setAsyncContext(MockAsyncContext asyncContext) {
		this.asyncContext = asyncContext;
	}

735
	@Override
736 737 738 739 740 741 742 743
	public AsyncContext getAsyncContext() {
		return this.asyncContext;
	}

	public void setDispatcherType(DispatcherType dispatcherType) {
		this.dispatcherType = dispatcherType;
	}

744
	@Override
745 746 747 748
	public DispatcherType getDispatcherType() {
		return this.dispatcherType;
	}

749

750
	// ---------------------------------------------------------------------
751
	// HttpServletRequest interface
752
	// ---------------------------------------------------------------------
753 754 755 756 757

	public void setAuthType(String authType) {
		this.authType = authType;
	}

758
	@Override
759 760 761 762
	public String getAuthType() {
		return this.authType;
	}

J
Juergen Hoeller 已提交
763
	public void setCookies(Cookie... cookies) {
764 765 766
		this.cookies = cookies;
	}

767
	@Override
768 769 770 771 772 773
	public Cookie[] getCookies() {
		return this.cookies;
	}

	/**
	 * Add a header entry for the given name.
774
	 * <p>If there was no entry for that header name before, the value will be used
775 776 777
	 * as-is. In case of an existing entry, a String array will be created,
	 * adding the given value (more specifically, its toString representation)
	 * as further element.
778
	 * <p>Multiple values can only be stored as list of Strings, following the
779 780
	 * Servlet spec (see {@code getHeaders} accessor). As alternative to
	 * repeated {@code addHeader} calls for individual elements, you can
781 782
	 * use a single call with an entire array or Collection of values as
	 * parameter.
783 784 785 786 787 788 789
	 * @see #getHeaderNames
	 * @see #getHeader
	 * @see #getHeaders
	 * @see #getDateHeader
	 * @see #getIntHeader
	 */
	public void addHeader(String name, Object value) {
790 791 792 793 794 795
		if (CONTENT_TYPE_HEADER.equalsIgnoreCase(name)) {
			setContentType((String) value);
			return;
		}
		doAddHeaderValue(name, value, false);
	}
796

797 798
	@SuppressWarnings("rawtypes")
	private void doAddHeaderValue(String name, Object value, boolean replace) {
799 800
		HeaderValueHolder header = HeaderValueHolder.getByName(this.headers, name);
		Assert.notNull(value, "Header value must not be null");
801
		if (header == null || replace) {
802 803 804 805 806 807 808 809 810 811 812 813 814 815
			header = new HeaderValueHolder();
			this.headers.put(name, header);
		}
		if (value instanceof Collection) {
			header.addValues((Collection) value);
		}
		else if (value.getClass().isArray()) {
			header.addValueArray(value);
		}
		else {
			header.addValue(value);
		}
	}

816
	@Override
817 818 819 820 821 822 823 824 825 826
	public long getDateHeader(String name) {
		HeaderValueHolder header = HeaderValueHolder.getByName(this.headers, name);
		Object value = (header != null ? header.getValue() : null);
		if (value instanceof Date) {
			return ((Date) value).getTime();
		}
		else if (value instanceof Number) {
			return ((Number) value).longValue();
		}
		else if (value != null) {
827 828
			throw new IllegalArgumentException(
					"Value for header '" + name + "' is neither a Date nor a Number: " + value);
829 830 831 832 833 834
		}
		else {
			return -1L;
		}
	}

835
	@Override
836 837 838 839 840
	public String getHeader(String name) {
		HeaderValueHolder header = HeaderValueHolder.getByName(this.headers, name);
		return (header != null ? header.getStringValue() : null);
	}

841
	@Override
842 843 844 845 846
	public Enumeration<String> getHeaders(String name) {
		HeaderValueHolder header = HeaderValueHolder.getByName(this.headers, name);
		return Collections.enumeration(header != null ? header.getStringValues() : new LinkedList<String>());
	}

847
	@Override
848 849 850 851
	public Enumeration<String> getHeaderNames() {
		return Collections.enumeration(this.headers.keySet());
	}

852
	@Override
853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868
	public int getIntHeader(String name) {
		HeaderValueHolder header = HeaderValueHolder.getByName(this.headers, name);
		Object value = (header != null ? header.getValue() : null);
		if (value instanceof Number) {
			return ((Number) value).intValue();
		}
		else if (value instanceof String) {
			return Integer.parseInt((String) value);
		}
		else if (value != null) {
			throw new NumberFormatException("Value for header '" + name + "' is not a Number: " + value);
		}
		else {
			return -1;
		}
	}
869

870
	public void setMethod(String method) {
871
		this.method = method;
872 873
	}

874
	@Override
875 876 877 878 879 880 881 882
	public String getMethod() {
		return this.method;
	}

	public void setPathInfo(String pathInfo) {
		this.pathInfo = pathInfo;
	}

883
	@Override
884 885 886 887
	public String getPathInfo() {
		return this.pathInfo;
	}

888
	@Override
889 890 891 892 893 894 895 896
	public String getPathTranslated() {
		return (this.pathInfo != null ? getRealPath(this.pathInfo) : null);
	}

	public void setContextPath(String contextPath) {
		this.contextPath = contextPath;
	}

897
	@Override
898 899 900 901 902 903 904 905
	public String getContextPath() {
		return this.contextPath;
	}

	public void setQueryString(String queryString) {
		this.queryString = queryString;
	}

906
	@Override
907 908 909 910 911 912 913 914
	public String getQueryString() {
		return this.queryString;
	}

	public void setRemoteUser(String remoteUser) {
		this.remoteUser = remoteUser;
	}

915
	@Override
916 917 918 919 920 921 922 923
	public String getRemoteUser() {
		return this.remoteUser;
	}

	public void addUserRole(String role) {
		this.userRoles.add(role);
	}

924
	@Override
925
	public boolean isUserInRole(String role) {
926 927
		return (this.userRoles.contains(role) || (this.servletContext instanceof MockServletContext &&
				((MockServletContext) this.servletContext).getDeclaredRoles().contains(role)));
928 929 930 931 932 933
	}

	public void setUserPrincipal(Principal userPrincipal) {
		this.userPrincipal = userPrincipal;
	}

934
	@Override
935 936 937 938
	public Principal getUserPrincipal() {
		return this.userPrincipal;
	}

939 940 941 942
	public void setRequestedSessionId(String requestedSessionId) {
		this.requestedSessionId = requestedSessionId;
	}

943
	@Override
944
	public String getRequestedSessionId() {
945
		return this.requestedSessionId;
946 947 948
	}

	public void setRequestURI(String requestURI) {
949
		this.requestURI = requestURI;
950 951
	}

952
	@Override
953 954 955 956
	public String getRequestURI() {
		return this.requestURI;
	}

957
	@Override
958
	public StringBuffer getRequestURL() {
959 960 961 962 963 964 965
		StringBuffer url = new StringBuffer(this.scheme).append("://").append(this.serverName);

		if (this.serverPort > 0
				&& (("http".equalsIgnoreCase(scheme) && this.serverPort != 80) || ("https".equalsIgnoreCase(scheme) && this.serverPort != 443))) {
			url.append(':').append(this.serverPort);
		}

966 967 968 969
		if (StringUtils.hasText(getRequestURI())) {
			url.append(getRequestURI());
		}

970 971 972 973 974 975 976
		return url;
	}

	public void setServletPath(String servletPath) {
		this.servletPath = servletPath;
	}

977
	@Override
978 979 980 981 982 983 984 985 986 987 988 989
	public String getServletPath() {
		return this.servletPath;
	}

	public void setSession(HttpSession session) {
		this.session = session;
		if (session instanceof MockHttpSession) {
			MockHttpSession mockSession = ((MockHttpSession) session);
			mockSession.access();
		}
	}

990
	@Override
991 992 993 994 995 996 997 998 999 1000 1001 1002 1003
	public HttpSession getSession(boolean create) {
		checkActive();
		// Reset session if invalidated.
		if (this.session instanceof MockHttpSession && ((MockHttpSession) this.session).isInvalid()) {
			this.session = null;
		}
		// Create new session if necessary.
		if (this.session == null && create) {
			this.session = new MockHttpSession(this.servletContext);
		}
		return this.session;
	}

1004
	@Override
1005 1006 1007 1008
	public HttpSession getSession() {
		return getSession(true);
	}

1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022
	/**
	 * The implementation of this (Servlet 3.1+) method calls
	 * {@link MockHttpSession#changeSessionId()} if the session is a mock session.
	 * Otherwise it simply returns the current session id.
	 * @since 4.0.3
	 */
	public String changeSessionId() {
		Assert.isTrue(this.session != null, "The request does not have a session");
		if (this.session instanceof MockHttpSession) {
			return ((MockHttpSession) session).changeSessionId();
		}
		return this.session.getId();
	}

1023 1024 1025 1026
	public void setRequestedSessionIdValid(boolean requestedSessionIdValid) {
		this.requestedSessionIdValid = requestedSessionIdValid;
	}

1027
	@Override
1028 1029 1030 1031 1032 1033 1034 1035
	public boolean isRequestedSessionIdValid() {
		return this.requestedSessionIdValid;
	}

	public void setRequestedSessionIdFromCookie(boolean requestedSessionIdFromCookie) {
		this.requestedSessionIdFromCookie = requestedSessionIdFromCookie;
	}

1036
	@Override
1037 1038 1039 1040 1041 1042 1043 1044
	public boolean isRequestedSessionIdFromCookie() {
		return this.requestedSessionIdFromCookie;
	}

	public void setRequestedSessionIdFromURL(boolean requestedSessionIdFromURL) {
		this.requestedSessionIdFromURL = requestedSessionIdFromURL;
	}

1045
	@Override
1046 1047 1048 1049
	public boolean isRequestedSessionIdFromURL() {
		return this.requestedSessionIdFromURL;
	}

1050
	@Override
P
Phillip Webb 已提交
1051
	@Deprecated
1052 1053 1054 1055
	public boolean isRequestedSessionIdFromUrl() {
		return isRequestedSessionIdFromURL();
	}

1056
	@Override
1057
	public boolean authenticate(HttpServletResponse response) throws IOException, ServletException {
1058
		throw new UnsupportedOperationException();
1059 1060
	}

1061
	@Override
1062
	public void login(String username, String password) throws ServletException {
1063
		throw new UnsupportedOperationException();
1064 1065
	}

1066
	@Override
1067 1068 1069 1070 1071 1072
	public void logout() throws ServletException {
		this.userPrincipal = null;
		this.remoteUser = null;
		this.authType = null;
	}

1073 1074 1075 1076
	public void addPart(Part part) {
		this.parts.put(part.getName(), part);
	}

1077
	@Override
1078 1079 1080 1081
	public Part getPart(String name) throws IOException, IllegalStateException, ServletException {
		return this.parts.get(name);
	}

1082
	@Override
1083 1084 1085 1086
	public Collection<Part> getParts() throws IOException, IllegalStateException, ServletException {
		return this.parts.values();
	}

1087
}