Merge commit 'e992c260c059c53c4aabc980db85efd58f190205'
[nikiroo-utils.git] / test_code / SerialServerTest.java
1 package be.nikiroo.utils.test_code;
2
3 import java.net.URL;
4
5 import be.nikiroo.utils.Version;
6 import be.nikiroo.utils.serial.server.ConnectActionClientObject;
7 import be.nikiroo.utils.serial.server.ConnectActionClientString;
8 import be.nikiroo.utils.serial.server.ConnectActionServerObject;
9 import be.nikiroo.utils.serial.server.ConnectActionServerString;
10 import be.nikiroo.utils.serial.server.ServerBridge;
11 import be.nikiroo.utils.serial.server.ServerObject;
12 import be.nikiroo.utils.serial.server.ServerString;
13 import be.nikiroo.utils.test.TestCase;
14 import be.nikiroo.utils.test.TestLauncher;
15
16 class SerialServerTest extends TestLauncher {
17 public SerialServerTest(String[] args) {
18 super("SerialServer test", args);
19
20 for (String key : new String[] { null,
21 "some super secret encryption key" }) {
22 for (boolean bridge : new Boolean[] { false, true }) {
23 final String skey = (key != null ? "(encrypted)"
24 : "(plain text)");
25 final String sbridge = (bridge ? " with bridge" : "");
26
27 addSeries(new SerialServerTest(args, key, skey, bridge,
28 sbridge, "ServerString"));
29
30 addSeries(new SerialServerTest(args, key, skey, bridge,
31 sbridge, new Object() {
32 @Override
33 public String toString() {
34 return "ServerObject";
35 }
36 }));
37 }
38 }
39 }
40
41 private SerialServerTest(final String[] args, final String key,
42 final String skey, final boolean bridge, final String sbridge,
43 final String title) {
44
45 super(title + " " + skey + sbridge, args);
46
47 addTest(new TestCase("Simple connection " + skey) {
48 @Override
49 public void test() throws Exception {
50 final String[] rec = new String[1];
51
52 ServerString server = new ServerString(this.getName(), 0, key) {
53 @Override
54 protected String onRequest(
55 ConnectActionServerString action, Version version,
56 String data, long id) throws Exception {
57 return null;
58 }
59
60 @Override
61 protected void onError(Exception e) {
62 }
63 };
64
65 int port = server.getPort();
66 assertEquals("A port should have been assigned", true, port > 0);
67
68 server.start();
69
70 ServerBridge br = null;
71 if (bridge) {
72 br = new ServerBridge(0, key, "", port, key);
73 br.setTraceHandler(null);
74
75 port = br.getPort();
76 assertEquals(
77 "A port should have been assigned to the bridge",
78 true, port > 0);
79
80 br.start();
81 }
82
83 try {
84 try {
85 new ConnectActionClientString(null, port, key) {
86 @Override
87 public void action(Version version)
88 throws Exception {
89 rec[0] = "ok";
90 }
91 }.connect();
92 } finally {
93 server.stop();
94 }
95 } finally {
96 if (br != null) {
97 br.stop();
98 }
99 }
100
101 assertNotNull("The client action was not run", rec[0]);
102 assertEquals("ok", rec[0]);
103 }
104 });
105
106 addTest(new TestCase("Simple exchange " + skey) {
107 final String[] sent = new String[1];
108 final String[] recd = new String[1];
109 final Exception[] err = new Exception[1];
110
111 @Override
112 public void test() throws Exception {
113 ServerString server = new ServerString(this.getName(), 0, key) {
114 @Override
115 protected String onRequest(
116 ConnectActionServerString action, Version version,
117 String data, long id) throws Exception {
118 sent[0] = data;
119 return "pong";
120 }
121
122 @Override
123 protected void onError(Exception e) {
124 err[0] = e;
125 }
126 };
127
128 int port = server.getPort();
129
130 server.start();
131
132 ServerBridge br = null;
133 if (bridge) {
134 br = new ServerBridge(0, key, "", port, key);
135 br.setTraceHandler(null);
136 port = br.getPort();
137 br.start();
138 }
139
140 try {
141 try {
142 new ConnectActionClientString(null, port, key) {
143 @Override
144 public void action(Version version)
145 throws Exception {
146 recd[0] = send("ping");
147 }
148 }.connect();
149 } finally {
150 server.stop();
151 }
152 } finally {
153 if (br != null) {
154 br.stop();
155 }
156 }
157
158 if (err[0] != null) {
159 fail("An exception was thrown: " + err[0].getMessage(),
160 err[0]);
161 }
162
163 assertEquals("ping", sent[0]);
164 assertEquals("pong", recd[0]);
165 }
166 });
167
168 addTest(new TestCase("Multiple exchanges " + skey) {
169 final String[] sent = new String[3];
170 final String[] recd = new String[3];
171 final Exception[] err = new Exception[1];
172
173 @Override
174 public void test() throws Exception {
175 ServerString server = new ServerString(this.getName(), 0, key) {
176 @Override
177 protected String onRequest(
178 ConnectActionServerString action, Version version,
179 String data, long id) throws Exception {
180 sent[0] = data;
181 action.send("pong");
182 sent[1] = action.rec();
183 return "pong2";
184 }
185
186 @Override
187 protected void onError(Exception e) {
188 err[0] = e;
189 }
190 };
191
192 int port = server.getPort();
193
194 server.start();
195
196 ServerBridge br = null;
197 if (bridge) {
198 br = new ServerBridge(0, key, "", port, key);
199 br.setTraceHandler(null);
200 port = br.getPort();
201 br.start();
202 }
203
204 try {
205 try {
206 new ConnectActionClientString(null, port, key) {
207 @Override
208 public void action(Version version)
209 throws Exception {
210 recd[0] = send("ping");
211 recd[1] = send("ping2");
212 }
213 }.connect();
214 } finally {
215 server.stop();
216 }
217 } finally {
218 if (br != null) {
219 br.stop();
220 }
221 }
222
223 if (err[0] != null) {
224 fail("An exception was thrown: " + err[0].getMessage(),
225 err[0]);
226 }
227
228 assertEquals("ping", sent[0]);
229 assertEquals("pong", recd[0]);
230 assertEquals("ping2", sent[1]);
231 assertEquals("pong2", recd[1]);
232 }
233 });
234
235 addTest(new TestCase("Multiple call from client " + skey) {
236 final String[] sent = new String[3];
237 final String[] recd = new String[3];
238 final Exception[] err = new Exception[1];
239
240 @Override
241 public void test() throws Exception {
242 ServerString server = new ServerString(this.getName(), 0, key) {
243 @Override
244 protected String onRequest(
245 ConnectActionServerString action, Version version,
246 String data, long id) throws Exception {
247 sent[Integer.parseInt(data)] = data;
248 return "" + (Integer.parseInt(data) * 2);
249 }
250
251 @Override
252 protected void onError(Exception e) {
253 err[0] = e;
254 }
255 };
256
257 int port = server.getPort();
258
259 server.start();
260
261 ServerBridge br = null;
262 if (bridge) {
263 br = new ServerBridge(0, key, "", port, key);
264 br.setTraceHandler(null);
265 port = br.getPort();
266 br.start();
267 }
268
269 try {
270 try {
271 new ConnectActionClientString(null, port, key) {
272 @Override
273 public void action(Version version)
274 throws Exception {
275 for (int i = 0; i < 3; i++) {
276 recd[i] = send("" + i);
277 }
278 }
279 }.connect();
280 } finally {
281 server.stop();
282 }
283 } finally {
284 if (br != null) {
285 br.stop();
286 }
287 }
288
289 if (err[0] != null) {
290 fail("An exception was thrown: " + err[0].getMessage(),
291 err[0]);
292 }
293
294 assertEquals("0", sent[0]);
295 assertEquals("0", recd[0]);
296 assertEquals("1", sent[1]);
297 assertEquals("2", recd[1]);
298 assertEquals("2", sent[2]);
299 assertEquals("4", recd[2]);
300 }
301 });
302 }
303
304 private SerialServerTest(final String[] args, final String key,
305 final String skey, final boolean bridge, final String sbridge,
306 final Object title) {
307
308 super(title + " " + skey + sbridge, args);
309
310 addTest(new TestCase("Simple connection " + skey) {
311 @Override
312 public void test() throws Exception {
313 final Object[] rec = new Object[1];
314
315 ServerObject server = new ServerObject(this.getName(), 0, key) {
316 @Override
317 protected Object onRequest(
318 ConnectActionServerObject action, Version version,
319 Object data, long id) throws Exception {
320 return null;
321 }
322
323 @Override
324 protected void onError(Exception e) {
325 }
326 };
327
328 int port = server.getPort();
329 assertEquals("A port should have been assigned", true, port > 0);
330
331 server.start();
332
333 ServerBridge br = null;
334 if (bridge) {
335 br = new ServerBridge(0, key, "", port, key);
336 br.setTraceHandler(null);
337 port = br.getPort();
338 br.start();
339 }
340
341 try {
342 try {
343 new ConnectActionClientObject(null, port, key) {
344 @Override
345 public void action(Version version)
346 throws Exception {
347 rec[0] = true;
348 }
349
350 @Override
351 protected void onError(Exception e) {
352 }
353 }.connect();
354 } finally {
355 server.stop();
356 }
357 } finally {
358 if (br != null) {
359 br.stop();
360 }
361 }
362
363 assertNotNull("The client action was not run", rec[0]);
364 assertEquals(true, (boolean) ((Boolean) rec[0]));
365 }
366 });
367
368 addTest(new TestCase("Simple exchange " + skey) {
369 final Object[] sent = new Object[1];
370 final Object[] recd = new Object[1];
371 final Exception[] err = new Exception[1];
372
373 @Override
374 public void test() throws Exception {
375 ServerObject server = new ServerObject(this.getName(), 0, key) {
376 @Override
377 protected Object onRequest(
378 ConnectActionServerObject action, Version version,
379 Object data, long id) throws Exception {
380 sent[0] = data;
381 return "pong";
382 }
383
384 @Override
385 protected void onError(Exception e) {
386 err[0] = e;
387 }
388 };
389
390 int port = server.getPort();
391
392 server.start();
393
394 ServerBridge br = null;
395 if (bridge) {
396 br = new ServerBridge(0, key, "", port, key);
397 br.setTraceHandler(null);
398 port = br.getPort();
399 br.start();
400 }
401
402 try {
403 try {
404 new ConnectActionClientObject(null, port, key) {
405 @Override
406 public void action(Version version)
407 throws Exception {
408 recd[0] = send("ping");
409 }
410 }.connect();
411 } finally {
412 server.stop();
413 }
414 } finally {
415 if (br != null) {
416 br.stop();
417 }
418 }
419
420 if (err[0] != null) {
421 fail("An exception was thrown: " + err[0].getMessage(),
422 err[0]);
423 }
424
425 assertEquals("ping", sent[0]);
426 assertEquals("pong", recd[0]);
427 }
428 });
429
430 addTest(new TestCase("Multiple exchanges " + skey) {
431 final Object[] sent = new Object[3];
432 final Object[] recd = new Object[3];
433 final Exception[] err = new Exception[1];
434
435 @Override
436 public void test() throws Exception {
437 ServerObject server = new ServerObject(this.getName(), 0, key) {
438 @Override
439 protected Object onRequest(
440 ConnectActionServerObject action, Version version,
441 Object data, long id) throws Exception {
442 sent[0] = data;
443 action.send("pong");
444 sent[1] = action.rec();
445 return "pong2";
446 }
447
448 @Override
449 protected void onError(Exception e) {
450 err[0] = e;
451 }
452 };
453
454 int port = server.getPort();
455
456 server.start();
457
458 ServerBridge br = null;
459 if (bridge) {
460 br = new ServerBridge(0, key, "", port, key);
461 br.setTraceHandler(null);
462 port = br.getPort();
463 br.start();
464 }
465
466 try {
467 try {
468 new ConnectActionClientObject(null, port, key) {
469 @Override
470 public void action(Version version)
471 throws Exception {
472 recd[0] = send("ping");
473 recd[1] = send("ping2");
474 }
475 }.connect();
476 } finally {
477 server.stop();
478 }
479 } finally {
480 if (br != null) {
481 br.stop();
482 }
483 }
484
485 if (err[0] != null) {
486 fail("An exception was thrown: " + err[0].getMessage(),
487 err[0]);
488 }
489
490 assertEquals("ping", sent[0]);
491 assertEquals("pong", recd[0]);
492 assertEquals("ping2", sent[1]);
493 assertEquals("pong2", recd[1]);
494 }
495 });
496
497 addTest(new TestCase("Object array of URLs " + skey) {
498 final Object[] sent = new Object[1];
499 final Object[] recd = new Object[1];
500 final Exception[] err = new Exception[1];
501
502 @Override
503 public void test() throws Exception {
504 ServerObject server = new ServerObject(this.getName(), 0, key) {
505 @Override
506 protected Object onRequest(
507 ConnectActionServerObject action, Version version,
508 Object data, long id) throws Exception {
509 sent[0] = data;
510 return new Object[] { "ACK" };
511 }
512
513 @Override
514 protected void onError(Exception e) {
515 err[0] = e;
516 }
517 };
518
519 int port = server.getPort();
520
521 server.start();
522
523 ServerBridge br = null;
524 if (bridge) {
525 br = new ServerBridge(0, key, "", port, key);
526 br.setTraceHandler(null);
527 port = br.getPort();
528 br.start();
529 }
530
531 try {
532 try {
533 new ConnectActionClientObject(null, port, key) {
534 @Override
535 public void action(Version version)
536 throws Exception {
537 recd[0] = send(new Object[] {
538 "key",
539 new URL(
540 "https://example.com/from_client"),
541 "https://example.com/from_client" });
542 }
543 }.connect();
544 } finally {
545 server.stop();
546 }
547 } finally {
548 if (br != null) {
549 br.stop();
550 }
551 }
552
553 if (err[0] != null) {
554 fail("An exception was thrown: " + err[0].getMessage(),
555 err[0]);
556 }
557
558 Object[] sento = (Object[]) (sent[0]);
559 Object[] recdo = (Object[]) (recd[0]);
560
561 assertEquals("key", sento[0]);
562 assertEquals("https://example.com/from_client",
563 ((URL) sento[1]).toString());
564 assertEquals("https://example.com/from_client", sento[2]);
565 assertEquals("ACK", recdo[0]);
566 }
567 });
568
569 addTest(new TestCase("Multiple call from client " + skey) {
570 final Object[] sent = new Object[3];
571 final Object[] recd = new Object[3];
572 final Exception[] err = new Exception[1];
573
574 @Override
575 public void test() throws Exception {
576 ServerObject server = new ServerObject(this.getName(), 0, key) {
577 @Override
578 protected Object onRequest(
579 ConnectActionServerObject action, Version version,
580 Object data, long id) throws Exception {
581 sent[(Integer) data] = data;
582 return ((Integer) data) * 2;
583 }
584
585 @Override
586 protected void onError(Exception e) {
587 err[0] = e;
588 }
589 };
590
591 int port = server.getPort();
592
593 server.start();
594
595 ServerBridge br = null;
596 if (bridge) {
597 br = new ServerBridge(0, key, "", port, key);
598 br.setTraceHandler(null);
599 port = br.getPort();
600 br.start();
601 }
602
603 try {
604 try {
605 new ConnectActionClientObject(null, port, key) {
606 @Override
607 public void action(Version version)
608 throws Exception {
609 for (int i = 0; i < 3; i++) {
610 recd[i] = send(i);
611 }
612 }
613 }.connect();
614 } finally {
615 server.stop();
616 }
617 } finally {
618 if (br != null) {
619 br.stop();
620 }
621 }
622
623 if (err[0] != null) {
624 fail("An exception was thrown: " + err[0].getMessage(),
625 err[0]);
626 }
627
628 assertEquals(0, sent[0]);
629 assertEquals(0, recd[0]);
630 assertEquals(1, sent[1]);
631 assertEquals(2, recd[1]);
632 assertEquals(2, sent[2]);
633 assertEquals(4, recd[2]);
634 }
635 });
636 }
637 }