| package fi.iki.elonen.samples.echo; |
| |
| /* |
| * #%L |
| * NanoHttpd-Websocket |
| * %% |
| * Copyright (C) 2012 - 2015 nanohttpd |
| * %% |
| * Redistribution and use in source and binary forms, with or without modification, |
| * are permitted provided that the following conditions are met: |
| * |
| * 1. Redistributions of source code must retain the above copyright notice, this |
| * list of conditions and the following disclaimer. |
| * |
| * 2. Redistributions in binary form must reproduce the above copyright notice, |
| * this list of conditions and the following disclaimer in the documentation |
| * and/or other materials provided with the distribution. |
| * |
| * 3. Neither the name of the nanohttpd nor the names of its contributors |
| * may be used to endorse or promote products derived from this software without |
| * specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND |
| * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
| * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
| * IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, |
| * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, |
| * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF |
| * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE |
| * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED |
| * OF THE POSSIBILITY OF SUCH DAMAGE. |
| * #L% |
| */ |
| |
| import java.net.URI; |
| import java.util.concurrent.TimeUnit; |
| |
| import org.eclipse.jetty.websocket.client.ClientUpgradeRequest; |
| import org.eclipse.jetty.websocket.client.WebSocketClient; |
| import org.junit.AfterClass; |
| import org.junit.Assert; |
| import org.junit.BeforeClass; |
| import org.junit.Test; |
| |
| import fi.iki.elonen.NanoWSD; |
| |
| public class EchoWebSocketsTest { |
| |
| private static NanoWSD server; |
| |
| @BeforeClass |
| public static void setUp() throws Exception { |
| EchoWebSocketsTest.server = new DebugWebSocketServer(9191, true); |
| EchoWebSocketsTest.server.start(); |
| } |
| |
| @AfterClass |
| public static void tearDown() throws Exception { |
| EchoWebSocketsTest.server.stop(); |
| } |
| |
| @Test |
| public void testWebsocketClient() throws Exception { |
| String destUri = "ws://localhost:9191"; |
| |
| WebSocketClient client = new WebSocketClient(); |
| SimpleEchoSocket socket = new SimpleEchoSocket(); |
| socket.getToSendMessages().add("Hello"); |
| socket.getToSendMessages().add("Thanks for the conversation."); |
| socket.getToSendMessages().add(createString(31000)); |
| socket.getToSendMessages().add(createString(65400)); |
| try { |
| client.start(); |
| URI echoUri = new URI(destUri); |
| ClientUpgradeRequest request = new ClientUpgradeRequest(); |
| client.connect(socket, echoUri, request); |
| System.out.printf("Connecting to : %s%n", echoUri); |
| socket.awaitClose(5, TimeUnit.SECONDS); |
| } catch (Throwable t) { |
| t.printStackTrace(); |
| } finally { |
| try { |
| client.stop(); |
| } catch (Exception e) { |
| e.printStackTrace(); |
| } |
| } |
| Assert.assertEquals(4, socket.getReceivedMessages().size()); |
| Assert.assertEquals("Hello", socket.getReceivedMessages().get(0)); |
| Assert.assertEquals("Thanks for the conversation.", socket.getReceivedMessages().get(1)); |
| |
| } |
| |
| private String createString(int i) { |
| StringBuilder builder = new StringBuilder(); |
| while (builder.length() < i) { |
| builder.append("A very long text."); |
| } |
| return builder.toString(); |
| } |
| } |