Adding new Arquillian tests for encoder/decoder

Author: arun-gupta

websocket/encoder/src/main/java/org/javaee7/websocket/encoder/MyEndpoint.java

@@ -49,7 +49,6 @@
         encoders = {MyMessageEncoder.class}, 
         decoders = {MyMessageDecoder.class})
 public class MyEndpoint {
-
     @OnMessage
     public MyMessage messageReceived(MyMessage message) {
         System.out.println("messageReceived: " + message);

websocket/encoder/src/main/java/org/javaee7/websocket/encoder/MyEndpointClientEmptyJSONArray.java

@@ -0,0 +1,34 @@
+package org.javaee7.websocket.encoder;
+
+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 MyEndpointClientEmptyJSONArray {
+    public static String JSON = "{}";
+    public static CountDownLatch latch= new CountDownLatch(3);
+
+    @OnOpen
+    public void onOpen(Session session) {
+        try {
+            session.getBasicRemote().sendText(JSON);
+        } catch (IOException ioe) {
+            ioe.printStackTrace();
+        }
+    }
+    
+    @OnMessage
+    public String processMessage(String message) {
+        latch.countDown();
+        return message;
+    }
+}

websocket/encoder/src/main/java/org/javaee7/websocket/encoder/MyEndpointClientJSONObject.java

@@ -0,0 +1,34 @@
+package org.javaee7.websocket.encoder;
+
+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 MyEndpointClientJSONObject {
+    public static CountDownLatch latch = new CountDownLatch(3);
+    public static String JSON = "{\"apple\" : \"red\", \"banana\": \"yellow\"}";
+
+    @OnOpen
+    public void onOpen(Session session) {
+        try {
+            session.getBasicRemote().sendText(JSON);
+        } catch (IOException ioe) {
+            ioe.printStackTrace();
+        }
+    }
+    
+    @OnMessage
+    public String processMessage(String message) {
+        latch.countDown();
+        return message;
+    }
+}

websocket/encoder/src/main/java/org/javaee7/websocket/encoder/MyMessageDecoder.java

@@ -52,10 +52,13 @@ public class MyMessageDecoder implements Decoder.Text<MyMessage> {
 
     @Override
     public MyMessage decode(String string) throws DecodeException {
-        System.out.println("decoding: " + string);
+        if (MyEndpointClientEmptyJSONArray.latch != null)
+            MyEndpointClientEmptyJSONArray.latch.countDown();
+        
+        if (MyEndpointClientJSONObject.latch != null)
+            MyEndpointClientJSONObject.latch.countDown();
+        
         MyMessage myMessage = new MyMessage(Json.createReader(new StringReader(string)).readObject());
-
-        System.out.println(myMessage.getJsonObject());
         return myMessage;
     }
 
@@ -65,12 +68,8 @@ public boolean willDecode(String string) {
     }
 
     @Override
-    public void init(EndpointConfig ec) {
-        System.out.println("init");
-    }
+    public void init(EndpointConfig ec) { }
 
     @Override
-    public void destroy() {
-        System.out.println("desroy");
-    }
+    public void destroy() { }
 }

websocket/encoder/src/main/java/org/javaee7/websocket/encoder/MyMessageEncoder.java

@@ -49,17 +49,18 @@
 public class MyMessageEncoder implements Encoder.Text<MyMessage> {
     @Override
     public String encode(MyMessage myMessage) throws EncodeException {
-//        return this.getClass().getName();
+        if (MyEndpointClientEmptyJSONArray.latch != null)
+            MyEndpointClientEmptyJSONArray.latch.countDown();
+        
+        if (MyEndpointClientJSONObject.latch != null)
+            MyEndpointClientJSONObject.latch.countDown();
+        
         return myMessage.getJsonObject().toString();
     }
 
     @Override
-    public void init(EndpointConfig ec) {
-        System.out.println("init");
-    }
+    public void init(EndpointConfig ec) { }
 
     @Override
-    public void destroy() {
-        System.out.println("destroy");
-    }
+    public void destroy() { }
 }

websocket/encoder/src/test/java/org/javaee7/websocket/encoder/EncoderEndpointTest.java

@@ -0,0 +1,100 @@
+package org.javaee7.websocket.encoder;
+
+import static org.junit.Assert.*;
+
+import java.io.IOException;
+import java.net.URI;
+import java.net.URISyntaxException;
+import java.util.concurrent.TimeUnit;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+import javax.websocket.ContainerProvider;
+import javax.websocket.DeploymentException;
+import javax.websocket.MessageHandler;
+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 org.junit.runner.RunWith;
+
+/**
+ * @author Nikos Ballas
+ * @author Arun Gupta
+ */
+@RunWith(Arquillian.class)
+public class EncoderEndpointTest {
+
+    @ArquillianResource
+    URI base;
+
+    /**
+     * Arquillian specific method for creating a file which can be deployed
+     * while executing the test.
+     */
+    @Deployment(testable = false)
+    public static WebArchive createDeployment() {
+        return ShrinkWrap.create(WebArchive.class)
+                .addClasses(MyEndpoint.class,
+                        MyMessage.class,
+                        MyMessageEncoder.class,
+                        MyMessageDecoder.class,
+                        MyEndpointClientEmptyJSONArray.class,
+                        MyEndpointClientJSONObject.class);
+    }
+
+    @Test
+    public void testEndpointEmptyJSONArray() throws URISyntaxException, DeploymentException, IOException, InterruptedException {
+        final Session session = connectToServer(MyEndpointClientEmptyJSONArray.class);
+        assertNotNull(session);
+        session.addMessageHandler(new MessageHandler.Whole<String>() {
+            @Override
+            public void onMessage(String text) {
+                assertEquals("{}", text);
+            }
+        });
+        assertTrue(MyEndpointClientEmptyJSONArray.latch.await(2, TimeUnit.SECONDS));
+    }
+
+    @Test
+    public void testEndpointEmptyJSONObject() throws URISyntaxException, DeploymentException, IOException, InterruptedException {
+        final String JSON = "{\"apple\" : \"red\", \"banana\": \"yellow\"}";
+        Session session = connectToServer(MyEndpointClientJSONObject.class);
+        assertNotNull(session);
+        session.addMessageHandler(new MessageHandler.Whole<String>() {
+            @Override
+            public void onMessage(String text) {
+                assertEquals(JSON, text);
+            }
+        });
+        assertTrue(MyEndpointClientJSONObject.latch.await(2, TimeUnit.SECONDS));
+    }
+
+    /**
+     * Method used to supply connection to the server by passing the naming of
+     * the websocket endpoint
+     *
+     * @param endpoint
+     * @return
+     * @throws DeploymentException
+     * @throws IOException
+     * @throws URISyntaxException
+     */
+    public Session connectToServer(Class endpoint) throws DeploymentException, IOException, URISyntaxException {
+        WebSocketContainer container = ContainerProvider.getWebSocketContainer();
+        URI uri = new URI("ws://"
+                + base.getHost()
+                + ":"
+                + base.getPort()
+                + "/"
+                + base.getPath()
+                + "/encoder");
+        return container.connectToServer(endpoint, uri);
+
+    }
+}