AGSA-initiated weblite intents should be rewritten if Chrome can use weblite

chrome/android/javatests/src/org/chromium/chrome/browser/TabsOpenedFromExternalAppTest.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;
 
 import android.content.Intent;
 import android.net.Uri;
 import android.os.Bundle;
 import android.os.Environment;
@@ skipping 159 matching lines @@
     protected void tearDown() throws Exception {
         mTestServer.stopAndDestroyServer();
         super.tearDown();
     }
 
     /**
      * Launch the specified URL as if it was triggered by an external application with id appId.
      * Returns when the URL has been navigated to.
      * @throws InterruptedException
      */
-    private void launchUrlFromExternalApp(String url, String appId, boolean createNewTab,
-            Bundle extras) throws InterruptedException {
+    private void launchUrlFromExternalApp(String url, String expectedUrl, String appId,
+            boolean createNewTab, Bundle extras) throws InterruptedException {
         final Intent intent = new Intent(Intent.ACTION_VIEW);
         if (appId != null) {
             intent.putExtra(Browser.EXTRA_APPLICATION_ID, appId);
         }
         if (createNewTab) {
             intent.putExtra(Browser.EXTRA_CREATE_NEW_TAB, true);
         }
         intent.setData(Uri.parse(url));
         if (extras != null) intent.putExtras(extras);
 
         final Tab originalTab = getActivity().getActivityTab();
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
                 getActivity().onNewIntent(intent);
             }
         });
         if (createNewTab) {
             CriteriaHelper.pollForUIThreadCriteria(new Criteria("Failed to select different tab") {
                 @Override
                 public boolean isSatisfied() {
                     return getActivity().getActivityTab() != originalTab;
                 }
             });
         }
-        ChromeTabUtils.waitForTabPageLoaded(getActivity().getActivityTab(), url);
+        ChromeTabUtils.waitForTabPageLoaded(getActivity().getActivityTab(), expectedUrl);
     }
 
     private void launchUrlFromExternalApp(String url, String appId, boolean createNewTab)
             throws InterruptedException {
-        launchUrlFromExternalApp(url, appId, createNewTab, null);
+        launchUrlFromExternalApp(url, url, appId, createNewTab, null);
     }
 
     /**
      * Tests that URLs opened from external apps can set an android-app scheme referrer.
      * @throws InterruptedException
      */
     @LargeTest
     @Feature({"Navigation"})
     public void testReferrer() throws InterruptedException {
         String url = mTestServer.getURL("/chrome/test/data/android/about.html");
         startMainActivityFromLauncher();
         Bundle extras = new Bundle();
         extras.putParcelable(Intent.EXTRA_REFERRER, Uri.parse(ANDROID_APP_REFERRER));
-        launchUrlFromExternalApp(url, EXTERNAL_APP_1_ID, true, extras);
+        launchUrlFromExternalApp(url, url, EXTERNAL_APP_1_ID, true, extras);
         CriteriaHelper.pollForCriteria(
                 new ReferrerCriteria(getActivity().getActivityTab(), ANDROID_APP_REFERRER), 2000,
                 200);
     }
 
     /**
      * Tests that URLs opened from external apps can set an android-app scheme referrer.
      * @throws InterruptedException
      */
     @LargeTest
     @Feature({"Navigation"})
     public void testCannotSetArbitraryReferrer() throws InterruptedException {
         String url = mTestServer.getURL("/chrome/test/data/android/about.html");
         startMainActivityFromLauncher();
         String referrer = "foobar://totally.legit.referrer";
         Bundle extras = new Bundle();
         extras.putParcelable(Intent.EXTRA_REFERRER, Uri.parse(referrer));
-        launchUrlFromExternalApp(url, EXTERNAL_APP_1_ID, true, extras);
+        launchUrlFromExternalApp(url, url, EXTERNAL_APP_1_ID, true, extras);
         CriteriaHelper.pollForCriteria(
                 new ReferrerCriteria(getActivity().getActivityTab(), ""), 2000, 200);
     }
 
     /**
      * Tests that URLs opened from the same external app don't create new tabs.
      * @throws InterruptedException
      */
     @LargeTest
     @Feature({"Navigation"})
@@ skipping 370 matching lines @@
                 getActivity().onBackPressed();
             }
         });
         CriteriaHelper.pollForUIThreadCriteria(new Criteria() {
             @Override
             public boolean isSatisfied() {
                 return getActivity().getTabModelSelector().getTotalTabCount() == 1;
             }
         });
     }
+
+    /**
+     * Tests that a Weblite url from an external app uses the lite_url param when Data Reduction
+     * Proxy previews are being used.
+     */
+    @MediumTest
+    @CommandLineFlags.Add({"enable-spdy-proxy-auth", "data-reduction-proxy-lo-fi=always-on",
+            "enable-data-reduction-proxy-lo-fi-preview"})
+    public void testLaunchWebLiteURL() throws InterruptedException {
+        startMainActivityFromLauncher();
+
+        String url = mTestServer.getURL("/chrome/test/data/android/about.html");
+
+        // Launch a first URL from an app.
+        launchUrlFromExternalApp("http://googleweblight.com/?lite_url=" + url, url,
+                EXTERNAL_APP_1_ID, false, null);
+
+        assertEquals("Selected tab is not on the right URL.",
+                url, getActivity().getActivityTab().getUrl());
+    }
+
+    /**
+     * Tests that a Weblite url from an external app does not use the lite_url param when Data
+     * Reduction Proxy previews are not being used.
+     */
+    @MediumTest
+    public void testLaunchWebLiteURLNoPreviews() throws InterruptedException {
+        startMainActivityFromLauncher();
+
+        String url = "http://googleweblight.com/?lite_url=chrome/test/data/android/about.html";
+
+        // Launch a first URL from an app.
+        launchUrlFromExternalApp(url, url, EXTERNAL_APP_1_ID, false, null);
+
+        assertEquals("Selected tab is not on the right URL.",
+                url, getActivity().getActivityTab().getUrl());
+    }
 }