Test for referrer policy

BUG=105028
TEST=ReferrerPolicyTest.*


Review URL: http://codereview.chromium.org/8823008

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@113307 0039d316-1c4b-4281-b951-d872f2087c98

chrome/browser/referrer_policy_browsertest.cc

+// Copyright (c) 2011 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/bind.h"
+#include "base/string_number_conversions.h"
+#include "base/utf_string_conversions.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
+#include "chrome/test/base/in_process_browser_test.h"
+#include "chrome/test/base/ui_test_utils.h"
+#include "net/test/test_server.h"
+
+namespace {
+
+const FilePath::CharType kDocRoot[] =
+    FILE_PATH_LITERAL("chrome/test/data/referrer_policy");
+
+}  // namespace
+
+class ReferrerPolicyTest : public InProcessBrowserTest {
+ public:
+  // Navigates to the given URL and waits for |num_navigations| to occur, and
+  // the title to change to |expected_title|.
+  void NavigateToURLAndWaitForTitle(const GURL& url,
+                                    const std::string& expected_title,
+                                    int num_navigations) {
+    ui_test_utils::TitleWatcher title_watcher(
+        browser()->GetSelectedTabContents(),
+        ASCIIToUTF16(expected_title));
+
+    ui_test_utils::NavigateToURLBlockUntilNavigationsComplete(
+        browser(), url, num_navigations);
+
+    EXPECT_EQ(title_watcher.WaitAndGetTitle(), ASCIIToUTF16(expected_title));
+  }
+
+  enum ExpectedReferrer {
+    ExpectedReferrerIsEmpty,
+    ExpectedReferrerIsFull,
+    ExpectedReferrerIsOrigin
+  };
+
+  // Navigates from a page with a given |referrer_policy| (optionally on https)
+  // to a page on http that reports the passed referrer and checks that it
+  // matches the |expected_referrer|.
+  void RunReferrerTest(const std::string referrer_policy,
+                       bool start_on_https,
+                       ExpectedReferrer expected_referrer) {
+    net::TestServer test_server(net::TestServer::TYPE_HTTP, FilePath(kDocRoot));
+    ASSERT_TRUE(test_server.Start());
+    net::TestServer ssl_test_server(net::TestServer::TYPE_HTTPS,
+                                    FilePath(kDocRoot));
+    ASSERT_TRUE(ssl_test_server.Start());
+
+    GURL start_url = test_server.GetURL(
+        std::string("files/referrer-policy-start.html?") + referrer_policy +
+        "&port=" + base::IntToString(test_server.host_port_pair().port()));
+    if (start_on_https) {
+      start_url = ssl_test_server.GetURL(
+          std::string("files/referrer-policy-start.html?") + referrer_policy +
+          "&port=" + base::IntToString(test_server.host_port_pair().port()));
+    }
+
+    std::string referrer;
+    switch (expected_referrer) {
+      case ExpectedReferrerIsEmpty:
+        referrer = "Referrer is empty";
+        break;
+      case ExpectedReferrerIsFull:
+        referrer = "Referrer is " + start_url.spec();
+        break;
+      case ExpectedReferrerIsOrigin:
+        referrer = "Referrer is " + start_url.GetWithEmptyPath().spec();
+        break;
+    }
+
+    NavigateToURLAndWaitForTitle(start_url, referrer, 2);
+  }
+};
+
+IN_PROC_BROWSER_TEST_F(ReferrerPolicyTest, Default) {
+  RunReferrerTest("default", false, ExpectedReferrerIsFull);
+}
+
+IN_PROC_BROWSER_TEST_F(ReferrerPolicyTest, Always) {
+  RunReferrerTest("always", false, ExpectedReferrerIsFull);
+}
+
+IN_PROC_BROWSER_TEST_F(ReferrerPolicyTest, Origin) {
+  RunReferrerTest("origin", false, ExpectedReferrerIsOrigin);
+}
+
+IN_PROC_BROWSER_TEST_F(ReferrerPolicyTest, Never) {
+  RunReferrerTest("never", false, ExpectedReferrerIsEmpty);
+}
+
+IN_PROC_BROWSER_TEST_F(ReferrerPolicyTest, HttpsDefault) {
+  RunReferrerTest("default", true, ExpectedReferrerIsEmpty);
+}
+
+IN_PROC_BROWSER_TEST_F(ReferrerPolicyTest, HttpsAlways) {
+  RunReferrerTest("always", true, ExpectedReferrerIsFull);
+}
+
+IN_PROC_BROWSER_TEST_F(ReferrerPolicyTest, HttpsOrigin) {
+  RunReferrerTest("origin", true, ExpectedReferrerIsOrigin);
+}
+
+IN_PROC_BROWSER_TEST_F(ReferrerPolicyTest, HttpsNever) {
+  RunReferrerTest("never", true, ExpectedReferrerIsEmpty);
+}
+
+IN_PROC_BROWSER_TEST_F(ReferrerPolicyTest, Redirect) {
+  net::TestServer test_server(net::TestServer::TYPE_HTTP, FilePath(kDocRoot));
+  ASSERT_TRUE(test_server.Start());
+  net::TestServer ssl_test_server(net::TestServer::TYPE_HTTPS,
+                                  FilePath(kDocRoot));
+  ASSERT_TRUE(ssl_test_server.Start());
+  GURL start_url = test_server.GetURL(
+      std::string("files/referrer-policy-redirect.html?") +
+      "ssl_port=" + base::IntToString(ssl_test_server.host_port_pair().port()) +
+      "&port=" + base::IntToString(test_server.host_port_pair().port()));
+  std::string referrer = "Referrer is " + start_url.GetWithEmptyPath().spec();
+  NavigateToURLAndWaitForTitle(start_url, referrer, 2);
+}

chrome/chrome_tests.gypi

@@ -2604,6 +2604,7 @@
         'browser/printing/print_preview_tab_controller_browsertest.cc',
         'browser/profiles/profile_manager_browsertest.cc',
         'browser/rlz/rlz_extension_apitest.cc',
+        'browser/referrer_policy_browsertest.cc',
         'browser/renderer_host/render_process_host_chrome_browsertest.cc',
         'browser/renderer_host/web_cache_manager_browsertest.cc',
         'browser/safe_browsing/safe_browsing_blocking_page_test.cc',

chrome/test/data/referrer_policy/referrer-policy-log.html

+<script>
+if (document.referrer == "")
+    document.title = "Referrer is empty";
+else
+    document.title = "Referrer is " + document.referrer;
+</script>

chrome/test/data/referrer_policy/referrer-policy-redirect.html

+<html>
+<head>
+<meta name="referrer" content="origin" />
+<script>
+function runTest() {
+    var matches =
+        RegExp(/\?ssl_port=(.*)&port=(.*)/).exec(document.location.search);
+    document.location =
+        "https://127.0.0.1:" + matches[1] + "/server-redirect?" +
+        "http://127.0.0.1:" + matches[2] + "/files/referrer-policy-log.html";
+}
+</script>
+</head>
+<body onload="runTest()">
+</body>
+</html>

chrome/test/data/referrer_policy/referrer-policy-start.html

+<html>
+<head>
+<script>
+function runTest() {
+    var matches = RegExp(/\?(.*)&port=(.*)/).exec(document.location.search);
+    var meta = document.createElement("meta");
+    meta.name = "referrer";
+    meta.content = matches[1];
+    document.head.appendChild(meta);
+    document.location =
+        "http://127.0.0.1:" + matches[2] + "/files/referrer-policy-log.html";
+}
+</script>
+</head>
+<body onload="runTest()">
+</body>
+</html>