customtabs: Add a test for external app redirection.

chrome/android/javatests/src/org/chromium/chrome/browser/customtabs/CustomTabExternalNavigationTest.java

 // Copyright 2015 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.chrome.browser.customtabs;
 
 import android.app.Activity;
 import android.os.Bundle;
 import android.support.test.filters.SmallTest;
 
 import org.chromium.base.test.util.RetryOnFailure;
 import org.chromium.chrome.browser.customtabs.CustomTabDelegateFactory.CustomTabNavigationDelegate;
 import org.chromium.chrome.browser.externalnav.ExternalNavigationHandler;
 import org.chromium.chrome.browser.externalnav.ExternalNavigationHandler.OverrideUrlLoadingResult;
 import org.chromium.chrome.browser.externalnav.ExternalNavigationParams;
 import org.chromium.chrome.browser.tab.Tab;
 import org.chromium.chrome.browser.tab.TabDelegateFactory;
+import org.chromium.content.browser.test.util.Criteria;
+import org.chromium.content.browser.test.util.CriteriaHelper;
+import org.chromium.content.browser.test.util.DOMUtils;
 import org.chromium.net.test.EmbeddedTestServer;
 
 /**
- * Instrumentation test for external navigation handling of a Custom Tab.
+ * Instrumentation tests for external navigation handling of a Custom Tab.
  */
 @RetryOnFailure
 public class CustomTabExternalNavigationTest extends CustomTabActivityTestBase {
-
     /**
      * A dummy activity that claims to handle "customtab://customtabtest".
      */
     public static class DummyActivityForSpecialScheme extends Activity {
         @Override
         protected void onCreate(Bundle savedInstanceState) {
             super.onCreate(savedInstanceState);
             finish();
         }
     }
@@ skipping 19 matching lines @@
         mTestServer = EmbeddedTestServer.createAndStartServer(getInstrumentation().getContext());
         super.setUp();
     }
 
     @Override
     public void tearDown() throws Exception {
         mTestServer.stopAndDestroyServer();
         super.tearDown();
     }
 
-    @Override
-    public void startMainActivity() throws InterruptedException {
-        super.startMainActivity();
-        startCustomTabActivityWithIntent(CustomTabsTestUtils.createMinimalCustomTabIntent(
-                getInstrumentation().getTargetContext(), mTestServer.getURL(TEST_PATH)));
-        Tab tab = getActivity().getActivityTab();
-        TabDelegateFactory delegateFactory = tab.getDelegateFactory();
-        assertTrue(delegateFactory instanceof CustomTabDelegateFactory);
-        CustomTabDelegateFactory customTabDelegateFactory =
-                ((CustomTabDelegateFactory) delegateFactory);
-        mUrlHandler = customTabDelegateFactory.getExternalNavigationHandler();
-        assertTrue(customTabDelegateFactory.getExternalNavigationDelegate()
-                instanceof CustomTabNavigationDelegate);
-        mNavigationDelegate = (CustomTabNavigationDelegate) customTabDelegateFactory
-                .getExternalNavigationDelegate();
-    }
-
     /**
      * For urls with special schemes and hosts, and there is exactly one activity having a matching
      * intent filter, the framework will make that activity the default handler of the special url.
      * This test tests whether chrome is able to start the default external handler.
      */
     @SmallTest
-    public void testExternalActivityStartedForDefaultUrl() {
+    public void testExternalActivityStartedForDefaultUrl() throws Exception {
+        startCustomTabActivity(mTestServer.getURL(TEST_PATH));
         final String testUrl = "customtab://customtabtest/intent";
         ExternalNavigationParams params = new ExternalNavigationParams.Builder(testUrl, false)
                 .build();
         OverrideUrlLoadingResult result = mUrlHandler.shouldOverrideUrlLoading(params);
         assertEquals(OverrideUrlLoadingResult.OVERRIDE_WITH_EXTERNAL_INTENT, result);
         assertTrue("A dummy activity should have been started to handle the special url.",
                 mNavigationDelegate.hasExternalActivityStarted());
     }
 
     /**
      * When loading a normal http url that chrome is able to handle, an intent picker should never
      * be shown, even if other activities such as {@link DummyActivityForHttp} claim to handle it.
      */
     @SmallTest
-    public void testIntentPickerNotShownForNormalUrl() {
+    public void testIntentPickerNotShownForNormalUrl() throws Exception {
+        startCustomTabActivity(mTestServer.getURL(TEST_PATH));
         final String testUrl = "http://customtabtest.com";
         ExternalNavigationParams params = new ExternalNavigationParams.Builder(testUrl, false)
                 .build();
         OverrideUrlLoadingResult result = mUrlHandler.shouldOverrideUrlLoading(params);
         assertEquals(OverrideUrlLoadingResult.NO_OVERRIDE, result);
         assertFalse("External activities should not be started to handle the url",
                 mNavigationDelegate.hasExternalActivityStarted());
     }
+
+    /**
+     * Launches a Custom Tab, clicks on an external app scheme link, and check that an external
+     * activity has been launched.
+     */
+    @SmallTest
+    public void testExternalApplicationLink() throws Exception {
+        warmUpAndWait();
+        startCustomTabActivity(mTestServer.getURL(TEST_PATH));
+
+        Tab tab = getActivity().getActivityTab();
+        assertTrue(DOMUtils.clickNode(tab.getContentViewCore(), "customtab-external-link"));
+
+        CriteriaHelper.pollUiThread(new Criteria("External Activity not launched.") {
+            @Override
+            public boolean isSatisfied() {
+                return mNavigationDelegate.hasExternalActivityStarted();
+            }
+        });
+    }
+
+    private void startCustomTabActivity(String url) throws InterruptedException {
+        super.startMainActivity();
+        startCustomTabActivityWithIntent(CustomTabsTestUtils.createMinimalCustomTabIntent(
+                getInstrumentation().getTargetContext(), url));
+        Tab tab = getActivity().getActivityTab();
+        TabDelegateFactory delegateFactory = tab.getDelegateFactory();
+        assertTrue(delegateFactory instanceof CustomTabDelegateFactory);
+        CustomTabDelegateFactory customTabDelegateFactory =
+                ((CustomTabDelegateFactory) delegateFactory);
+        mUrlHandler = customTabDelegateFactory.getExternalNavigationHandler();
+        assertTrue(customTabDelegateFactory.getExternalNavigationDelegate()
+                           instanceof CustomTabNavigationDelegate);
+        mNavigationDelegate = (CustomTabNavigationDelegate)
+                                      customTabDelegateFactory.getExternalNavigationDelegate();
+    }
 }