SocketRealTest.java

/*
 * Copyright (C) 2015 Jacek Marchwicki <jacek.marchwicki@gmail.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License
 */

package com.example;

import com.appunite.websocket.rx.RxWebSockets;
import com.appunite.websocket.rx.object.GsonObjectSerializer;
import com.appunite.websocket.rx.object.RxObjectWebSockets;
import com.example.model.DataMessage;
import com.example.model.Message;
import com.example.model.MessageType;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import okhttp3.OkHttpClient;
import okhttp3.Request;

import org.junit.Before;
import org.junit.Ignore;
import org.junit.Test;

import java.util.logging.Logger;

import rx.Observable;
import rx.Subscription;
import rx.functions.Func1;
import rx.schedulers.Schedulers;

public class SocketRealTest {

 private Socket socket;

 @Before
 public void setUp() throws Exception {
 final Gson gson = new GsonBuilder()
 .registerTypeAdapter(Message.class, new Message.Deserializer())
 .registerTypeAdapter(MessageType.class, new MessageType.SerializerDeserializer())
 .create();

 final RxWebSockets webSockets = new RxWebSockets(new OkHttpClient(),
 new Request.Builder()
 .get()
 .url("ws://10.10.0.2:8080/ws")
 .addHeader("Sec-WebSocket-Protocol", "chat")
 .build());
 final RxObjectWebSockets jsonWebSockets = new RxObjectWebSockets(webSockets, new GsonObjectSerializer(gson, Message.class));
 final SocketConnection socketConnection = new SocketConnectionImpl(jsonWebSockets, Schedulers.computation());
 socket = new Socket(socketConnection, Schedulers.computation());

 }

 @Test
 @Ignore
 public void testName() throws Exception {
 socket.sendMessageOnceWhenConnected(new Func1<String, Observable<Object>>() {
 @Override
 public Observable<Object> call(String messageId) {
 return Observable.<Object>just(new DataMessage(messageId, "some message"));
 }
 })
 .subscribe(LoggingObservables.logging(Logger.getLogger("Rx"), "SendMessage"));

 socket.sendPingWhenConnected();
 socket.sendPingEvery5seconds();
 final Subscription subscribe = socket.connection()
 .subscribeOn(Schedulers.io())
 .subscribe();
 Thread.sleep(10000);
 subscribe.unsubscribe();
 Thread.sleep(10000);
 }

}