/
ProtocolTest.java
145 lines (135 loc) · 6.21 KB
/
ProtocolTest.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
package org.atmosphere.vibe.server;
import java.io.IOException;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentSkipListSet;
import javax.servlet.ServletContext;
import javax.servlet.ServletContextEvent;
import javax.servlet.ServletContextListener;
import javax.servlet.ServletException;
import javax.servlet.ServletRegistration;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.exec.CommandLine;
import org.apache.commons.exec.DefaultExecutor;
import org.atmosphere.vibe.platform.Action;
import org.atmosphere.vibe.platform.VoidAction;
import org.atmosphere.vibe.platform.server.atmosphere2.AtmosphereBridge;
import org.atmosphere.vibe.server.ServerSocket.Reply;
import org.eclipse.jetty.server.ServerConnector;
import org.eclipse.jetty.servlet.ServletContextHandler;
import org.junit.Test;
public class ProtocolTest {
@Test
public void protocol() throws Exception {
final Set<String> sockets = new ConcurrentSkipListSet<String>();
final DefaultServer server = new DefaultServer();
server.socketAction(new Action<ServerSocket>() {
@Override
public void on(final ServerSocket socket) {
sockets.add(socket.id());
socket.on("close", new VoidAction() {
@Override
public void on() {
sockets.remove(socket.id());
}
})
.on("echo", new Action<Object>() {
@Override
public void on(Object data) {
socket.send("echo", data);
}
})
.on("/reply/inbound", new Action<Reply<Map<String, Object>>>() {
@Override
public void on(Reply<Map<String, Object>> reply) {
Map<String, Object> data = reply.data();
switch ((String) data.get("type")) {
case "resolved":
reply.resolve(data.get("data"));
break;
case "rejected":
reply.reject(data.get("data"));
break;
}
}
})
.on("/reply/outbound", new Action<Map<String, Object>>() {
@Override
public void on(Map<String, Object> data) {
switch ((String) data.get("type")) {
case "resolved":
socket.send("test", data.get("data"), new Action<Object>() {
@Override
public void on(Object data) {
socket.send("done", data);
}
});
break;
case "rejected":
socket.send("test", data.get("data"), null, new Action<Object>() {
@Override
public void on(Object data) {
socket.send("done", data);
}
});
break;
}
}
});
}
});
org.eclipse.jetty.server.Server jetty = new org.eclipse.jetty.server.Server();
ServerConnector connector = new ServerConnector(jetty);
connector.setPort(8000);
jetty.addConnector(connector);
ServletContextHandler handler = new ServletContextHandler();
handler.addEventListener(new ServletContextListener() {
@Override
@SuppressWarnings("serial")
public void contextInitialized(ServletContextEvent event) {
ServletContext context = event.getServletContext();
// /setup
ServletRegistration regSetup = context.addServlet("/setup", new HttpServlet() {
@Override
protected void doGet(HttpServletRequest req, HttpServletResponse res) throws ServletException, IOException {
Map<String, String[]> params = req.getParameterMap();
server.setTransports(params.get("transports")[0].split(","));
if (params.containsKey("heartbeat")) {
server.setHeartbeat(Integer.parseInt(params.get("heartbeat")[0]));
}
if (params.containsKey("_heartbeat")) {
server.set_heartbeat(Integer.parseInt(params.get("_heartbeat")[0]));
}
}
});
regSetup.addMapping("/setup");
// /alive
ServletRegistration regAlive = context.addServlet("/alive", new HttpServlet() {
@Override
protected void doGet(HttpServletRequest req, HttpServletResponse res) throws ServletException, IOException {
res.getWriter().print(sockets.contains(req.getParameter("id")));
}
});
regAlive.addMapping("/alive");
// /vibe
new AtmosphereBridge(context, "/vibe").httpAction(server.httpAction()).websocketAction(server.websocketAction());
}
@Override
public void contextDestroyed(ServletContextEvent sce) {}
});
jetty.setHandler(handler);
jetty.start();
CommandLine cmdLine = CommandLine.parse("./src/test/resources/node/node")
.addArgument("./src/test/resources/runner")
.addArgument("--vibe.transports")
.addArgument("ws,sse,streamxhr,streamxdr,streamiframe,longpollajax,longpollxdr,longpolljsonp")
.addArgument("--vibe.extension")
.addArgument("reply");
DefaultExecutor executor = new DefaultExecutor();
// The exit value of mocha is the number of failed tests.
executor.execute(cmdLine);
jetty.stop();
}
}