1 package org.cateproject.view.mock;
2
3 import java.io.IOException;
4
5 import javax.servlet.RequestDispatcher;
6 import javax.servlet.ServletException;
7 import javax.servlet.ServletRequest;
8 import javax.servlet.ServletResponse;
9
10 import org.springframework.mock.web.MockHttpServletRequest;
11 import org.springframework.web.servlet.DispatcherServlet;
12
13 public class MockRequestDispatcher implements RequestDispatcher {
14
15 private String resource;
16 private DispatcherServlet dispatcherServlet;
17
18 public MockRequestDispatcher(DispatcherServlet dispatcherServlet) {
19 this.dispatcherServlet = dispatcherServlet;
20 }
21
22 public void setResource(String resource) {
23 this.resource = resource;
24 }
25
26 public void forward(ServletRequest request, ServletResponse response) throws ServletException, IOException {
27
28 }
29
30 public void include(ServletRequest request, ServletResponse response) throws ServletException, IOException {
31 try {
32 MockHttpServletRequest mockRequest = (MockHttpServletRequest) request;
33 mockRequest.setRequestURI(resource);
34 dispatcherServlet.service(mockRequest, response);
35 } catch(Exception e) {
36 throw new ServletException(e);
37 }
38
39 }
40
41 }