Implementing RTC debugging session objects (client and server parts).

components/devtools_bridge/android/javatests/src/org/chromium/components/devtools_bridge/SessionControlMessagesTest.java

Index: components/devtools_bridge/android/javatests/src/org/chromium/components/devtools_bridge/SessionControlMessagesTest.java
diff --git a/components/devtools_bridge/android/javatests/src/org/chromium/components/devtools_bridge/SessionControlMessagesTest.java b/components/devtools_bridge/android/javatests/src/org/chromium/components/devtools_bridge/SessionControlMessagesTest.java
new file mode 100644
index 0000000000000000000000000000000000000000..a31816865333fb11b47bd283dafffaffcedeefd1
--- /dev/null
+++ b/components/devtools_bridge/android/javatests/src/org/chromium/components/devtools_bridge/SessionControlMessagesTest.java
@@ -0,0 +1,94 @@
+// Copyright 2014 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+package org.chromium.components.devtools_bridge;
+
+import android.test.InstrumentationTestCase;
+import android.test.suitebuilder.annotation.SmallTest;
+
+import junit.framework.Assert;
+
+import org.chromium.components.devtools_bridge.SessionControlMessages.ClientMessage;
+import org.chromium.components.devtools_bridge.SessionControlMessages.ClientMessageHandler;
+import org.chromium.components.devtools_bridge.SessionControlMessages.IceExchangeMessage;
+import org.chromium.components.devtools_bridge.SessionControlMessages.InvalidFormatException;
+import org.chromium.components.devtools_bridge.SessionControlMessages.Message;
+import org.chromium.components.devtools_bridge.SessionControlMessages.MessageHandler;
+import org.chromium.components.devtools_bridge.SessionControlMessages.ServerMessage;
+import org.chromium.components.devtools_bridge.SessionControlMessages.ServerMessageHandler;
+import org.chromium.components.devtools_bridge.SessionControlMessages.UnknownRequestMessage;
+import org.chromium.components.devtools_bridge.SessionControlMessages.UnknownResponseMessage;
+
+/**
+ * Tests for {@link SessionControlMessages}
+ */
+public class SessionControlMessagesTest extends InstrumentationTestCase {
+    private static final String UNKNOWN_REQUEST_TYPE = "@unknown request@";
+    private static final String UNKNOWN_REQUEST = "{\"type\": \"" + UNKNOWN_REQUEST_TYPE + "\"}";
+
+    @SmallTest
+    public void testIceExchangeMessage() {
+        recode(new IceExchangeMessage());
+    }
+
+    @SmallTest
+    public void testUnknownRequest() throws InvalidFormatException {
+        UnknownRequestMessage request =
+                (UnknownRequestMessage) ClientMessageReader.readMessage(UNKNOWN_REQUEST);
+        UnknownResponseMessage response = ServerMessageReader.recode(request.createResponse());
+        Assert.assertEquals(UNKNOWN_REQUEST_TYPE, response.rawRequestType);
+    }
+
+    private <T extends ServerMessage> T recode(T message) {
+        assert message != null;
+        return ServerMessageReader.recode(message);
+    }
+
+    @SuppressWarnings("unchecked")
+    private static <T> T cast(T prototype, Object object) {
+        Assert.assertNotNull(object);
+        if (prototype.getClass() == object.getClass())
+            return (T) object;
+        else
+            throw new ClassCastException();
+    }
+
+    private static void checkedRead(MessageHandler handler, Message<?> message) {
+        try {
+            handler.readMessage(SessionControlMessages.toByteArray(message));
+        } catch (InvalidFormatException e) {
+            Assert.fail(e.toString());
+        }
+    }
+
+    private static class ServerMessageReader extends ServerMessageHandler {
+        private ServerMessage mLastMessage;
+
+        @Override
+        protected void onMessage(ServerMessage message) {
+            mLastMessage = message;
+        }
+
+        public static <T extends ServerMessage> T recode(T message) {
+            ServerMessageReader handler = new ServerMessageReader();
+            checkedRead(handler, message);
+            return cast(message, handler.mLastMessage);
+        }
+    }
+
+    private static class ClientMessageReader extends ClientMessageHandler {
+        private ClientMessage mLastMessage;
+
+        @Override
+        protected void onMessage(ClientMessage message) {
+            mLastMessage = message;
+        }
+
+        public static ClientMessage readMessage(String json) throws InvalidFormatException {
+            ClientMessageReader reader = new ClientMessageReader();
+            reader.readMessage(json.getBytes());
+            return reader.mLastMessage;
+        }
+    }
+}