Fixed compilation

jacek-marchwicki · jacek-marchwicki · commit 61116cbbefaa · 2015-08-31T17:07:20.000+02:00
Change-Id: Ib8d699d6743cfb605fc7eda0aaf9ee1056c84d90
diff --git a/websockets-rxjava-example/src/test/java/com/example/RxJsonWebSocketsRealTest.java b/websockets-rxjava-example/src/test/java/com/example/RxJsonWebSocketsRealTest.java
@@ -18,6 +18,7 @@
 
 import com.appunite.websocket.NewWebSocket;
 import com.appunite.websocket.rx.RxWebSockets;
+import com.appunite.websocket.rx.object.GsonObjectSerializer;
 import com.appunite.websocket.rx.object.RxObjectWebSockets;
 import com.example.model.Message;
 import com.example.model.MessageType;
@@ -63,7 +64,7 @@ public void setUp() throws Exception {
                 SERVER_URI,
                 ImmutableList.of("chat"),
                 ImmutableList.<Header>of());
-        socket = new RxObjectWebSockets(rxWebSockets, gson, Message.class);;
+        socket = new RxObjectWebSockets(rxWebSockets, new GsonObjectSerializer(gson, Message.class));
     }
 
     @Test
diff --git a/websockets-rxjava-example/src/test/java/com/example/SocketRealTest.java b/websockets-rxjava-example/src/test/java/com/example/SocketRealTest.java
@@ -18,6 +18,7 @@
 
 import com.appunite.websocket.NewWebSocket;
 import com.appunite.websocket.rx.*;
+import com.appunite.websocket.rx.object.GsonObjectSerializer;
 import com.appunite.websocket.rx.object.RxObjectWebSockets;
 import com.example.model.DataMessage;
 import com.example.model.MessageType;
@@ -67,7 +68,7 @@ public void setUp() throws Exception {
                 SERVER_URI,
                 ImmutableList.of("chat"),
                 ImmutableList.<Header>of());
-        final RxObjectWebSockets jsonWebSockets = new RxObjectWebSockets(webSockets, gson, Message.class);
+        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());
 
diff --git a/websockets-rxjava-example/src/test/java/com/example/SocketTest.java b/websockets-rxjava-example/src/test/java/com/example/SocketTest.java
@@ -17,6 +17,7 @@
 package com.example;
 
 import com.appunite.websocket.NotConnectedException;
+import com.appunite.websocket.rx.object.ObjectParseException;
 import com.appunite.websocket.rx.object.ObjectWebSocketSender;
 import com.appunite.websocket.rx.object.messages.RxObjectEvent;
 import com.appunite.websocket.rx.object.messages.RxObjectEventConnected;
@@ -82,7 +83,8 @@ public void testConnection_registerIsSent() throws Exception {
         }
     }
 
-    private void register() throws IOException, InterruptedException, NotConnectedException {
+    private void register() throws IOException, InterruptedException, NotConnectedException,
+            ObjectParseException {
         connection.onNext(new RxObjectEventConnected(sender), 0);
         testScheduler.triggerActions();
         verify(sender).sendObjectMessage(new RegisterMessage("asdf"));
