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.io.PrintWriter;
18 import java.util.Random;
19
20 import javax.servlet.ServletException;
21 import javax.servlet.http.HttpServlet;
22 import javax.servlet.http.HttpServletRequest;
23 import javax.servlet.http.HttpServletResponse;
24 import javax.servlet.http.HttpSession;
25
26 import org.eclipse.jetty.client.ContentExchange;
27 import org.eclipse.jetty.client.HttpClient;
28 import org.eclipse.jetty.http.HttpMethods;
29 import org.junit.Test;
30
31
32
33
34 public abstract class AbstractSessionMigrationTest
35 {
36 public abstract AbstractTestServer createServer (int port);
37
38 @Test
39 public void testSessionMigration() throws Exception
40 {
41 Random random = new Random(System.nanoTime());
42
43 String contextPath = "";
44 String servletMapping = "/server";
45 int port1 = random.nextInt(50000) + 10000;
46 AbstractTestServer server1 = createServer(port1);
47 server1.addContext(contextPath).addServlet(TestServlet.class, servletMapping);
48 server1.start();
49 try
50 {
51 int port2 = random.nextInt(50000) + 10000;
52 AbstractTestServer server2 = createServer(port2);
53 server2.addContext(contextPath).addServlet(TestServlet.class, servletMapping);
54 server2.start();
55 try
56 {
57 HttpClient client = new HttpClient();
58 client.setConnectorType(HttpClient.CONNECTOR_SOCKET);
59 client.start();
60 try
61 {
62
63 int value = 1;
64 ContentExchange exchange1 = new ContentExchange(true);
65 exchange1.setMethod(HttpMethods.POST);
66 exchange1.setURL("http://localhost:" + port1 + contextPath + servletMapping + "?action=set&value=" + value);
67 client.send(exchange1);
68 exchange1.waitForDone();
69 assert exchange1.getResponseStatus() == HttpServletResponse.SC_OK;
70 String sessionCookie = exchange1.getResponseFields().getStringField("Set-Cookie");
71 assert sessionCookie != null;
72
73 sessionCookie = sessionCookie.replaceFirst("(\\W)(P|p)ath=", "$1\\$Path=");
74
75
76
77 ContentExchange exchange2 = new ContentExchange(true);
78 exchange2.setMethod(HttpMethods.GET);
79 exchange2.setURL("http://localhost:" + port2 + contextPath + servletMapping + "?action=get");
80 exchange2.getRequestFields().add("Cookie", sessionCookie);
81 client.send(exchange2);
82 exchange2.waitForDone();
83 assert exchange2.getResponseStatus() == HttpServletResponse.SC_OK;
84 String response = exchange2.getResponseContent();
85 assert response.trim().equals(String.valueOf(value));
86 }
87 finally
88 {
89 client.stop();
90 }
91 }
92 finally
93 {
94 server2.stop();
95 }
96 }
97 finally
98 {
99 server1.stop();
100 }
101 }
102
103 public static class TestServlet extends HttpServlet
104 {
105 @Override
106 protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException
107 {
108 doPost(request, response);
109 }
110
111 @Override
112 protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException
113 {
114 HttpSession session = request.getSession(false);
115 if (session == null) session = request.getSession(true);
116
117 String action = request.getParameter("action");
118 if ("set".equals(action))
119 {
120 int value = Integer.parseInt(request.getParameter("value"));
121 session.setAttribute("value", value);
122 PrintWriter writer = response.getWriter();
123 writer.println(value);
124 writer.flush();
125 }
126 else if ("get".equals(action))
127 {
128 int value = (Integer)session.getAttribute("value");
129 PrintWriter writer = response.getWriter();
130 writer.println(value);
131 writer.flush();
132 }
133 }
134 }
135 }