Arquillian tests for async websocket endpoint. Part of #86

airborn · airborn · commit ad3d9b15e3d0 · 2013-12-05T00:30:20.000+01:00
diff --git a/pom.xml b/pom.xml
@@ -113,6 +113,12 @@
             <version>20131018</version>
             <scope>test</scope>
         </dependency>
+        <dependency>
+            <groupId>com.jayway.awaitility</groupId>
+            <artifactId>awaitility</artifactId>
+            <version>1.3.5</version>
+            <scope>test</scope>
+        </dependency>
     </dependencies>
     <pluginRepositories>
         <pluginRepository>
diff --git a/websocket/endpoint-async/src/main/java/org/javaee7/websocket/endpoint/async/MyAsyncEndpointByteBuffer.java b/websocket/endpoint-async/src/main/java/org/javaee7/websocket/endpoint/async/MyAsyncEndpointByteBuffer.java
@@ -0,0 +1,18 @@
+package org.javaee7.websocket.endpoint.async;
+
+import java.nio.ByteBuffer;
+import javax.websocket.OnMessage;
+import javax.websocket.Session;
+import javax.websocket.server.ServerEndpoint;
+
+/**
+ * @author Jacek Jackowiak
+ */
+@ServerEndpoint("/bytebuffer")
+public class MyAsyncEndpointByteBuffer {
+
+    @OnMessage
+    public void echoByteBuffer(ByteBuffer data, Session session) {
+        session.getAsyncRemote().sendBinary(data);
+    }
+}
diff --git a/websocket/endpoint-async/src/main/java/org/javaee7/websocket/endpoint/async/MyAsyncEndpointByteBufferClient.java b/websocket/endpoint-async/src/main/java/org/javaee7/websocket/endpoint/async/MyAsyncEndpointByteBufferClient.java
@@ -0,0 +1,24 @@
+package org.javaee7.websocket.endpoint.async;
+
+import java.nio.ByteBuffer;
+import javax.websocket.ClientEndpoint;
+import javax.websocket.OnMessage;
+
+/**
+ *
+ * @author Jacek Jackowiak
+ */
+@ClientEndpoint
+public class MyAsyncEndpointByteBufferClient {
+
+    private ByteBuffer receivedMessage;
+
+    @OnMessage
+    public void onMessage(ByteBuffer msg) {
+        receivedMessage = msg;
+    }
+
+    public ByteBuffer getReceivedMessage() {
+        return receivedMessage;
+    }
+}
diff --git a/websocket/endpoint-async/src/main/java/org/javaee7/websocket/endpoint/async/MyAsyncEndpointText.java b/websocket/endpoint-async/src/main/java/org/javaee7/websocket/endpoint/async/MyAsyncEndpointText.java
@@ -46,11 +46,11 @@
 /**
  * @author Arun Gupta
  */
-@ServerEndpoint("/websocket")
-public class MyEndpoint {
-    
+@ServerEndpoint("/text")
+public class MyAsyncEndpointText {
+
     @OnMessage
-    public void echoText(String test, Session session) {
-        session.getAsyncRemote().sendText(test);
+    public void echoText(String text, Session session) {
+        session.getAsyncRemote().sendText(text);
     }
 }
diff --git a/websocket/endpoint-async/src/main/java/org/javaee7/websocket/endpoint/async/MyAsyncEndpointTextClient.java b/websocket/endpoint-async/src/main/java/org/javaee7/websocket/endpoint/async/MyAsyncEndpointTextClient.java
@@ -0,0 +1,22 @@
+package org.javaee7.websocket.endpoint.async;
+
+import javax.websocket.*;
+
+/**
+ * 
+ * @author Jacek Jackowiak
+ */
+@ClientEndpoint
+public class MyAsyncEndpointTextClient {
+
+    private String receivedMessage;
+
+    @OnMessage
+    public void onMessage(String msg) {
+        receivedMessage = msg;
+    }
+
+    public String getReceivedMessage() {
+        return receivedMessage;
+    }
+}
diff --git a/websocket/endpoint-async/src/test/java/org/javaee7/websocket/endpoint/async/MyAsyncEndpointTest.java b/websocket/endpoint-async/src/test/java/org/javaee7/websocket/endpoint/async/MyAsyncEndpointTest.java
@@ -0,0 +1,85 @@
+package org.javaee7.websocket.endpoint.async;
+
+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;
+
+import javax.websocket.*;
+import java.io.IOException;
+import java.io.UnsupportedEncodingException;
+import java.net.URI;
+import java.net.URISyntaxException;
+import java.net.URL;
+import java.nio.ByteBuffer;
+
+import static com.jayway.awaitility.Awaitility.*;
+import static org.hamcrest.Matchers.*;
+import static org.hamcrest.MatcherAssert.*;
+
+/**
+ *
+ * @author Jacek Jackowiak
+ */
+@RunWith(Arquillian.class)
+public class MyAsyncEndpointTest {
+
+    private static final String TEST_MESSAGE = "test";
+
+    @ArquillianResource
+    private URL base;
+
+    @Deployment(testable = false)
+    public static WebArchive deploy() throws URISyntaxException {
+        return ShrinkWrap.create(WebArchive.class)
+                .addClass(MyAsyncEndpointText.class)
+                .addClass(MyAsyncEndpointByteBuffer.class);
+    }
+
+    @Test
+    public void shouldReceiveAsyncTextMessage() throws URISyntaxException, IOException, DeploymentException {
+        MyAsyncEndpointTextClient endpoint = new MyAsyncEndpointTextClient();
+        Session session = connectToEndpoint(endpoint, "text");
+
+        session.getAsyncRemote().sendText(TEST_MESSAGE);
+
+        await().untilCall(to(endpoint).getReceivedMessage(), is(equalTo(TEST_MESSAGE)));
+    }
+
+    @Test
+    public void shouldReceiveAsyncByteBufferMessage() throws URISyntaxException, IOException, DeploymentException {
+        final ByteBuffer buffer = ByteBuffer.wrap(TEST_MESSAGE.getBytes());
+        MyAsyncEndpointByteBufferClient endpoint = new MyAsyncEndpointByteBufferClient();
+        Session session = connectToEndpoint(endpoint, "bytebuffer");
+
+        session.getAsyncRemote().sendBinary(buffer);
+
+        await().untilCall(to(endpoint).getReceivedMessage(), is(notNullValue()));
+        String receivedString = bufferToString(endpoint.getReceivedMessage());
+        assertThat(receivedString, is(equalTo(TEST_MESSAGE)));
+    }
+
+    private String bufferToString(ByteBuffer buffer) throws UnsupportedEncodingException {
+        byte[] bytes = new byte[buffer.remaining()];
+        buffer.duplicate().get(bytes);
+        return new String(bytes, "UTF-8");
+    }
+
+    private Session connectToEndpoint(Object endpoint, String uriPart) throws URISyntaxException, DeploymentException, IOException {
+        URI uri = getURI(uriPart);
+        WebSocketContainer container = ContainerProvider.getWebSocketContainer();
+        return container.connectToServer(endpoint, uri);
+    }
+
+    private URI getURI(String uriPart) throws URISyntaxException {
+        return new URI("ws://"
+                + base.getHost()
+                + ":"
+                + base.getPort()
+                + base.getPath()
+                + uriPart);
+    }
+}
