Moving code for multiple-http-servlet-request article from spring-core to spring-mvc (#8329)

2019-12-10 18:37:06 +00:00
parent 96d4c3c95b
commit c0a4e8019b
12 changed files with 0 additions and 0 deletions
@@ -0,0 +1,62 @@
+package org.baeldung.cachedrequest;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStream;
+
+import org.junit.After;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.springframework.mock.web.MockHttpServletRequest;
+import org.springframework.util.StreamUtils;
+
+import junit.framework.TestCase;
+
+@RunWith(MockitoJUnitRunner.class)
+public class CachedBodyHttpServletRequestUnitTest extends TestCase {
+
+    private CachedBodyServletInputStream servletInputStream;
+
+    @After
+    public void cleanUp() throws IOException {
+        if (null != servletInputStream) {
+            servletInputStream.close();
+        }
+    }
+
+    @Test
+    public void testGivenHttpServletRequestWithBody_whenCalledGetInputStream_ThenGetsServletInputStreamWithSameBody() throws IOException {
+        // Given
+        byte[] cachedBody = "{\"firstName\" :\"abc\",\"lastName\" : \"xyz\",\"age\" : 30\"}".getBytes();
+        MockHttpServletRequest mockeddHttpServletRequest = new MockHttpServletRequest();
+        mockeddHttpServletRequest.setContent(cachedBody);
+        CachedBodyHttpServletRequest request = new CachedBodyHttpServletRequest(mockeddHttpServletRequest);
+
+        // when
+        InputStream inputStream = request.getInputStream();
+
+        // then
+        assertEquals(new String(cachedBody), new String(StreamUtils.copyToByteArray(inputStream)));
+    }
+
+    @Test
+    public void testGivenHttpServletRequestWithBody_whenCalledGetReader_ThenGetBufferedReaderWithSameBody() throws IOException {
+        // Given
+        byte[] cachedBody = "{\"firstName\" :\"abc\",\"lastName\" : \"xyz\",\"age\" : 30\"}".getBytes();
+        MockHttpServletRequest mockeddHttpServletRequest = new MockHttpServletRequest();
+        mockeddHttpServletRequest.setContent(cachedBody);
+        CachedBodyHttpServletRequest request = new CachedBodyHttpServletRequest(mockeddHttpServletRequest);
+
+        // when
+        BufferedReader bufferedReader = request.getReader();
+
+        // then
+        String line = "";
+        StringBuilder builder = new StringBuilder();
+        while ((line = bufferedReader.readLine()) != null) {
+            builder.append(line);
+        }
+        assertEquals(new String(cachedBody), builder.toString());
+    }
+}
@@ -0,0 +1,98 @@
+package org.baeldung.cachedrequest;
+
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+
+import javax.servlet.ReadListener;
+
+import org.junit.After;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mockito;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.springframework.util.StreamUtils;
+
+import junit.framework.TestCase;
+
+@RunWith(MockitoJUnitRunner.class)
+public class CachedBodyServletInputStreamUnitTest extends TestCase {
+
+    private CachedBodyServletInputStream servletInputStream;
+
+    @After
+    public void cleanUp() throws IOException {
+        if (null != servletInputStream) {
+            servletInputStream.close();
+        }
+    }
+
+    @Test
+    public void testGivenServletInputStreamCreated_whenCalledisFinished_Thenfalse() {
+        // Given
+        byte[] cachedBody = "{\"firstName\" :\"abc\",\"lastName\" : \"xyz\",\"age\" : 30\"}".getBytes();
+        servletInputStream = new CachedBodyServletInputStream(cachedBody);
+
+        // when
+        boolean finished = servletInputStream.isFinished();
+
+        // then
+        assertFalse(finished);
+    }
+
+    @Test
+    public void testGivenServletInputStreamCreatedAndBodyRead_whenCalledisFinished_ThenTrue() throws IOException {
+        // Given
+        byte[] cachedBody = "{\"firstName\" :\"abc\",\"lastName\" : \"xyz\",\"age\" : 30\"}".getBytes();
+        servletInputStream = new CachedBodyServletInputStream(cachedBody);
+        StreamUtils.copyToByteArray(servletInputStream);
+
+        // when
+        boolean finished = servletInputStream.isFinished();
+
+        // then
+        assertTrue(finished);
+    }
+
+    @Test
+    public void testGivenServletInputStreamCreatedAndBodyRead_whenCalledIsReady_ThenTrue() throws IOException {
+        // Given
+        byte[] cachedBody = "{\"firstName\" :\"abc\",\"lastName\" : \"xyz\",\"age\" : 30\"}".getBytes();
+        servletInputStream = new CachedBodyServletInputStream(cachedBody);
+
+        // when
+        boolean ready = servletInputStream.isReady();
+
+        // then
+        assertTrue(ready);
+    }
+
+    @Test
+    public void testGivenServletInputStreamCreated_whenCalledIsRead_ThenReturnsBody() throws IOException {
+        // Given
+        byte[] cachedBody = "{\"firstName\" :\"abc\",\"lastName\" : \"xyz\",\"age\" : 30\"}".getBytes();
+        servletInputStream = new CachedBodyServletInputStream(cachedBody);
+
+        // when
+        ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
+        int len = 0;
+        byte[] buffer = new byte[1024];
+        while ((len = servletInputStream.read(buffer)) != -1) {
+            byteArrayOutputStream.write(buffer, 0, len);
+        }
+
+        // then
+        assertEquals(new String(cachedBody), new String(byteArrayOutputStream.toByteArray()));
+    }
+
+    @Test(expected = UnsupportedOperationException.class)
+    public void testGivenServletInputStreamCreated_whenCalledIsRead_ThenThrowsException() throws IOException {
+        // Given
+        byte[] cachedBody = "{\"firstName\" :\"abc\",\"lastName\" : \"xyz\",\"age\" : 30\"}".getBytes();
+        servletInputStream = new CachedBodyServletInputStream(cachedBody);
+
+        // when
+        servletInputStream.setReadListener(Mockito.mock(ReadListener.class));
+
+    }
+
+}
@@ -0,0 +1,39 @@
+package org.baeldung.cachedrequest;
+
+import java.io.IOException;
+
+import javax.servlet.FilterChain;
+import javax.servlet.ServletException;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mockito;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.springframework.mock.web.MockHttpServletRequest;
+import org.springframework.mock.web.MockHttpServletResponse;
+
+import junit.framework.TestCase;
+
+@RunWith(MockitoJUnitRunner.class)
+public class ContentCachingFilterUnitTest extends TestCase {
+
+    @InjectMocks
+    private ContentCachingFilter filterToTest;
+
+    @Test
+    public void testGivenHttpRequest_WhenDoFilter_thenCreatesRequestWrapperObject() throws IOException, ServletException {
+        // Given
+        MockHttpServletRequest mockedRequest = new MockHttpServletRequest();
+        MockHttpServletResponse mockedResponse = new MockHttpServletResponse();
+        FilterChain mockedFilterChain = Mockito.mock(FilterChain.class);
+
+        // when
+        filterToTest.doFilter(mockedRequest, mockedResponse, mockedFilterChain);
+
+        // then
+        Mockito.verify(mockedFilterChain, Mockito.times(1))
+            .doFilter(Mockito.any(CachedBodyHttpServletRequest.class), Mockito.any(MockHttpServletResponse.class));
+    }
+
+}
@@ -0,0 +1,64 @@
+package org.baeldung.cachedrequest;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.MediaType;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit4.SpringRunner;
+import org.springframework.test.context.web.AnnotationConfigWebContextLoader;
+import org.springframework.test.context.web.WebAppConfiguration;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.ResultActions;
+import org.springframework.test.web.servlet.setup.MockMvcBuilders;
+import org.springframework.web.context.WebApplicationContext;
+
+import java.io.IOException;
+
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+
+@RunWith(SpringRunner.class)
+@ContextConfiguration(loader = AnnotationConfigWebContextLoader.class, classes = { HttpRequestDemoConfig.class, ContentCachingFilter.class, PrintRequestContentFilter.class, PersonController.class })
+@WebAppConfiguration
+public class PersonControllerIntegrationTest {
+
+    @Autowired
+    private WebApplicationContext wac;
+
+    private MockMvc mockMvc;
+
+    ObjectMapper objectMapper = new ObjectMapper();
+
+    @Autowired
+    private ContentCachingFilter contentCachingFilter;
+
+    @Autowired
+    private PrintRequestContentFilter printRequestContentFilter;
+
+    @Before
+    public void setup() throws Exception {
+
+        this.mockMvc = MockMvcBuilders.webAppContextSetup(this.wac)
+                .addFilter(contentCachingFilter, "/**")
+                .addFilter(printRequestContentFilter, "/**")
+                .build();
+    }
+
+    @Test
+    public void whenValidInput_thenCreateBook() throws IOException, Exception {
+        // assign - given
+        Person person = new Person("sumit", "abc", 100);
+
+        // act - when
+        ResultActions result = mockMvc.perform(post("/person").accept(MediaType.APPLICATION_JSON)
+                .contentType(MediaType.APPLICATION_JSON)
+                .content(objectMapper.writeValueAsString(person)));
+
+        // assert - then
+        result.andExpect(status().isNoContent());
+    }
+
+}
@@ -0,0 +1,40 @@
+package org.baeldung.cachedrequest;
+
+import java.io.IOException;
+
+import javax.servlet.FilterChain;
+import javax.servlet.ServletException;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mockito;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.springframework.mock.web.MockHttpServletRequest;
+import org.springframework.mock.web.MockHttpServletResponse;
+
+import junit.framework.TestCase;
+
+@RunWith(MockitoJUnitRunner.class)
+public class PrintRequestContentFilterUnitTest extends TestCase {
+
+    @InjectMocks
+    private PrintRequestContentFilter filterToTest;
+
+    @Test
+    public void testGivenHttpRequest_WhenDoFilter_thenReadsBody() throws IOException, ServletException {
+        // Given
+        MockHttpServletRequest mockedRequest = new MockHttpServletRequest();
+        MockHttpServletResponse mockedResponse = new MockHttpServletResponse();
+        FilterChain mockedFilterChain = Mockito.mock(FilterChain.class);
+        CachedBodyHttpServletRequest cachedBodyHttpServletRequest = new CachedBodyHttpServletRequest(mockedRequest);
+
+        // when
+        filterToTest.doFilter(cachedBodyHttpServletRequest, mockedResponse, mockedFilterChain);
+
+        // then
+        Mockito.verify(mockedFilterChain, Mockito.times(1))
+            .doFilter(cachedBodyHttpServletRequest, mockedResponse);
+    }
+
+}