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

chrome/android/javatests/src/org/chromium/chrome/browser/document/DocumentModeTest.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.document;
 
 import android.app.Activity;
 import android.app.Instrumentation;
 import android.app.PendingIntent;
 import android.content.Intent;
 import android.net.Uri;
 import android.os.Build;
 import android.test.FlakyTest;
 import android.test.suitebuilder.annotation.MediumTest;
+import android.test.suitebuilder.annotation.SmallTest;
 import android.text.TextUtils;
 import android.view.View;
 
 import org.chromium.base.ApplicationStatus;
 import org.chromium.base.ThreadUtils;
+import org.chromium.base.test.util.CommandLineFlags;
 import org.chromium.base.test.util.MinAndroidSdkLevel;
 import org.chromium.base.test.util.UrlUtils;
 import org.chromium.chrome.browser.ChromeActivity;
 import org.chromium.chrome.browser.ChromeApplication;
 import org.chromium.chrome.browser.ChromeTabbedActivity;
 import org.chromium.chrome.browser.IntentHandler;
 import org.chromium.chrome.browser.ShortcutHelper;
 import org.chromium.chrome.browser.tab.Tab;
 import org.chromium.chrome.browser.tabmodel.TabModel;
+import org.chromium.chrome.browser.tabmodel.TabModelSelector;
 import org.chromium.chrome.browser.tabmodel.TabModelUtils;
 import org.chromium.chrome.browser.tabmodel.document.DocumentTabModelSelector;
 import org.chromium.chrome.browser.tabmodel.document.OffTheRecordDocumentTabModel;
 import org.chromium.chrome.test.util.ActivityUtils;
 import org.chromium.chrome.test.util.ApplicationTestUtils;
 import org.chromium.chrome.test.util.DisableInTabbedMode;
+import org.chromium.chrome.test.util.TestHttpServerClient;
 import org.chromium.content.browser.test.util.Criteria;
 import org.chromium.content.browser.test.util.CriteriaHelper;
 import org.chromium.content.browser.test.util.TouchCommon;
 import org.chromium.content_public.browser.LoadUrlParams;
 
 import java.lang.ref.WeakReference;
 import java.util.List;
 
 /**
  * General tests for how Document mode Activities interact with each other.
@@ skipping 598 matching lines @@
      * set to null immediately afterward.
      */
     @MediumTest
     public void testWindowOpenWithOpenerSuppressed() throws Exception {
         Intent lastIntent = performNewWindowTest(ONCLICK_NO_REFERRER_LINK,
                 "window.open page, opener set to null", true, "Page 4", false);
         assertEquals("Intent wasn't fired with about:blank",
                 "about:blank", IntentHandler.getUrlFromIntent(lastIntent));
     }
 
+    /**
+     * Tests that a Weblite url from an external app uses the lite_url param when Data Reduction
+     * Proxy previews are being used.
+     */
+    @SmallTest
+    @CommandLineFlags.Add({"enable-spdy-proxy-auth", "data-reduction-proxy-lo-fi=always-on",
+            "enable-data-reduction-proxy-lo-fi-preview"})
+    public void testLaunchWebLiteURL() throws Exception {
+        String url = TestHttpServerClient.getUrl("chrome/test/data/android/about.html");
+        launchViaViewIntent(false, "http://googleweblight.com/?lite_url=" + url,
+                "Page 1");
+
+        TabModelSelector selector = ChromeApplication.getDocumentTabModelSelector();
+        assertEquals(1, selector.getTotalTabCount());
+        assertEquals(url, selector.getModel(false).getTabAt(0).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.
+     */
+    @SmallTest
+    public void testLaunchWebLiteURLNoPreviews() throws Exception {
+        String url = TestHttpServerClient.getUrl(
+                "http://googleweblight.com/?lite_url=chrome/test/data/android/about.html");
+        launchViaViewIntent(false, url, "Page 1");
+
+        TabModelSelector selector = ChromeApplication.getDocumentTabModelSelector();
+        assertEquals(1, selector.getTotalTabCount());
+        assertEquals(url, selector.getModel(false).getTabAt(0).getUrl());
+    }
+
     private Intent performNewWindowTest(String url, String title, boolean checkWindowOpenSuccess,
             String openTitle, boolean waitLongerForOpenedPage) throws Exception {
         launchViaLaunchDocumentInstance(false, url, title);
 
         final DocumentActivity firstActivity =
                 (DocumentActivity) ApplicationStatus.getLastTrackedFocusedActivity();
 
         // Save the current tab ID.
         final DocumentTabModelSelector selector =
                 ChromeApplication.getDocumentTabModelSelector();
@@ skipping 25 matching lines @@
                 firstActivity, ApplicationStatus.getLastTrackedFocusedActivity());
 
         if (checkWindowOpenSuccess) {
             assertEquals("New WebContents was not created",
                     SUCCESS_URL, firstActivity.getActivityTab().getUrl());
         }
 
         return lastActivity.getIntent();
     }
 }