Upstream JavaBridge tests.

content/shell/android/javatests/src/org/chromium/content_shell/ContentShellTestBase.java

 // Copyright (c) 2012 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.content_shell;
 
 import android.content.ComponentName;
 import android.content.Intent;
 import android.net.Uri;
 import android.test.ActivityInstrumentationTestCase2;
 
+import org.chromium.content.browser.ContentView;
+import org.chromium.content.browser.ContentViewClient;
+import org.chromium.content.browser.LoadUrlParams;
+import org.chromium.content.browser.test.TestContentViewClient;
+
+import java.util.HashMap;
+import java.util.Map;
+import java.util.concurrent.TimeUnit;
+
 /**
  * Base test class for all ContentShell based tests.
  */
 public class ContentShellTestBase extends ActivityInstrumentationTestCase2<ContentShellActivity> {
+    private Map<ContentView, TestContentViewClient> mTestContentViewClients
+            = new HashMap<ContentView, TestContentViewClient>();

[klundberg, 2012/09/07 16:15:29]

I don't think you should add this to ContentShellTestBase as not all sub-classes
will need this.

Instead, create another ContentViewTestBase class and add the ContentView
specific things to this class.

Otherwise, we will just end up in the same situation as MainActivityTestBase is
in.

[Ramya, 2012/09/19 21:31:57]

Done.

+
+    protected static int WAIT_TIMEOUT_SECONDS = 15;
 
     public ContentShellTestBase() {
         super(ContentShellActivity.class);
     }
 
     /**
      * Starts the ContentShell activity and loads the given URL.
      */
     protected ContentShellActivity launchContentShellWithUrl(String url) {
         Intent intent = new Intent(Intent.ACTION_MAIN);
         intent.addCategory(Intent.CATEGORY_LAUNCHER);
         intent.addFlags(Intent.FLAG_ACTIVITY_NEW_TASK);
         intent.setData(Uri.parse(url));
         intent.setComponent(new ComponentName(getInstrumentation().getTargetContext(),
               ContentShellActivity.class));
-        return (ContentShellActivity) getInstrumentation().startActivitySync(intent);
+        setActivityIntent(intent);
+        return getActivity();
+    }
+
+    /**
+     * Sets the {@link TestContentViewClient} associated with the specified
+     * {@link ContentView}.
+     *
+     * Must be called on the UI thread.
+     */
+    public void setTestContentViewClient(
+            final ContentView contentView, final TestContentViewClient client) {
+        contentView.setContentViewClient(client);
+        mTestContentViewClients.put(contentView, client);
+    }
+
+    /**
+     * Gets the {@link TestContentViewClient} associated with the specified
+     * {@link ContentView}.
+     *
+     * This will not work correctly if the test code invokes
+     * {link {@link ContentView#setContentViewClient(ContentViewClient)} directly.
+     *
+     * @return {@link TestContentViewClient}
+     */
+    public TestContentViewClient getTestContentViewClient(ContentView contentView) {
+        return mTestContentViewClients.get(contentView);
+    }
+
+    /**
+     * Loads data on the UI thread and blocks until onPageFinished is called.
+     */
+    @Deprecated
+    protected void loadDataSync(final ContentView chromeView, final String data,
+                                final String mimeType, final boolean isBase64Encoded)
+            throws Throwable {
+        TestContentViewClient.OnPageFinishedHelper onPageFinishedHelper =
+                getTestContentViewClient(chromeView).getOnPageFinishedHelper();
+        int currentCallCount = onPageFinishedHelper.getCallCount();
+        loadDataAsync(chromeView, data, mimeType, isBase64Encoded);
+        onPageFinishedHelper.waitForCallback(currentCallCount, 1, WAIT_TIMEOUT_SECONDS,
+                TimeUnit.SECONDS);

[klundberg, 2012/09/07 16:15:30]

This should be in a utility class, not in a test base class.

[Ramya, 2012/09/19 21:31:57]

As discussed, leaving it here till the Util files are moved to base. 

+    }
+
+    /**
+     * Loads data on the UI thread but does not block.
+     */
+    @Deprecated
+    protected void loadDataAsync(final ContentView chromeView, final String data,
+                                 final String mimeType, final boolean isBase64Encoded)
+            throws Throwable {

[klundberg, 2012/09/07 16:15:30]

Same comment as above.

[Ramya, 2012/09/19 21:31:57]

See comment above. 

+        runTestOnUiThread(new Runnable() {
+            @Override
+            public void run() {
+                chromeView.loadUrl(LoadUrlParams.createLoadDataParams(
+                        data, mimeType, isBase64Encoded));
+            }
+        });
+    }
+
+    /**
+     * Allows access to the {@link ContentView}.
+     *
+     * @return The first {@link ContentView}
+     */
+    public ContentView getContentView() {
+        return getActivity().getActiveContentView();
     }
 }