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