websocket/websocket-client/src/test/java/org/opendaylight/yangtools/websocket/WebSocketClientTest.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.yangtools.websocket;
   9
  10 import io.netty.handler.codec.http.websocketx.CloseWebSocketFrame;
  11 import java.net.URI;
  12 import java.net.URISyntaxException;
  13 import org.junit.Before;
  14 import org.junit.Test;
  15 import org.opendaylight.yangtools.websocket.client.WebSocketIClient;
  16 import org.opendaylight.yangtools.websocket.server.WebSocketServer;
  17 import org.slf4j.Logger;
  18 import org.slf4j.LoggerFactory;
  19
  20 public class WebSocketClientTest {
  21     private static final Logger logger = LoggerFactory.getLogger(WebSocketClientTest.class.toString());
  22     private static final String MESSAGE = "Take me to your leader!";
  23     private static final int port = 8080;
  24     private Thread webSocketServerThread;
  25
  26
  27     @Before
  28     public void startWebSocketServer(){
  29         try {
  30             WebSocketServer webSocketServer = new WebSocketServer(port);
  31             webSocketServerThread = new Thread(webSocketServer);
  32             webSocketServerThread.setDaemon(false);
  33             webSocketServerThread.start();
  34         } catch (Exception e) {
  35             logger.trace("Error starting websocket server");
  36         }
  37     }
  38     @Test
  39     public void connectAndSendData(){
  40
  41         URI uri = null;
  42         try {
  43             uri = new URI("ws://localhost:8080/websocket");
  44             ClientMessageCallback messageCallback = new ClientMessageCallback();
  45             WebSocketIClient wsClient = new WebSocketIClient(uri,messageCallback);
  46             try {
  47                 wsClient.connect();
  48                 wsClient.writeAndFlush(MESSAGE);
  49                 wsClient.writeAndFlush(new CloseWebSocketFrame());
  50                 webSocketServerThread.interrupt();
  51             } catch (InterruptedException e) {
  52                 logger.info("WebSocket client couldn't connect to : "+uri);
  53             }
  54         } catch (URISyntaxException e) {
  55             logger.info("There is an error in URL sytnax {}",e);
  56         }
  57     }
  58
  59     private class ClientMessageCallback implements org.opendaylight.yangtools.websocket.client.callback.ClientMessageCallback {
  60         @Override
  61         public void onMessageReceived(Object message) {
  62            logger.info("received message {}",message);
  63            System.out.println("received message : " + message);
  64         }
  65     }
  66 }