| // Copyright 2013 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 "base/command_line.h" |
| #include "base/strings/utf_string_conversions.h" |
| #include "chrome/browser/ui/browser.h" |
| #include "chrome/browser/ui/tabs/tab_strip_model.h" |
| #include "chrome/common/chrome_switches.h" |
| #include "chrome/test/base/in_process_browser_test.h" |
| #include "chrome/test/base/ui_test_utils.h" |
| #include "content/public/test/browser_test_utils.h" |
| |
| namespace { |
| |
| const char kPrefetchPage[] = "files/prerender/simple_prefetch.html"; |
| |
| class PrefetchBrowserTestBase : public InProcessBrowserTest { |
| public: |
| explicit PrefetchBrowserTestBase(bool do_prefetching) |
| : do_prefetching_(do_prefetching) {} |
| |
| virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { |
| if (do_prefetching_) { |
| command_line->AppendSwitchASCII( |
| switches::kForceFieldTrials, |
| "Prefetch/ExperimentYes/"); |
| } else { |
| command_line->AppendSwitchASCII( |
| switches::kForceFieldTrials, |
| "Prefetch/ExperimentNo/"); |
| } |
| } |
| |
| private: |
| bool do_prefetching_; |
| }; |
| |
| class PrefetchBrowserTest : public PrefetchBrowserTestBase { |
| public: |
| PrefetchBrowserTest() |
| : PrefetchBrowserTestBase(true) {} |
| }; |
| |
| class PrefetchBrowserTestNoPrefetching : public PrefetchBrowserTestBase { |
| public: |
| PrefetchBrowserTestNoPrefetching() |
| : PrefetchBrowserTestBase(false) {} |
| }; |
| |
| IN_PROC_BROWSER_TEST_F(PrefetchBrowserTest, PrefetchOn) { |
| ASSERT_TRUE(test_server()->Start()); |
| GURL url = test_server()->GetURL(kPrefetchPage); |
| |
| const base::string16 expected_title = ASCIIToUTF16("link onload"); |
| content::TitleWatcher title_watcher( |
| browser()->tab_strip_model()->GetActiveWebContents(), |
| expected_title); |
| ui_test_utils::NavigateToURL(browser(), url); |
| |
| EXPECT_EQ(expected_title, title_watcher.WaitAndGetTitle()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(PrefetchBrowserTestNoPrefetching, PrefetchOff) { |
| ASSERT_TRUE(test_server()->Start()); |
| GURL url = test_server()->GetURL(kPrefetchPage); |
| |
| const base::string16 expected_title = ASCIIToUTF16("link onerror"); |
| content::TitleWatcher title_watcher( |
| browser()->tab_strip_model()->GetActiveWebContents(), |
| expected_title); |
| ui_test_utils::NavigateToURL(browser(), url); |
| |
| EXPECT_EQ(expected_title, title_watcher.WaitAndGetTitle()); |
| } |
| |
| } // namespace |
| |