|
4 | 4 | import java.net.*; |
5 | 5 | import java.util.Set; |
6 | 6 | import java.util.concurrent.*; |
| 7 | + |
7 | 8 | import org.java_websocket.WebSocket; |
8 | 9 | import org.java_websocket.handshake.*; |
9 | 10 | import org.java_websocket.client.*; |
10 | 11 | import org.java_websocket.server.WebSocketServer; |
11 | 12 | import org.java_websocket.util.SocketUtil; |
12 | 13 | import org.java_websocket.enums.ReadyState; |
13 | | -import org.junit.Test; |
14 | | -import static org.junit.Assert.*; |
| 14 | +import org.junit.jupiter.api.Test; |
| 15 | +import org.junit.jupiter.api.Timeout; |
| 16 | + |
| 17 | +import static org.junit.jupiter.api.Assertions.*; |
15 | 18 |
|
16 | 19 | public class ConnectBlockingTest { |
17 | 20 |
|
18 | | - @Test(timeout = 1000) |
19 | | - public void test_ConnectBlockingCleanup() throws Throwable { |
| 21 | + @Test |
| 22 | + @Timeout(1000) |
| 23 | + public void test_ConnectBlockingCleanup() throws Throwable { |
| 24 | + |
| 25 | + Set<Thread> threadSet1 = Thread.getAllStackTraces().keySet(); |
| 26 | + final CountDownLatch ready = new CountDownLatch(1); |
| 27 | + final CountDownLatch accepted = new CountDownLatch(1); |
| 28 | + |
| 29 | + final int port = SocketUtil.getAvailablePort(); |
| 30 | + |
| 31 | + /* TCP server which listens to a port, but does not answer handshake */ |
| 32 | + Thread server = new Thread(new Runnable() { |
| 33 | + @Override |
| 34 | + public void run() { |
| 35 | + try { |
| 36 | + ServerSocket serverSocket = new ServerSocket(port); |
| 37 | + serverSocket.setReuseAddress(true); |
| 38 | + ready.countDown(); |
| 39 | + Socket clientSocket = serverSocket.accept(); |
| 40 | + accepted.countDown(); |
| 41 | + } catch (Throwable t) { |
| 42 | + assertInstanceOf(InterruptedException.class, t); |
| 43 | + } |
| 44 | + } |
| 45 | + }); |
| 46 | + server.start(); |
| 47 | + ready.await(); |
20 | 48 |
|
21 | | - Set<Thread> threadSet1 = Thread.getAllStackTraces().keySet(); |
22 | | - final CountDownLatch ready = new CountDownLatch(1); |
23 | | - final CountDownLatch accepted = new CountDownLatch(1); |
| 49 | + WebSocketClient client = new WebSocketClient(URI.create("ws://localhost:" + port)) { |
| 50 | + @Override |
| 51 | + public void onOpen(ServerHandshake handshake) { |
| 52 | + } |
24 | 53 |
|
25 | | - final int port = SocketUtil.getAvailablePort(); |
| 54 | + @Override |
| 55 | + public void onClose(int code, String reason, boolean remote) { |
| 56 | + } |
26 | 57 |
|
27 | | - /* TCP server which listens to a port, but does not answer handshake */ |
28 | | - Thread server = new Thread(new Runnable() { |
29 | | - @Override |
30 | | - public void run() { |
31 | | - try { |
32 | | - ServerSocket serverSocket = new ServerSocket(port); |
33 | | - ready.countDown(); |
34 | | - Socket clientSocket = serverSocket.accept(); |
35 | | - accepted.countDown(); |
36 | | - } catch (Throwable t) { |
37 | | - assertTrue(t instanceof InterruptedException); |
38 | | - } |
39 | | - } |
40 | | - }); |
41 | | - server.start(); |
42 | | - ready.await(); |
| 58 | + @Override |
| 59 | + public void onMessage(String message) { |
| 60 | + } |
43 | 61 |
|
44 | | - WebSocketClient client = new WebSocketClient(URI.create("ws://localhost:" + port)) { |
45 | | - @Override |
46 | | - public void onOpen(ServerHandshake handshake) { |
47 | | - } |
48 | | - @Override |
49 | | - public void onClose(int code, String reason, boolean remote) {} |
50 | | - @Override |
51 | | - public void onMessage(String message) {} |
52 | | - @Override |
53 | | - public void onError(Exception ex) { |
54 | | - ex.printStackTrace(); |
55 | | - } |
56 | | - }; |
57 | | - boolean connected = client.connectBlocking(100, TimeUnit.MILLISECONDS); |
58 | | - assertEquals("TCP socket should have been accepted", 0, accepted.getCount()); |
59 | | - assertFalse("WebSocket should not be connected (as server didn't send handshake)", connected); |
| 62 | + @Override |
| 63 | + public void onError(Exception ex) { |
| 64 | + ex.printStackTrace(); |
| 65 | + } |
| 66 | + }; |
| 67 | + boolean connected = client.connectBlocking(100, TimeUnit.MILLISECONDS); |
| 68 | + assertEquals( 0, accepted.getCount(), "TCP socket should have been accepted"); |
| 69 | + assertFalse(connected, "WebSocket should not be connected (as server didn't send handshake)"); |
60 | 70 |
|
61 | | - server.interrupt(); |
62 | | - server.join(); |
| 71 | + server.interrupt(); |
| 72 | + server.join(); |
63 | 73 |
|
64 | | - Set<Thread> threadSet2 = Thread.getAllStackTraces().keySet(); |
65 | | - assertEquals("no threads left over", threadSet1, threadSet2); |
66 | | - assertTrue("WebSocket is in closed state", client.getReadyState() == ReadyState.CLOSED || client.getReadyState() == ReadyState.NOT_YET_CONNECTED); |
67 | | - } |
| 74 | + Set<Thread> threadSet2 = Thread.getAllStackTraces().keySet(); |
| 75 | + assertEquals(threadSet1, threadSet2, "no threads left over"); |
| 76 | + assertTrue(client.getReadyState() == ReadyState.CLOSED || client.getReadyState() == ReadyState.NOT_YET_CONNECTED, "WebSocket is in closed state"); |
| 77 | + } |
68 | 78 | } |
0 commit comments