Handle post errors

1 files changed, 21 insertions, 0 deletions

diff --git a/src/test/java/com/juick/server/tests/ServerTests.java b/src/test/java/com/juick/server/tests/ServerTests.java
index 418526c7..e0abe136 100644
--- a/src/test/java/com/juick/server/tests/ServerTests.java
+++ b/src/test/java/com/juick/server/tests/ServerTests.java
@@ -2749,4 +2749,25 @@ public class ServerTests {
         var replies = message.getReplies();
         assertThat(replies.size(), is(1));
     }
+    @Test
+    @Transactional
+    public void unsupportedMediaShouldBeHandledCorrectly() throws Exception {
+        ClassPathResource newMedia = new ClassPathResource("sample1.dng");
+        byte[] newMediaData = IOUtils.toByteArray(newMedia.getInputStream());
+        var response = mockMvc.perform(MockMvcRequestBuilders.multipart("/api/post")
+                .file(new MockMultipartFile("attach", "sample1.dng", "image/dng", newMediaData))
+                .param("body", "test")
+                .with(httpBasic(freefdName, freefdPassword))).andExpect(status().isBadRequest()).andReturn();
+        var result = jsonMapper.readValue(response.getResponse().getContentAsString(), CommandResult.class);
+        assertThat(result.getText(), is("Wrong file type: tif"));
+        var r = commandsManager.processCommand(freefd, "tst", emptyUri);
+        response = mockMvc.perform(MockMvcRequestBuilders.multipart("/api/comment")
+                .file(new MockMultipartFile("attach", "sample1.dng", "image/dng", newMediaData))
+                .param("body", "test")
+                .param("mid", String.valueOf(r.getNewMessage().get().getMid()))
+                .with(httpBasic(freefdName, freefdPassword))).andExpect(status().isBadRequest()).andReturn();
+        result = jsonMapper.readValue(response.getResponse().getContentAsString(), CommandResult.class);
+        assertThat(result.getText(), is("Wrong file type: tif"));
+
+    }
 }