MyEndpointTest.java

package org.javaee7.websocket.binary.test;

import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.assertNotNull;

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.javaee7.websocket.binary.MyEndpointByteArray;
import org.javaee7.websocket.binary.MyEndpointByteBuffer;
import org.javaee7.websocket.binary.MyEndpointClient;
import org.javaee7.websocket.binary.MyEndpointInputStream;
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 MyEndpointTest {

    private static final String RESPONSE = "Hello World!";

    @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(MyEndpointByteBuffer.class,
                        MyEndpointByteArray.class,
                        MyEndpointInputStream.class,
                        MyEndpointClient.class);
    }

    /**
     * The basic test method for the class {@link MyEndpointByteBuffer}
     *
     * @throws URISyntaxException
     * @throws DeploymentException
     * @throws IOException
     */
    @Test
    public void testEndpointByteBuffer() throws URISyntaxException, DeploymentException, IOException, InterruptedException {
        MyEndpointClient.latch = new CountDownLatch(1);
        Session session = connectToServer("bytebuffer");
        assertNotNull(session);
        assertTrue(MyEndpointClient.latch.await(2, TimeUnit.SECONDS));
        assertNotNull(MyEndpointClient.response);
        assertArrayEquals(RESPONSE.getBytes(), MyEndpointClient.response);
    }

    /**
     * The basic test method for the class {
     *
     * @MyEndpointByteArray }
     *
     * @throws DeploymentException
     * @throws IOException
     * @throws URISyntaxException
     */
    @Test
    public void testEndpointByteArray() throws DeploymentException, IOException, URISyntaxException, InterruptedException {
        MyEndpointClient.latch = new CountDownLatch(1);
        Session session = connectToServer("bytearray");
        assertNotNull(session);
        assertTrue(MyEndpointClient.latch.await(2, TimeUnit.SECONDS));
        assertNotNull(MyEndpointClient.response);
        assertArrayEquals(RESPONSE.getBytes(), MyEndpointClient.response);
    }

    /**
     * The basic test method for the class {
     *
     * @MyEndpointInputStream }
     *
     * @throws DeploymentException
     * @throws IOException
     * @throws URISyntaxException
     */
    @Test
    public void testEndpointInputStream() throws DeploymentException, IOException, URISyntaxException, InterruptedException {
        MyEndpointClient.latch = new CountDownLatch(1);
        Session session = connectToServer("inputstream");
        assertNotNull(session);
        assertTrue(MyEndpointClient.latch.await(2, TimeUnit.SECONDS));
        assertNotNull(MyEndpointClient.response);
        assertArrayEquals(RESPONSE.getBytes(), MyEndpointClient.response);
    }

    /**
     * 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(String endpoint) throws DeploymentException, IOException, URISyntaxException {
        WebSocketContainer container = ContainerProvider.getWebSocketContainer();
        URI uri = new URI("ws://"
                        + base.getHost()
                        + ":"
                        + base.getPort()
                        + "/"
                        + base.getPath()
                        + "/"
                        + endpoint);
        System.out.println("Connecting to: " + uri);
        return container.connectToServer(MyEndpointClient.class, uri);
    }
}