From 9bc4663ead7ce90c5749537b6debaebf770f1280 Mon Sep 17 00:00:00 2001 From: Rossen Stoyanchev Date: Thu, 20 Jun 2013 20:39:35 -0400 Subject: [PATCH] Add accept(String) method to mock request builder Issue: SPR-10390 --- .../request/MockHttpServletRequestBuilder.java | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/spring-test-mvc/src/main/java/org/springframework/test/web/servlet/request/MockHttpServletRequestBuilder.java b/spring-test-mvc/src/main/java/org/springframework/test/web/servlet/request/MockHttpServletRequestBuilder.java index 3a7d016980..142131fcf5 100644 --- a/spring-test-mvc/src/main/java/org/springframework/test/web/servlet/request/MockHttpServletRequestBuilder.java +++ b/spring-test-mvc/src/main/java/org/springframework/test/web/servlet/request/MockHttpServletRequestBuilder.java @@ -25,6 +25,7 @@ import java.util.List; import java.util.Locale; import java.util.Map; import java.util.Map.Entry; + import javax.servlet.ServletContext; import javax.servlet.ServletRequest; import javax.servlet.http.Cookie; @@ -188,6 +189,21 @@ public class MockHttpServletRequestBuilder implements RequestBuilder, Mergeable return this; } + /** + * Set the 'Accept' header to the given media type(s). + * + * @param mediaTypes one or more media types + */ + public MockHttpServletRequestBuilder accept(String... mediaTypes) { + Assert.notEmpty(mediaTypes, "No 'Accept' media types"); + List result = new ArrayList(mediaTypes.length); + for (String mediaType : mediaTypes) { + result.add(MediaType.parseMediaType(mediaType)); + } + this.headers.set("Accept", MediaType.toString(result)); + return this; + } + /** * Set the request body. *