src/chrome/browser/prefetch/prefetch_browsertest.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/command_line.h"
   6 #include "base/prefs/pref_service.h"
   7 #include "base/strings/utf_string_conversions.h"
   8 #include "chrome/browser/profiles/profile.h"
   9 #include "chrome/browser/ui/browser.h"
  10 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  11 #include "chrome/common/chrome_switches.h"
  12 #include "chrome/common/pref_names.h"
  13 #include "chrome/test/base/in_process_browser_test.h"
  14 #include "chrome/test/base/ui_test_utils.h"
  15 #include "content/public/test/browser_test_utils.h"
  16
  17 namespace {
  18
  19 const char kPrefetchPage[] = "files/prerender/simple_prefetch.html";
  20
  21 class PrefetchBrowserTestBase : public InProcessBrowserTest {
  22  public:
  23   explicit PrefetchBrowserTestBase(bool do_predictive_networking,
  24                                    bool do_prefetch_field_trial)
  25       : do_predictive_networking_(do_predictive_networking),
  26         do_prefetch_field_trial_(do_prefetch_field_trial) {}
  27
  28   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
  29     if (do_prefetch_field_trial_) {
  30       command_line->AppendSwitchASCII(
  31           switches::kForceFieldTrials,
  32           "Prefetch/ExperimentYes/");
  33     } else {
  34       command_line->AppendSwitchASCII(
  35           switches::kForceFieldTrials,
  36           "Prefetch/ExperimentNo/");
  37     }
  38   }
  39
  40   virtual void SetUpOnMainThread() OVERRIDE {
  41     browser()->profile()->GetPrefs()->SetBoolean(
  42         prefs::kNetworkPredictionEnabled, do_predictive_networking_);
  43   }
  44
  45   bool RunPrefetchExperiment(bool expect_success, Browser* browser) {
  46     CHECK(test_server()->Start());
  47     GURL url = test_server()->GetURL(kPrefetchPage);
  48
  49     const base::string16 expected_title =
  50         expect_success ? base::ASCIIToUTF16("link onload")
  51                        : base::ASCIIToUTF16("link onerror");
  52     content::TitleWatcher title_watcher(
  53         browser->tab_strip_model()->GetActiveWebContents(), expected_title);
  54     ui_test_utils::NavigateToURL(browser, url);
  55     return expected_title == title_watcher.WaitAndGetTitle();
  56   }
  57
  58  private:
  59   bool do_predictive_networking_;
  60   bool do_prefetch_field_trial_;
  61 };
  62
  63 class PrefetchBrowserTestPredictionOnExpOn : public PrefetchBrowserTestBase {
  64  public:
  65   PrefetchBrowserTestPredictionOnExpOn()
  66       : PrefetchBrowserTestBase(true, true) {}
  67 };
  68
  69 class PrefetchBrowserTestPredictionOnExpOff : public PrefetchBrowserTestBase {
  70  public:
  71   PrefetchBrowserTestPredictionOnExpOff()
  72       : PrefetchBrowserTestBase(true, false) {}
  73 };
  74
  75 class PrefetchBrowserTestPredictionOffExpOn : public PrefetchBrowserTestBase {
  76  public:
  77   PrefetchBrowserTestPredictionOffExpOn()
  78       : PrefetchBrowserTestBase(false, true) {}
  79 };
  80
  81 class PrefetchBrowserTestPredictionOffExpOff : public PrefetchBrowserTestBase {
  82  public:
  83   PrefetchBrowserTestPredictionOffExpOff()
  84       : PrefetchBrowserTestBase(false, false) {}
  85 };
  86
  87 // Privacy option is on, experiment is on.  Prefetch should succeed.
  88 IN_PROC_BROWSER_TEST_F(PrefetchBrowserTestPredictionOnExpOn, PredOnExpOn) {
  89   EXPECT_TRUE(RunPrefetchExperiment(true, browser()));
  90 }
  91
  92 // Privacy option is on, experiment is off.  Prefetch should be dropped.
  93 IN_PROC_BROWSER_TEST_F(PrefetchBrowserTestPredictionOnExpOff, PredOnExpOff) {
  94   EXPECT_TRUE(RunPrefetchExperiment(false, browser()));
  95 }
  96
  97 // Privacy option is off, experiment is on.  Prefetch should be dropped.
  98 IN_PROC_BROWSER_TEST_F(PrefetchBrowserTestPredictionOffExpOn, PredOffExpOn) {
  99   EXPECT_TRUE(RunPrefetchExperiment(false, browser()));
 100 }
 101
 102 // Privacy option is off, experiment is off.  Prefetch should be dropped.
 103 IN_PROC_BROWSER_TEST_F(PrefetchBrowserTestPredictionOffExpOff, PredOffExpOff) {
 104   EXPECT_TRUE(RunPrefetchExperiment(false, browser()));
 105 }
 106
 107 // Bug 339909: When in incognito mode the browser crashed due to an
 108 // uninitialized preference member. Verify that it no longer does.
 109 IN_PROC_BROWSER_TEST_F(PrefetchBrowserTestPredictionOnExpOn, IncognitoTest) {
 110   Profile* incognito_profile = browser()->profile()->GetOffTheRecordProfile();
 111   Browser* incognito_browser = new Browser(
 112       Browser::CreateParams(incognito_profile, browser()->host_desktop_type()));
 113
 114   // Navigate just to have a tab in this window, otherwise there is no
 115   // WebContents for the incognito browser.
 116   ui_test_utils::OpenURLOffTheRecord(browser()->profile(), GURL("about:blank"));
 117
 118   EXPECT_TRUE(RunPrefetchExperiment(true, incognito_browser));
 119 }
 120
 121 }  // namespace
 122