1
2
3
4
5
6
7
8
9
10
11
12
13
14 package org.eclipse.jetty.server.session;
15
16 import java.io.IOException;
17 import java.util.Collections;
18 import java.util.Random;
19
20 import javax.servlet.RequestDispatcher;
21 import javax.servlet.ServletContext;
22 import javax.servlet.ServletException;
23 import javax.servlet.http.HttpServlet;
24 import javax.servlet.http.HttpServletRequest;
25 import javax.servlet.http.HttpServletResponse;
26 import javax.servlet.http.HttpSession;
27
28 import org.eclipse.jetty.client.ContentExchange;
29 import org.eclipse.jetty.client.HttpClient;
30 import org.eclipse.jetty.http.HttpMethods;
31 import org.eclipse.jetty.servlet.ServletContextHandler;
32 import org.junit.Test;
33
34
35
36
37 public abstract class AbstractServerCrossContextSessionTest
38 {
39
40 public abstract AbstractTestServer createServer(int port);
41
42 @Test
43 public void testCrossContextDispatch() throws Exception
44 {
45 Random random = new Random(System.nanoTime());
46
47 String contextA = "/contextA";
48 String contextB = "/contextB";
49 String servletMapping = "/server";
50 int port = random.nextInt(50000) + 10000;
51 AbstractTestServer server = createServer(port);
52 ServletContextHandler ctxA = server.addContext(contextA);
53 ctxA.addServlet(TestServletA.class, servletMapping);
54 ServletContextHandler ctxB = server.addContext(contextB);
55 ctxB.addServlet(TestServletB.class, servletMapping);
56 server.start();
57 try
58 {
59 HttpClient client = new HttpClient();
60 client.setConnectorType(HttpClient.CONNECTOR_SOCKET);
61 client.start();
62 try
63 {
64
65 ContentExchange exchange = new ContentExchange(true);
66 exchange.setMethod(HttpMethods.GET);
67 exchange.setURL("http://localhost:" + port + contextA + servletMapping);
68 client.send(exchange);
69 exchange.waitForDone();
70 assert exchange.getResponseStatus() == HttpServletResponse.SC_OK;
71 }
72 finally
73 {
74 client.stop();
75 }
76 }
77 finally
78 {
79 server.stop();
80 }
81 }
82
83 public static class TestServletA extends HttpServlet
84 {
85 @Override
86 protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException
87 {
88 HttpSession session = request.getSession(false);
89 if (session == null) session = request.getSession(true);
90
91
92 session.setAttribute("A", "A");
93 System.out.println("A: session.getAttributeNames() = " + Collections.list(session.getAttributeNames()));
94
95
96
97 ServletContext contextB = getServletContext().getContext("/contextB");
98 RequestDispatcher dispatcherB = contextB.getRequestDispatcher(request.getServletPath());
99 dispatcherB.forward(request, response);
100
101
102 Object objectB = session.getAttribute("B");
103 assert objectB == null;
104 System.out.println("A: session.getAttributeNames() = " + Collections.list(session.getAttributeNames()));
105 }
106 }
107
108 public static class TestServletB extends HttpServlet
109 {
110 @Override
111 protected void doGet(HttpServletRequest request, HttpServletResponse httpServletResponse) throws ServletException, IOException
112 {
113 HttpSession session = request.getSession(false);
114 if (session == null) session = request.getSession(true);
115
116
117 Object objectA = session.getAttribute("A");
118 assert objectA == null;
119
120
121 session.setAttribute("B", "B");
122 System.out.println("B: session.getAttributeNames() = " + Collections.list(session.getAttributeNames()));
123 }
124 }
125 }