Add unit tests for content type with charset

Endava · Mar 11, 2022 · d234cb9 · d234cb9
1 parent 60621d8
commit d234cb9
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/src/test/java/com/endava/cats/generator/simple/OpenApiUtilsTest.java b/src/test/java/com/endava/cats/generator/simple/OpenApiUtilsTest.java
@@ -9,6 +9,9 @@
 import org.junit.jupiter.api.Test;
 import org.mockito.Mockito;
 
+import java.util.AbstractMap;
+import java.util.Set;
+
 @QuarkusTest
 class OpenApiUtilsTest {
 
@@ -39,4 +42,26 @@ void shouldReturnDefaultContentType() {
         MediaType result = OpenApiUtils.getMediaTypeFromContent(content, "non-app/json");
         Assertions.assertThat(result).isNull();
     }
+
+    @Test
+    void shouldReturnContentTypeWhenContainingCharset() {
+        Content content = Mockito.mock(Content.class);
+        MediaType mediaType = Mockito.mock(MediaType.class);
+        MediaType mediaTypeCharset = Mockito.mock(MediaType.class);
+
+        Mockito.when(content.get("app/json")).thenReturn(mediaType);
+        Mockito.when(content.entrySet()).thenReturn(Set.of(new AbstractMap.SimpleEntry<>("app/json2", mediaTypeCharset)));
+        MediaType actual = OpenApiUtils.getMediaTypeFromContent(content, "app/json2");
+        Assertions.assertThat(actual).isSameAs(mediaTypeCharset).isNotSameAs(mediaType);
+    }
+
+    @Test
+    void shouldReturnTrueWhenContentTypeWithCharset() {
+        Content content = Mockito.mock(Content.class);
+        MediaType mediaType = Mockito.mock(MediaType.class);
+        Mockito.when(content.keySet()).thenReturn(Set.of("application/json;charset=UTF-8"));
+
+        boolean actual = OpenApiUtils.hasContentType(content, "application/json");
+        Assertions.assertThat(actual).isTrue();
+    }
 }