BAEL-5715 Updated Code

2022-10-14 20:16:00 +05:30
parent 97dbab9102
commit 083b1b42d7
4 changed files with 112 additions and 0 deletions
@@ -0,0 +1,43 @@
+package com.baeldung.postman.controller;
+
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.content;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.http.MediaType;
+import org.springframework.mock.web.MockMultipartFile;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
+
+import com.baeldung.postman.PostmanUploadDemo;
+import com.baeldung.postman.model.JsonRequest;
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+@SpringBootTest(classes = PostmanUploadDemo.class)
+@AutoConfigureMockMvc
+public class PostmanUploadControllerTest {
+    @Autowired
+    private MockMvc mockMvc;
+
+    @Test
+    public void givenJson_whenUploaded_thenSuccessReturned() throws Exception {
+        JsonRequest request = new JsonRequest(1, "John Doe");
+        this.mockMvc.perform(MockMvcRequestBuilders.post("/uploadJson")
+            .contentType(MediaType.APPLICATION_JSON)
+            .content(new ObjectMapper().writeValueAsString(request)))
+          .andExpect(status().isOk())
+          .andExpect(content().string("1John Doe"));
+    }
+
+    @Test
+    public void givenFile_whenUploaded_thenSuccessReturned() throws Exception {
+        MockMultipartFile request = new MockMultipartFile("dummy", "{\"key\": \"value\"}".getBytes());
+        this.mockMvc.perform(MockMvcRequestBuilders.multipart("/uploadFile")
+            .file("file", request.getBytes()))
+          .andExpect(status().isOk())
+          .andExpect(content().string("file received successfully"));
+    }
+}