HTML MessagePort as mojo::MessagePipeHandle

android_webview/javatests/src/org/chromium/android_webview/test/PostMessageTest.java

 // 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.android_webview.test;
 
 import static org.chromium.base.test.util.ScalableTimeout.scaleTimeout;
 
-import android.os.Handler;
 import android.support.test.filters.SmallTest;
 import android.webkit.JavascriptInterface;
 
 import org.chromium.android_webview.AwContents;
 import org.chromium.android_webview.test.util.CommonResources;
 import org.chromium.base.ThreadUtils;
 import org.chromium.base.annotations.SuppressFBWarnings;
 import org.chromium.base.test.util.DisabledTest;
 import org.chromium.base.test.util.Feature;
 import org.chromium.base.test.util.RetryOnFailure;
 import org.chromium.content.browser.AppWebMessagePort;
-import org.chromium.content.browser.AppWebMessagePortService;
 import org.chromium.content.browser.test.util.Criteria;
 import org.chromium.content.browser.test.util.CriteriaHelper;
 import org.chromium.content.browser.test.util.TestCallbackHelperContainer.OnPageFinishedHelper;
 import org.chromium.content_public.browser.MessagePort;
 import org.chromium.net.test.util.TestWebServer;
 
 import java.util.concurrent.Callable;
 import java.util.concurrent.CountDownLatch;
 
 /**
@@ skipping 462 matching lines @@
         });
         boolean ignore = latch.await(TIMEOUT, java.util.concurrent.TimeUnit.MILLISECONDS);
     }
 
     // Create two message channels, and while they are in pending state, transfer the
     // second one in the first one.
     @SmallTest
     @Feature({"AndroidWebView", "Android-PostMessage"})
     public void testPendingPortCanBeTransferredInPendingPort() throws Throwable {
         loadPage(TITLE_FROM_POSTMESSAGE_TO_CHANNEL);
-        final TestMessagePort testPort =
-                new TestMessagePort(getAwBrowserContext().getMessagePortService());
         runTestOnUiThread(new Runnable() {
             @Override
             public void run() {
                 AppWebMessagePort[] channel1 = mAwContents.createMessageChannel();
                 mAwContents.postMessageToFrame(
                         null, "1", mWebServer.getBaseUrl(), new AppWebMessagePort[] {channel1[1]});
                 AppWebMessagePort[] channel2 = mAwContents.createMessageChannel();
                 channel1[0].postMessage("2", new AppWebMessagePort[] {channel2[0]});
             }
         });
@@ skipping 279 matching lines @@
                             }
                         }, null);
                         p[0].postMessage("2", null);
                     }
                 }, null);
             }
         });
         expectTitle("24");
     }
 
-    private static class TestMessagePort extends AppWebMessagePort {
-        private boolean mReady;
-        private AppWebMessagePort mPort;
-        private final Object mLock = new Object();
-
-        public TestMessagePort(AppWebMessagePortService service) {
-            super(service);
-        }
-
-        public void setMessagePort(AppWebMessagePort port) {
-            mPort = port;
-        }
-
-        public void setReady(boolean ready) {
-            synchronized (mLock) {
-                mReady = ready;
-            }
-        }
-        @Override
-        public boolean isReady() {
-            synchronized (mLock) {
-                return mReady;
-            }
-        }
-        @Override
-        public int portId() {
-            return mPort.portId();
-        }
-        @Override
-        public void setPortId(int id) {
-            mPort.setPortId(id);
-        }
-        @Override
-        public void close() {
-            mPort.close();
-        }
-        @Override
-        public boolean isClosed() {
-            return mPort.isClosed();
-        }
-        @Override
-        public void setMessageCallback(MessageCallback messageCallback, Handler handler) {
-            mPort.setMessageCallback(messageCallback, handler);
-        }
-        @Override
-        public void onMessage(String message, AppWebMessagePort[] sentPorts) {
-            mPort.onMessage(message, sentPorts);
-        }
-        @Override
-        public void postMessage(String message, MessagePort[] sentPorts)
-                throws IllegalStateException {
-            mPort.postMessage(message, sentPorts);
-        }
-    }
-
-    // Post a message with a pending port to a frame and then post a message that
-    // is pending after that. Make sure that when first message becomes ready,
-    // the subsequent not-ready message is not sent.
-    @SmallTest
-    @Feature({"AndroidWebView", "Android-PostMessage"})
-    public void testPostMessageToFrameNotSendsPendingMessages() throws Throwable {
-        loadPage(TITLE_FROM_POSTMESSAGE_TO_FRAME);
-        final TestMessagePort testPort =
-                new TestMessagePort(getAwBrowserContext().getMessagePortService());
-        runTestOnUiThread(new Runnable() {
-            @Override
-            public void run() {
-                AppWebMessagePort[] channel = mAwContents.createMessageChannel();
-                mAwContents.postMessageToFrame(
-                        null, "1", mWebServer.getBaseUrl(), new AppWebMessagePort[] {channel[1]});
-                mAwContents.postMessageToFrame(null, "2", mWebServer.getBaseUrl(), null);
-                AppWebMessagePort[] channel2 = mAwContents.createMessageChannel();
-                // Test port is in a pending state so it should not be transferred.
-                testPort.setMessagePort(channel2[0]);
-                mAwContents.postMessageToFrame(
-                        null, "3", mWebServer.getBaseUrl(), new AppWebMessagePort[] {testPort});
-            }
-        });
-        expectTitle("12");
-    }
-
     private static final String WORKER_MESSAGE = "from_worker";
 
     // Listen for messages. Pass port 1 to worker and use port 2 to receive messages from
     // from worker.
     private static final String TEST_PAGE_FOR_PORT_TRANSFER =
             "<!DOCTYPE html><html><body>"
             + "    <script>"
             + "        var worker = new Worker(\"worker.js\");"
             + "        onmessage = function (e) {"
             + "            if (e.data == \"" + WEBVIEW_MESSAGE + "\") {"
@@ skipping 195 matching lines @@
                     }
                 }, null);
                 mAwContents.postMessageToFrame(null, WEBVIEW_MESSAGE, mWebServer.getBaseUrl(),
                         new AppWebMessagePort[] {channel[1]});
             }
         });
         channelContainer.waitForMessage();
         assertEquals("12", channelContainer.getMessage());
     }
 }