Adding a new test for WebSocket Chat

Author: arun-gupta

websocket/chat/src/main/java/org/javaee7/websocket/chat/ChatClientEndpoint1.java

@@ -0,0 +1,34 @@
+package org.javaee7.websocket.chat;
+
+import java.io.IOException;
+import java.util.concurrent.CountDownLatch;
+
+import javax.websocket.ClientEndpoint;
+import javax.websocket.OnMessage;
+import javax.websocket.OnOpen;
+import javax.websocket.Session;
+
+/**
+ * @author Arun Gupta
+ */
+@ClientEndpoint
+public class ChatClientEndpoint1 {
+    public static String TEXT = "Client1 joins";
+    public static CountDownLatch latch;
+    public static String response;
+
+    @OnOpen
+    public void onOpen(Session session) {
+        try {
+            session.getBasicRemote().sendText(TEXT);
+        } catch (IOException ioe) {
+            ioe.printStackTrace();
+        }
+    }
+    
+    @OnMessage
+    public void processMessage(String message) {
+        response = message;
+        latch.countDown();
+    }
+}

websocket/chat/src/main/java/org/javaee7/websocket/chat/ChatClientEndpoint2.java

@@ -0,0 +1,34 @@
+package org.javaee7.websocket.chat;
+
+import java.io.IOException;
+import java.util.concurrent.CountDownLatch;
+
+import javax.websocket.ClientEndpoint;
+import javax.websocket.OnMessage;
+import javax.websocket.OnOpen;
+import javax.websocket.Session;
+
+/**
+ * @author Arun Gupta
+ */
+@ClientEndpoint
+public class ChatClientEndpoint2 {
+    public static String TEXT = "Client2 joins";
+    public static CountDownLatch latch;
+    public static String response;
+
+    @OnOpen
+    public void onOpen(Session session) {
+        try {
+            session.getBasicRemote().sendText(TEXT);
+        } catch (IOException ioe) {
+            ioe.printStackTrace();
+        }
+    }
+    
+    @OnMessage
+    public void processMessage(String message) {
+        response = message;
+        latch.countDown();
+    }
+}

websocket/chat/src/main/java/org/javaee7/websocket/chat/ChatEndpoint.java

@@ -52,6 +52,7 @@
 public class ChatEndpoint {
     @OnMessage
     public void message(String message, Session client) throws IOException, EncodeException {
+        System.out.println("message: " + message);
         for (Session peer : client.getOpenSessions()) {
             peer.getBasicRemote().sendText(message);
         }

websocket/chat/src/test/java/org/javaee7/websocket/chat/ChatTest.java

@@ -0,0 +1,68 @@
+package org.javaee7.websocket.chat;
+
+import java.io.IOException;
+import java.net.URI;
+import java.net.URISyntaxException;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.TimeUnit;
+import javax.websocket.ContainerProvider;
+import javax.websocket.DeploymentException;
+import javax.websocket.Session;
+import javax.websocket.WebSocketContainer;
+import org.jboss.arquillian.container.test.api.Deployment;
+import org.jboss.arquillian.junit.Arquillian;
+import org.jboss.arquillian.test.api.ArquillianResource;
+import org.jboss.shrinkwrap.api.ShrinkWrap;
+import org.jboss.shrinkwrap.api.spec.WebArchive;
+import org.junit.Test;
+import static org.junit.Assert.*;
+import org.junit.runner.RunWith;
+
+/**
+ * @author Arun Gupta
+ */
+@RunWith(Arquillian.class)
+public class ChatTest {
+    
+    @ArquillianResource
+    URI base;
+
+    @Deployment(testable = false)
+    public static WebArchive createDeployment() {
+        return ShrinkWrap.create(WebArchive.class)
+                .addClasses(ChatEndpoint.class,
+                        ChatClientEndpoint1.class,
+                        ChatClientEndpoint2.class);
+    }
+    
+    @Test
+    public void testConnect() throws URISyntaxException, DeploymentException, IOException, InterruptedException {
+        ChatClientEndpoint1.latch = new CountDownLatch(1);
+        final Session session1 = connectToServer(ChatClientEndpoint1.class);
+        assertNotNull(session1);
+        assertTrue(ChatClientEndpoint1.latch.await(2, TimeUnit.SECONDS));
+        
+        assertEquals(ChatClientEndpoint1.TEXT, ChatClientEndpoint1.response);
+
+        ChatClientEndpoint1.latch = new CountDownLatch(1);
+        ChatClientEndpoint2.latch = new CountDownLatch(1);
+        final Session session2 = connectToServer(ChatClientEndpoint2.class);
+        assertNotNull(session2);
+        assertTrue(ChatClientEndpoint1.latch.await(2, TimeUnit.SECONDS));
+        assertTrue(ChatClientEndpoint2.latch.await(2, TimeUnit.SECONDS));
+        assertEquals(ChatClientEndpoint2.TEXT, ChatClientEndpoint1.response);
+        assertEquals(ChatClientEndpoint2.TEXT, ChatClientEndpoint2.response);
+    }
+    
+
+    public Session connectToServer(Class<?> endpoint) throws DeploymentException, IOException, URISyntaxException {
+        WebSocketContainer container = ContainerProvider.getWebSocketContainer();
+        URI uri = new URI("ws://"
+                + base.getHost()
+                + ":"
+                + base.getPort()
+                + base.getPath()
+                + "chat");
+        return container.connectToServer(endpoint, uri);
+    }
+}