chore: add contentType to accept Spring MedieType (#1625)

rest-assured · Nov 18, 2022 · afbf13b · afbf13b
1 parent 14ef2c6
commit afbf13b
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 0 deletions.
diff --git a/...src/main/java/io/restassured/module/mockmvc/internal/MockMvcRequestSpecificationImpl.java b/...src/main/java/io/restassured/module/mockmvc/internal/MockMvcRequestSpecificationImpl.java
@@ -191,6 +191,11 @@ public MockMvcRequestSpecification contentType(ContentType contentType) {
         return header(CONTENT_TYPE, contentType.toString());
     }
 
+    public MockMvcRequestSpecification contentType(MediaType mediaType) {
+        notNull(mediaType, "mediaType");
+        return header(CONTENT_TYPE, mediaType.toString());
+    }
+
     public MockMvcRequestSpecification contentType(String contentType) {
         notNull(contentType, "contentType");
         return header(CONTENT_TYPE, contentType);

diff --git a/...rc/main/java/io/restassured/module/mockmvc/specification/MockMvcRequestSpecification.java b/...rc/main/java/io/restassured/module/mockmvc/specification/MockMvcRequestSpecification.java
@@ -57,6 +57,16 @@ public interface MockMvcRequestSpecification extends MockMvcRequestSender {
      */
     MockMvcRequestSpecification contentType(ContentType contentType);
 
+    /**
+     * Specify the content type of the request.
+     *
+     * @param mediaType The content type of the request
+     * @return The request specification
+     * @see ContentType
+     * @see MediaType
+     */
+    MockMvcRequestSpecification contentType(MediaType mediaType);
+
     /**
      * Specify the content type of the request.
      *

diff --git a/...a/io/restassured/module/webtestclient/internal/WebTestClientRequestSpecificationImpl.java b/...a/io/restassured/module/webtestclient/internal/WebTestClientRequestSpecificationImpl.java
@@ -118,6 +118,12 @@ public WebTestClientRequestSpecification contentType(ContentType contentType) {
         return header(CONTENT_TYPE, contentType.toString());
     }
 
+    @Override
+    public WebTestClientRequestSpecification contentType(MediaType mediaType) {
+        notNull(mediaType, "mediaType");
+        return header(CONTENT_TYPE, mediaType.toString());
+    }
+
     @Override
     public WebTestClientRequestSpecification contentType(String contentType) {
         notNull(contentType, "contentType");

diff --git a/.../io/restassured/module/webtestclient/specification/WebTestClientRequestSpecification.java b/.../io/restassured/module/webtestclient/specification/WebTestClientRequestSpecification.java
@@ -46,6 +46,16 @@ public interface WebTestClientRequestSpecification extends WebTestClientRequestS
 	 */
 	WebTestClientRequestSpecification contentType(ContentType contentType);
 
+	/**
+	 * Specify the content type of the request.
+	 *
+	 * @param mediaType The content type of the request
+	 * @return The request specification
+	 *
+	 * @see ContentType
+	 */
+	WebTestClientRequestSpecification contentType(MediaType mediaType);
+
 	/**
 	 * Specify the content type of the request.
 	 *