Code Snippet

Just another Code Snippet site

[Java] Mock ServletOutputStream

    class MockServletOutputStream extends ServletOutputStream {

        FileWriter writer;

        public MockServletOutputStream(File file) {
            try {
                writer = new FileWriter(file);
            } catch (IOException e) {
                // TODO
            }
        }

        /**
         * {@inheritDoc}
         */
        @Override
        public void write(int b) throws IOException {
            writer.write(b);
        }

        @Override
        public void close() {
            try {
                super.close();
                writer.flush();
                writer.close();
            } catch (IOException e) {
                // TODO
            }
        }
    }
@Test
public void test1() {

        File output = new File("target/test-classes/output.txt");
        MockServletOutputStream stream = new MockServletOutputStream(output);

        HttpServletResponse response = mock(HttpServletResponse.class);
        when(response.getOutputStream()).thenReturn(stream);



}


Comments are currently closed.