1 package com.atlassian.johnson.filters;
2
3 import org.junit.Test;
4
5 import javax.servlet.http.HttpServletResponse;
6 import java.io.IOException;
7 import java.io.PrintWriter;
8
9 import static org.easymock.EasyMock.*;
10
11 public class Johnson503FilterTest
12 {
13 @Test
14 public void testHandleError() throws IOException
15 {
16 PrintWriter writer = createStrictMock(PrintWriter.class);
17 writer.flush();
18
19 HttpServletResponse response = createStrictMock(HttpServletResponse.class);
20 response.setStatus(eq(HttpServletResponse.SC_SERVICE_UNAVAILABLE));
21 expect(response.getWriter()).andReturn(writer);
22
23 replay(response, writer);
24
25 Johnson503Filter filter = new Johnson503Filter();
26 filter.handleError(null, null, response);
27
28 verify(response, writer);
29 }
30
31 @Test
32 public void testHandleNotSetup() throws IOException
33 {
34 PrintWriter writer = createStrictMock(PrintWriter.class);
35 writer.flush();
36
37 HttpServletResponse response = createStrictMock(HttpServletResponse.class);
38 response.setStatus(eq(HttpServletResponse.SC_SERVICE_UNAVAILABLE));
39 expect(response.getWriter()).andReturn(writer);
40
41 replay(response, writer);
42
43 Johnson503Filter johnson503Filter = new Johnson503Filter();
44 johnson503Filter.handleNotSetup(null, response);
45
46 verify(response, writer);
47 }
48 }