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