Prerender: exclude AGSA from the PrerenderSilence experiment

chrome/browser/prerender/prerender_unittest.cc

 // 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.
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <map>
 #include <utility>
 
@@ skipping 1213 matching lines @@
           url,
           ORIGIN_EXTERNAL_REQUEST,
           FINAL_STATUS_MANAGER_SHUTDOWN);
   std::unique_ptr<PrerenderHandle> prerender_handle(
       prerender_manager()->AddPrerenderFromExternalRequest(
           url, content::Referrer(), nullptr, gfx::Rect(kSize)));
   EXPECT_FALSE(prerender_handle);
   EXPECT_FALSE(prerender_contents->prerendering_has_started());
 }
 
-// Checks that the "PrerenderSilence experiment does not disable offline
+// Checks that the "PrerenderSilence" experiment does not disable offline
 // prerendering.
 TEST_F(PrerenderTest, PrerenderSilenceAllowsOffline) {
   // Set the time to 30 seconds before the experiment expires.
   ASSERT_TRUE(base::FieldTrialList::CreateFieldTrial(
       "PrerenderSilence", "ExperimentYes_expires_2016-12-20T00:01:00Z"));
   ASSERT_TRUE(OverridePrerenderManagerTime("2016-12-20T00:00:30Z",
                                            prerender_manager()));
   GURL url("http://www.google.com/");
   DummyPrerenderContents* prerender_contents =
       prerender_manager()->CreateNextPrerenderContents(
           url, ORIGIN_OFFLINE, FINAL_STATUS_MANAGER_SHUTDOWN);
   std::unique_ptr<PrerenderHandle> prerender_handle =
       prerender_manager()->AddPrerenderForOffline(url, nullptr, kSize);
   EXPECT_TRUE(prerender_handle);
   EXPECT_TRUE(prerender_handle->IsPrerendering());
   EXPECT_TRUE(prerender_contents->prerendering_has_started());
   EXPECT_EQ(prerender_contents, prerender_handle->contents());
   EXPECT_EQ(ORIGIN_OFFLINE, prerender_handle->contents()->origin());
 }
 
-// Checks that the "PrerenderSilence" experiment disables prerendering.
+// Checks that the "PrerenderSilence" experiment does not disable
+// forced-cellular prerendering.
+TEST_F(PrerenderTest, PrerenderSilenceAllowsForcedCellular) {
+  // Set the time to 30 seconds before the experiment expires.
+  ASSERT_TRUE(base::FieldTrialList::CreateFieldTrial(
+      "PrerenderSilence", "ExperimentYes_expires_2016-12-20T00:01:00Z"));
+  ASSERT_TRUE(OverridePrerenderManagerTime("2016-12-20T00:00:30Z",
+                                           prerender_manager()));
+  GURL url("http://www.google.com/");
+  DummyPrerenderContents* prerender_contents =
+      prerender_manager()->CreateNextPrerenderContents(
+          url, ORIGIN_EXTERNAL_REQUEST_FORCED_CELLULAR,
+          FINAL_STATUS_MANAGER_SHUTDOWN);
+  std::unique_ptr<PrerenderHandle> prerender_handle =
+      prerender_manager()->AddPrerenderOnCellularFromExternalRequest(
+          url, content::Referrer(), nullptr, gfx::Rect(kSize));
+  EXPECT_TRUE(prerender_handle);
+  EXPECT_TRUE(prerender_handle->IsPrerendering());
+  EXPECT_TRUE(prerender_contents->prerendering_has_started());
+  EXPECT_EQ(prerender_contents, prerender_handle->contents());
+  EXPECT_EQ(ORIGIN_EXTERNAL_REQUEST_FORCED_CELLULAR,
+            prerender_handle->contents()->origin());
+}
+
+// Checks that the "PrerenderSilence" experiment disables prerendering and
+// re-enables it after expiration.
 TEST_F(PrerenderTest, PrerenderSilenceDisallowsNonOffline) {
   ASSERT_TRUE(base::FieldTrialList::CreateFieldTrial(
       "PrerenderSilence", "ExperimentYes_expires_2016-12-20T00:02:00Z"));
-  ASSERT_TRUE(OverridePrerenderManagerTime("2016-12-20T00:01:00Z",
-                                           prerender_manager()));
-  const Origin origins[] = {
+  Origin prerender_silence_origins[] = {
       ORIGIN_GWS_PRERENDER,
       ORIGIN_OMNIBOX,
       ORIGIN_NONE,
       ORIGIN_LINK_REL_PRERENDER_SAMEDOMAIN,
       ORIGIN_LINK_REL_PRERENDER_CROSSDOMAIN,
       ORIGIN_EXTERNAL_REQUEST,
       ORIGIN_INSTANT,
       ORIGIN_LINK_REL_NEXT,
-      ORIGIN_EXTERNAL_REQUEST_FORCED_CELLULAR,
   };
-  for (const Origin& origin : origins) {
+
+  // Prerender before expiration of the experiment.
+  ASSERT_TRUE(OverridePrerenderManagerTime("2016-12-20T00:01:00Z",
+                                           prerender_manager()));
+  for (const Origin& origin : prerender_silence_origins) {
     EXPECT_TRUE(
         prerender_manager()->IsPrerenderSilenceExperimentForTesting(origin));
   }
+
+  // Prerender after expiration of the experiment.
+  ASSERT_TRUE(OverridePrerenderManagerTime("2016-12-20T00:03:00Z",
+                                           prerender_manager()));
+  for (const Origin& origin : prerender_silence_origins) {
+    EXPECT_FALSE(
+        prerender_manager()->IsPrerenderSilenceExperimentForTesting(origin));
+  }
 }
 
 // Checks that prerendering is enabled after expiration of the
 // "PrerenderSilence" experiment.
 TEST_F(PrerenderTest, PrerenderSilenceAllowsAfterExpiration) {
   ASSERT_TRUE(base::FieldTrialList::CreateFieldTrial(
       "PrerenderSilence", "ExperimentYes_expires_2016-12-20T00:02:00Z"));
   ASSERT_TRUE(OverridePrerenderManagerTime("2016-12-20T00:01:60Z",
                                            prerender_manager()));
   GURL url("http://www.google.com/");
@@ skipping 596 matching lines @@
   prerender_contents->SetPrerenderMode(PREFETCH_ONLY);
   EXPECT_TRUE(prerender_contents->IsValidHttpMethod("GET"));
   EXPECT_TRUE(prerender_contents->IsValidHttpMethod("HEAD"));
   EXPECT_FALSE(prerender_contents->IsValidHttpMethod("OPTIONS"));
   EXPECT_FALSE(prerender_contents->IsValidHttpMethod("POST"));
   EXPECT_FALSE(prerender_contents->IsValidHttpMethod("TRACE"));
   EXPECT_FALSE(prerender_contents->IsValidHttpMethod("WHATEVER"));
 }
 
 }  // namespace prerender