Play Framework/Scala example source code file (WebSocketTest.java)

This example Play Framework source code file (WebSocketTest.java) is included in my "Source Code Warehouse" project. The intent of this project is to help you more easily find Play Framework (and Scala) source code examples by using tags.

All credit for the original source code belongs to Play Framework; I'm just trying to make examples easier to find. (For my Scala work, see my Scala examples and tutorials.)

Play Framework tags/keywords

api, countdownlatch, exception, iterate, lib, library, override, play, play framework, seconds, test, testchannel, throwable, websocket, websockettest

The WebSocketTest.java Play Framework example source code

/*
 * Copyright (C) 2009-2013 Typesafe Inc. <http://www.typesafe.com>
 */
package play.mvc;

import java.util.concurrent.CountDownLatch;
import org.junit.Test;
import play.api.libs.iteratee.TestChannel;
import play.libs.F;

import static java.util.concurrent.TimeUnit.SECONDS;
import static org.junit.Assert.assertTrue;

public class WebSocketTest {

    @Test
    public void testWebSocket() throws Throwable {
        final CountDownLatch ready = new CountDownLatch(1);
        final CountDownLatch message = new CountDownLatch(1);
        final CountDownLatch close = new CountDownLatch(1);

        final TestChannel<String> testChannel = new TestChannel<String>();
        final WebSocket.Out<String> wsOut = new WebSocket.Out<String>() {
            @Override
            public void write(String frame) {
                testChannel.push(frame);
            }
            @Override
            public void close() {
                testChannel.eofAndEnd();
            }
        };

        final WebSocket.In<String> wsIn = new WebSocket.In<String>();

        WebSocket<String> webSocket = new WebSocket<String>() {
            @Override
            public void onReady(WebSocket.In<String> in, WebSocket.Out<String> out) {
                ready.countDown();
                in.onMessage(new F.Callback<String>() {
                    @Override
                    public void invoke(String m) {
                        message.countDown();
                    }
                });
                in.onClose(new F.Callback0() {
                    @Override
                    public void invoke() {
                        close.countDown();
                    }
                });
                out.write("message");
                out.close();
            }
        };

        webSocket.onReady(wsIn, wsOut);

        assertTrue("WebSocket.onReady callback was not invoked", ready.await(1, SECONDS));

        for (F.Callback<String> callback : wsIn.callbacks) {
            callback.invoke("message");
        }

        assertTrue("WebSocket.In.onMessage callback was not invoked", message.await(1, SECONDS));

        for (F.Callback0 callback : wsIn.closeCallbacks) {
            callback.invoke();
        }

        assertTrue("WebSocket.In.onClose callback was not invoked", close.await(1, SECONDS));

        testChannel.expect("message");
        testChannel.expectEOF();
        testChannel.expectEnd();
        testChannel.expectEmpty();
    }

    @Test
    public void testWhenReadyFactory() throws Exception {
        final CountDownLatch ready = new CountDownLatch(1);

        WebSocket<String> webSocket = WebSocket.whenReady(new F.Callback2<WebSocket.In<String>, WebSocket.Out<String>>() {
            @Override
            public void invoke(WebSocket.In<String> in, WebSocket.Out<String> out) {
                ready.countDown();
            }
        });

        webSocket.onReady(null, null);

        assertTrue("WebSocket.onReady callback was not invoked", ready.await(1, SECONDS));
    }
}