|
1 /* |
|
2 * Copyright (c) 2017, Oracle and/or its affiliates. All rights reserved. |
|
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
|
4 * |
|
5 * This code is free software; you can redistribute it and/or modify it |
|
6 * under the terms of the GNU General Public License version 2 only, as |
|
7 * published by the Free Software Foundation. |
|
8 * |
|
9 * This code is distributed in the hope that it will be useful, but WITHOUT |
|
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
|
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
|
12 * version 2 for more details (a copy is included in the LICENSE file that |
|
13 * accompanied this code). |
|
14 * |
|
15 * You should have received a copy of the GNU General Public License version |
|
16 * 2 along with this work; if not, write to the Free Software Foundation, |
|
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
|
18 * |
|
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
|
20 * or visit www.oracle.com if you need additional information or have any |
|
21 * questions. |
|
22 */ |
|
23 |
1 package jdk.incubator.http.internal.websocket; |
24 package jdk.incubator.http.internal.websocket; |
2 |
25 |
3 import jdk.incubator.http.WebSocket; |
26 import jdk.incubator.http.WebSocket; |
4 import jdk.incubator.http.WebSocket.MessagePart; |
27 import jdk.incubator.http.WebSocket.MessagePart; |
5 |
28 |
6 import java.nio.ByteBuffer; |
29 import java.nio.ByteBuffer; |
|
30 import java.util.ArrayList; |
|
31 import java.util.List; |
|
32 import java.util.Objects; |
|
33 import java.util.concurrent.CompletableFuture; |
7 import java.util.concurrent.CompletionStage; |
34 import java.util.concurrent.CompletionStage; |
|
35 |
|
36 import static jdk.incubator.http.internal.websocket.TestSupport.fullCopy; |
8 |
37 |
9 public class MockListener implements WebSocket.Listener { |
38 public class MockListener implements WebSocket.Listener { |
10 |
39 |
11 private final long bufferSize; |
40 private final long bufferSize; |
12 private long count; |
41 private long count; |
|
42 private final List<ListenerInvocation> invocations = new ArrayList<>(); |
|
43 private final CompletableFuture<?> lastCall = new CompletableFuture<>(); |
13 |
44 |
14 /* |
45 /* |
15 * Typical buffer sizes: 1, n, Long.MAX_VALUE |
46 * Typical buffer sizes: 1, n, Long.MAX_VALUE |
16 */ |
47 */ |
17 public MockListener(long bufferSize) { |
48 public MockListener(long bufferSize) { |
22 } |
53 } |
23 |
54 |
24 @Override |
55 @Override |
25 public void onOpen(WebSocket webSocket) { |
56 public void onOpen(WebSocket webSocket) { |
26 System.out.printf("onOpen(%s)%n", webSocket); |
57 System.out.printf("onOpen(%s)%n", webSocket); |
27 replenishDemandIfNeeded(webSocket); |
58 invocations.add(new OnOpen(webSocket)); |
|
59 onOpen0(webSocket); |
|
60 } |
|
61 |
|
62 protected void onOpen0(WebSocket webSocket) { |
|
63 replenish(webSocket); |
28 } |
64 } |
29 |
65 |
30 @Override |
66 @Override |
31 public CompletionStage<?> onText(WebSocket webSocket, |
67 public CompletionStage<?> onText(WebSocket webSocket, |
32 CharSequence message, |
68 CharSequence message, |
33 MessagePart part) { |
69 MessagePart part) { |
34 System.out.printf("onText(%s, %s, %s)%n", webSocket, message, part); |
70 System.out.printf("onText(%s, %s, %s)%n", webSocket, message, part); |
35 replenishDemandIfNeeded(webSocket); |
71 invocations.add(new OnText(webSocket, message.toString(), part)); |
|
72 return onText0(webSocket, message, part); |
|
73 } |
|
74 |
|
75 protected CompletionStage<?> onText0(WebSocket webSocket, |
|
76 CharSequence message, |
|
77 MessagePart part) { |
|
78 replenish(webSocket); |
36 return null; |
79 return null; |
37 } |
80 } |
38 |
81 |
39 @Override |
82 @Override |
40 public CompletionStage<?> onBinary(WebSocket webSocket, |
83 public CompletionStage<?> onBinary(WebSocket webSocket, |
41 ByteBuffer message, |
84 ByteBuffer message, |
42 MessagePart part) { |
85 MessagePart part) { |
43 System.out.printf("onBinary(%s, %s, %s)%n", webSocket, message, part); |
86 System.out.printf("onBinary(%s, %s, %s)%n", webSocket, message, part); |
44 replenishDemandIfNeeded(webSocket); |
87 invocations.add(new OnBinary(webSocket, fullCopy(message), part)); |
|
88 return onBinary0(webSocket, message, part); |
|
89 } |
|
90 |
|
91 protected CompletionStage<?> onBinary0(WebSocket webSocket, |
|
92 ByteBuffer message, |
|
93 MessagePart part) { |
|
94 replenish(webSocket); |
45 return null; |
95 return null; |
46 } |
96 } |
47 |
97 |
48 @Override |
98 @Override |
49 public CompletionStage<?> onPing(WebSocket webSocket, ByteBuffer message) { |
99 public CompletionStage<?> onPing(WebSocket webSocket, ByteBuffer message) { |
50 System.out.printf("onPing(%s, %s)%n", webSocket, message); |
100 System.out.printf("onPing(%s, %s)%n", webSocket, message); |
51 replenishDemandIfNeeded(webSocket); |
101 invocations.add(new OnPing(webSocket, fullCopy(message))); |
|
102 return onPing0(webSocket, message); |
|
103 } |
|
104 |
|
105 protected CompletionStage<?> onPing0(WebSocket webSocket, ByteBuffer message) { |
|
106 replenish(webSocket); |
52 return null; |
107 return null; |
53 } |
108 } |
54 |
109 |
55 @Override |
110 @Override |
56 public CompletionStage<?> onPong(WebSocket webSocket, ByteBuffer message) { |
111 public CompletionStage<?> onPong(WebSocket webSocket, ByteBuffer message) { |
57 System.out.printf("onPong(%s, %s)%n", webSocket, message); |
112 System.out.printf("onPong(%s, %s)%n", webSocket, message); |
58 replenishDemandIfNeeded(webSocket); |
113 invocations.add(new OnPong(webSocket, fullCopy(message))); |
|
114 return onPong0(webSocket, message); |
|
115 } |
|
116 |
|
117 protected CompletionStage<?> onPong0(WebSocket webSocket, ByteBuffer message) { |
|
118 replenish(webSocket); |
59 return null; |
119 return null; |
60 } |
120 } |
61 |
121 |
62 @Override |
122 @Override |
63 public CompletionStage<?> onClose(WebSocket webSocket, |
123 public CompletionStage<?> onClose(WebSocket webSocket, |
64 int statusCode, |
124 int statusCode, |
65 String reason) { |
125 String reason) { |
66 System.out.printf("onClose(%s, %s, %s)%n", webSocket, statusCode, reason); |
126 System.out.printf("onClose(%s, %s, %s)%n", webSocket, statusCode, reason); |
|
127 invocations.add(new OnClose(webSocket, statusCode, reason)); |
|
128 lastCall.complete(null); |
67 return null; |
129 return null; |
68 } |
130 } |
69 |
131 |
70 @Override |
132 @Override |
71 public void onError(WebSocket webSocket, Throwable error) { |
133 public void onError(WebSocket webSocket, Throwable error) { |
72 System.out.printf("onError(%s, %s)%n", webSocket, error); |
134 System.out.printf("onError(%s, %s)%n", webSocket, error); |
73 } |
135 invocations.add(new OnError(webSocket, error == null ? null : error.getClass())); |
74 |
136 lastCall.complete(null); |
75 private void replenishDemandIfNeeded(WebSocket webSocket) { |
137 } |
|
138 |
|
139 public CompletableFuture<?> onCloseOrOnErrorCalled() { |
|
140 return lastCall.copy(); |
|
141 } |
|
142 |
|
143 protected void replenish(WebSocket webSocket) { |
76 if (--count <= 0) { |
144 if (--count <= 0) { |
77 count = bufferSize - bufferSize / 2; |
145 count = bufferSize - bufferSize / 2; |
78 System.out.printf("request(%s)%n", count); |
146 } |
79 webSocket.request(count); |
147 webSocket.request(count); |
|
148 } |
|
149 |
|
150 public List<ListenerInvocation> invocations() { |
|
151 return new ArrayList<>(invocations); |
|
152 } |
|
153 |
|
154 public abstract static class ListenerInvocation { |
|
155 |
|
156 public static OnOpen onOpen(WebSocket webSocket) { |
|
157 return new OnOpen(webSocket); |
|
158 } |
|
159 |
|
160 public static OnText onText(WebSocket webSocket, |
|
161 String text, |
|
162 MessagePart part) { |
|
163 return new OnText(webSocket, text, part); |
|
164 } |
|
165 |
|
166 public static OnBinary onBinary(WebSocket webSocket, |
|
167 ByteBuffer data, |
|
168 MessagePart part) { |
|
169 return new OnBinary(webSocket, data, part); |
|
170 } |
|
171 |
|
172 public static OnPing onPing(WebSocket webSocket, |
|
173 ByteBuffer data) { |
|
174 return new OnPing(webSocket, data); |
|
175 } |
|
176 |
|
177 public static OnPong onPong(WebSocket webSocket, |
|
178 ByteBuffer data) { |
|
179 return new OnPong(webSocket, data); |
|
180 } |
|
181 |
|
182 public static OnClose onClose(WebSocket webSocket, |
|
183 int statusCode, |
|
184 String reason) { |
|
185 return new OnClose(webSocket, statusCode, reason); |
|
186 } |
|
187 |
|
188 public static OnError onError(WebSocket webSocket, |
|
189 Class<? extends Throwable> clazz) { |
|
190 return new OnError(webSocket, clazz); |
|
191 } |
|
192 |
|
193 final WebSocket webSocket; |
|
194 |
|
195 private ListenerInvocation(WebSocket webSocket) { |
|
196 this.webSocket = webSocket; |
|
197 } |
|
198 } |
|
199 |
|
200 public static final class OnOpen extends ListenerInvocation { |
|
201 |
|
202 public OnOpen(WebSocket webSocket) { |
|
203 super(webSocket); |
|
204 } |
|
205 |
|
206 @Override |
|
207 public boolean equals(Object o) { |
|
208 if (this == o) return true; |
|
209 if (o == null || getClass() != o.getClass()) return false; |
|
210 ListenerInvocation that = (ListenerInvocation) o; |
|
211 return Objects.equals(webSocket, that.webSocket); |
|
212 } |
|
213 |
|
214 @Override |
|
215 public int hashCode() { |
|
216 return Objects.hashCode(webSocket); |
|
217 } |
|
218 } |
|
219 |
|
220 public static final class OnText extends ListenerInvocation { |
|
221 |
|
222 final String text; |
|
223 final MessagePart part; |
|
224 |
|
225 public OnText(WebSocket webSocket, String text, MessagePart part) { |
|
226 super(webSocket); |
|
227 this.text = text; |
|
228 this.part = part; |
|
229 } |
|
230 |
|
231 @Override |
|
232 public boolean equals(Object o) { |
|
233 if (this == o) return true; |
|
234 if (o == null || getClass() != o.getClass()) return false; |
|
235 OnText onText = (OnText) o; |
|
236 return Objects.equals(text, onText.text) && |
|
237 part == onText.part && |
|
238 Objects.equals(webSocket, onText.webSocket); |
|
239 } |
|
240 |
|
241 @Override |
|
242 public int hashCode() { |
|
243 return Objects.hash(text, part, webSocket); |
|
244 } |
|
245 } |
|
246 |
|
247 public static final class OnBinary extends ListenerInvocation { |
|
248 |
|
249 final ByteBuffer data; |
|
250 final MessagePart part; |
|
251 |
|
252 public OnBinary(WebSocket webSocket, ByteBuffer data, MessagePart part) { |
|
253 super(webSocket); |
|
254 this.data = data; |
|
255 this.part = part; |
|
256 } |
|
257 |
|
258 @Override |
|
259 public boolean equals(Object o) { |
|
260 if (this == o) return true; |
|
261 if (o == null || getClass() != o.getClass()) return false; |
|
262 OnBinary onBinary = (OnBinary) o; |
|
263 return Objects.equals(data, onBinary.data) && |
|
264 part == onBinary.part && |
|
265 Objects.equals(webSocket, onBinary.webSocket); |
|
266 } |
|
267 |
|
268 @Override |
|
269 public int hashCode() { |
|
270 return Objects.hash(data, part, webSocket); |
|
271 } |
|
272 } |
|
273 |
|
274 public static final class OnPing extends ListenerInvocation { |
|
275 |
|
276 final ByteBuffer data; |
|
277 |
|
278 public OnPing(WebSocket webSocket, ByteBuffer data) { |
|
279 super(webSocket); |
|
280 this.data = data; |
|
281 } |
|
282 |
|
283 @Override |
|
284 public boolean equals(Object o) { |
|
285 if (this == o) return true; |
|
286 if (o == null || getClass() != o.getClass()) return false; |
|
287 OnPing onPing = (OnPing) o; |
|
288 return Objects.equals(data, onPing.data) && |
|
289 Objects.equals(webSocket, onPing.webSocket); |
|
290 } |
|
291 |
|
292 @Override |
|
293 public int hashCode() { |
|
294 return Objects.hash(data, webSocket); |
|
295 } |
|
296 } |
|
297 |
|
298 public static final class OnPong extends ListenerInvocation { |
|
299 |
|
300 final ByteBuffer data; |
|
301 |
|
302 public OnPong(WebSocket webSocket, ByteBuffer data) { |
|
303 super(webSocket); |
|
304 this.data = data; |
|
305 } |
|
306 |
|
307 @Override |
|
308 public boolean equals(Object o) { |
|
309 if (this == o) return true; |
|
310 if (o == null || getClass() != o.getClass()) return false; |
|
311 OnPong onPong = (OnPong) o; |
|
312 return Objects.equals(data, onPong.data) && |
|
313 Objects.equals(webSocket, onPong.webSocket); |
|
314 } |
|
315 |
|
316 @Override |
|
317 public int hashCode() { |
|
318 return Objects.hash(data, webSocket); |
|
319 } |
|
320 } |
|
321 |
|
322 public static final class OnClose extends ListenerInvocation { |
|
323 |
|
324 final int statusCode; |
|
325 final String reason; |
|
326 |
|
327 public OnClose(WebSocket webSocket, int statusCode, String reason) { |
|
328 super(webSocket); |
|
329 this.statusCode = statusCode; |
|
330 this.reason = reason; |
|
331 } |
|
332 |
|
333 @Override |
|
334 public boolean equals(Object o) { |
|
335 if (this == o) return true; |
|
336 if (o == null || getClass() != o.getClass()) return false; |
|
337 OnClose onClose = (OnClose) o; |
|
338 return statusCode == onClose.statusCode && |
|
339 Objects.equals(reason, onClose.reason) && |
|
340 Objects.equals(webSocket, onClose.webSocket); |
|
341 } |
|
342 |
|
343 @Override |
|
344 public int hashCode() { |
|
345 return Objects.hash(statusCode, reason, webSocket); |
|
346 } |
|
347 } |
|
348 |
|
349 public static final class OnError extends ListenerInvocation { |
|
350 |
|
351 final Class<? extends Throwable> clazz; |
|
352 |
|
353 public OnError(WebSocket webSocket, Class<? extends Throwable> clazz) { |
|
354 super(webSocket); |
|
355 this.clazz = clazz; |
|
356 } |
|
357 |
|
358 @Override |
|
359 public boolean equals(Object o) { |
|
360 if (this == o) return true; |
|
361 if (o == null || getClass() != o.getClass()) return false; |
|
362 OnError onError = (OnError) o; |
|
363 return Objects.equals(clazz, onError.clazz) && |
|
364 Objects.equals(webSocket, onError.webSocket); |
|
365 } |
|
366 |
|
367 @Override |
|
368 public int hashCode() { |
|
369 return Objects.hash(clazz, webSocket); |
80 } |
370 } |
81 } |
371 } |
82 } |
372 } |