1// Copyright (c) 2012 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/strings/utf_string_conversions.h" 7#include "chrome/browser/ui/browser.h" 8#include "chrome/browser/ui/tabs/tab_strip_model.h" 9#include "chrome/common/chrome_switches.h" 10#include "chrome/test/base/in_process_browser_test.h" 11#include "chrome/test/base/ui_test_utils.h" 12#include "content/public/test/browser_test_utils.h" 13 14namespace { 15 16const char kPrefetchPage[] = "files/prerender/simple_prefetch.html"; 17 18class PrefetchBrowserTestBase : public InProcessBrowserTest { 19 public: 20 explicit PrefetchBrowserTestBase(bool do_prefetching) 21 : do_prefetching_(do_prefetching) {} 22 23 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { 24 if (do_prefetching_) { 25 command_line->AppendSwitchASCII( 26 switches::kPrerenderMode, 27 switches::kPrerenderModeSwitchValuePrefetchOnly); 28 } else { 29 command_line->AppendSwitchASCII( 30 switches::kPrerenderMode, 31 switches::kPrerenderModeSwitchValueDisabled); 32 } 33 } 34 35 private: 36 bool do_prefetching_; 37}; 38 39class PrefetchBrowserTest : public PrefetchBrowserTestBase { 40 public: 41 PrefetchBrowserTest() 42 : PrefetchBrowserTestBase(true) {} 43}; 44 45class PrefetchBrowserTestNoPrefetching : public PrefetchBrowserTestBase { 46 public: 47 PrefetchBrowserTestNoPrefetching() 48 : PrefetchBrowserTestBase(false) {} 49}; 50 51IN_PROC_BROWSER_TEST_F(PrefetchBrowserTest, PrefetchOn) { 52 ASSERT_TRUE(test_server()->Start()); 53 GURL url = test_server()->GetURL(kPrefetchPage); 54 55 const string16 expected_title = ASCIIToUTF16("link onload"); 56 content::TitleWatcher title_watcher( 57 browser()->tab_strip_model()->GetActiveWebContents(), 58 expected_title); 59 ui_test_utils::NavigateToURL(browser(), url); 60 61 EXPECT_EQ(expected_title, title_watcher.WaitAndGetTitle()); 62} 63 64IN_PROC_BROWSER_TEST_F(PrefetchBrowserTestNoPrefetching, PrefetchOff) { 65 ASSERT_TRUE(test_server()->Start()); 66 GURL url = test_server()->GetURL(kPrefetchPage); 67 68 const string16 expected_title = ASCIIToUTF16("link onerror"); 69 content::TitleWatcher title_watcher( 70 browser()->tab_strip_model()->GetActiveWebContents(), 71 expected_title); 72 ui_test_utils::NavigateToURL(browser(), url); 73 74 EXPECT_EQ(expected_title, title_watcher.WaitAndGetTitle()); 75} 76 77} // namespace 78 79