Added test that prestes working with websockets

jacek-marchwicki · jacek-marchwicki · commit a3ebf26da2a2 · 2014-09-30T16:09:33.000+02:00
Change-Id: I85a423a1f9f3c9e3b1e57fc214b3c1bc38882aea
diff --git a/AndroidSocketIO/src/androidTest/java/com/appunite/websocket/WebsocketTest.java b/AndroidSocketIO/src/androidTest/java/com/appunite/websocket/WebsocketTest.java
@@ -0,0 +1,139 @@
+package com.appunite.websocket;
+
+import android.net.Uri;
+import android.test.AndroidTestCase;
+
+import com.appunite.socketio.NotConnectedException;
+
+import org.hamcrest.Matchers;
+import org.json.JSONObject;
+import org.mockito.ArgumentCaptor;
+
+import java.io.IOException;
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.AtomicReference;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.*;
+import static org.mockito.Matchers.argThat;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.reset;
+import static org.mockito.Mockito.timeout;
+import static org.mockito.Mockito.verify;
+
+public class WebsocketTest extends AndroidTestCase {
+
+    private Uri mUri;
+    private WebSocketListener mListener;
+
+    @Override
+    public void setUp() throws Exception {
+        super.setUp();
+        mUri = Uri.parse("wss://some.test.website.com/");
+        mListener = mock(WebSocketListener.class);
+    }
+
+    public void testPreconditions() throws Exception {
+        assertThat(mUri, is(notNullValue()));
+    }
+
+    public void testNoError() throws Exception {
+        final WebSocket webSocket = new WebSocket(mListener);
+        final AtomicReference<Exception> reference = new AtomicReference<Exception>();
+        new Thread(new Runnable() {
+            @Override
+            public void run() {
+                try {
+                    webSocket.connect(mUri);
+                } catch (IOException e) {
+                    reference.set(e);
+                } catch (WrongWebsocketResponse e) {
+                    reference.set(e);
+                } catch (InterruptedException e) {
+                }
+            }
+        }).start();
+
+        Thread.sleep(2000);
+        final Exception exception = reference.get();
+        if (exception != null) {
+            throw exception;
+        }
+
+        webSocket.interrupt();
+    }
+
+    public void testConnection() throws Exception {
+        final WebSocket webSocket = new WebSocket(mListener);
+
+        final AtomicBoolean interrupted = new AtomicBoolean(false);
+        new Thread(new Runnable() {
+            @Override
+            public void run() {
+                try {
+                    webSocket.connect(mUri);
+                } catch (IOException ignore) {
+                } catch (WrongWebsocketResponse ignore) {
+                } catch (InterruptedException e) {
+                    interrupted.set(true);
+                }
+            }
+        }).start();
+
+        verify(mListener, timeout(2000)).onConnected();
+
+        webSocket.interrupt();
+        Thread.sleep(1000);
+
+        assertThat(interrupted.get(), Matchers.is(equalTo(true)));
+    }
+
+    public void testSendData() throws Exception {
+        final WebSocket webSocket = authorize();
+
+        webSocket.interrupt();
+    }
+
+    public void testRequestUsers() throws Exception {
+        final WebSocket webSocket = authorize();
+
+        webSocket.sendStringMessage("{\"action\": \"observe\", \"users\": [12354]}");
+        verify(mListener, timeout(2000)).onStringMessage(argThat(containsString("\"12354\"")));
+
+        webSocket.interrupt();
+    }
+
+    private WebSocket connectWebsocket() throws IOException, InterruptedException, NotConnectedException {
+        final WebSocket webSocket = new WebSocket(mListener);
+
+        new Thread(new Runnable() {
+            @Override
+            public void run() {
+                try {
+                    webSocket.connect(mUri);
+                } catch (IOException ignore) {
+                } catch (WrongWebsocketResponse ignore) {
+                } catch (InterruptedException ignore) {
+                }
+            }
+        }).start();
+
+        verify(mListener, timeout(2000)).onConnected();
+        reset(mListener);
+        return webSocket;
+    }
+
+    private WebSocket authorize() throws IOException, InterruptedException, NotConnectedException {
+        final WebSocket webSocket = connectWebsocket();
+
+        webSocket.sendStringMessage("{\"action\": \"auth\", \"auth_token\": \"1ZizC3fpskE3YEHexfgX\"}");
+
+        final ArgumentCaptor<String> responseMessage = ArgumentCaptor.forClass(String.class);
+        verify(mListener, timeout(2000)).onStringMessage(responseMessage.capture());
+        reset(mListener);
+
+        assertThat(responseMessage.getValue(), is(containsString("auth")));
+        assertThat(responseMessage.getValue(), is(containsString("OK")));
+        return webSocket;
+    }
+}
