diff options
| -rw-r--r-- | chrome/browser/net/cookie_policy_browsertest.cc | 87 | ||||
| -rwxr-xr-x | chrome/chrome.gyp | 1 |
2 files changed, 88 insertions, 0 deletions
diff --git a/chrome/browser/net/cookie_policy_browsertest.cc b/chrome/browser/net/cookie_policy_browsertest.cc new file mode 100644 index 0000000..f218575 --- /dev/null +++ b/chrome/browser/net/cookie_policy_browsertest.cc @@ -0,0 +1,87 @@ +// Copyright (c) 2009 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 "chrome/browser/browser.h" +#include "chrome/browser/net/url_request_context_getter.h" +#include "chrome/browser/profile.h" +#include "chrome/common/pref_names.h" +#include "chrome/common/pref_service.h" +#include "chrome/test/in_process_browser_test.h" +#include "chrome/test/ui_test_utils.h" + +class CookiePolicyBrowserTest : public InProcessBrowserTest { + public: + CookiePolicyBrowserTest() {} + + private: + DISALLOW_COPY_AND_ASSIGN(CookiePolicyBrowserTest); +}; + +// Visits a page that sets a first-party cookie. +IN_PROC_BROWSER_TEST_F(CookiePolicyBrowserTest, AllowFirstPartyCookies) { + HTTPTestServer* server = StartHTTPServer(); + ASSERT_TRUE(server != NULL); + + PrefService* prefs = browser()->profile()->GetPrefs(); + prefs->SetInteger(prefs::kCookieBehavior, + net::CookiePolicy::BLOCK_THIRD_PARTY_COOKIES); + net::CookiePolicy::Type policy_type = net::CookiePolicy::FromInt( + prefs->GetInteger(prefs::kCookieBehavior)); + ASSERT_EQ(net::CookiePolicy::BLOCK_THIRD_PARTY_COOKIES, policy_type); + + net::CookieStore* cookie_store = + browser()->profile()->GetRequestContext()->GetCookieStore(); + + GURL url = server->TestServerPage("set-cookie?cookie1"); + + std::string cookie = cookie_store->GetCookies(url); + ASSERT_EQ("", cookie); + + ui_test_utils::NavigateToURL(browser(), url); + + cookie = cookie_store->GetCookies(url); + EXPECT_EQ("cookie1", cookie); +} + + +// Visits a page that is a redirect across domain boundary to a page that sets +// a first-party cookie. +IN_PROC_BROWSER_TEST_F(CookiePolicyBrowserTest, + AllowFirstPartyCookiesRedirect) { + HTTPTestServer* server = StartHTTPServer(); + ASSERT_TRUE(server != NULL); + + PrefService* prefs = browser()->profile()->GetPrefs(); + prefs->SetInteger(prefs::kCookieBehavior, + net::CookiePolicy::BLOCK_THIRD_PARTY_COOKIES); + net::CookiePolicy::Type policy_type = net::CookiePolicy::FromInt( + prefs->GetInteger(prefs::kCookieBehavior)); + ASSERT_EQ(net::CookiePolicy::BLOCK_THIRD_PARTY_COOKIES, policy_type); + + net::CookieStore* cookie_store = + browser()->profile()->GetRequestContext()->GetCookieStore(); + + GURL url = server->TestServerPage("server-redirect?"); + + GURL redirected_url = server->TestServerPage("set-cookie?cookie2"); + // Change the host name from localhost to www.example.com so it triggers + // third-party cookie blocking if the first party for cookies URL is not + // changed when we follow a redirect. + ASSERT_EQ("localhost", redirected_url.host()); + GURL::Replacements replacements; + std::string new_host("www.example.com"); + replacements.SetHostStr(new_host); + redirected_url = redirected_url.ReplaceComponents(replacements); + + std::string cookie = cookie_store->GetCookies(redirected_url); + ASSERT_EQ("", cookie); + + host_resolver()->AddRule("www.example.com", "127.0.0.1"); + + ui_test_utils::NavigateToURL(browser(), + GURL(url.spec() + redirected_url.spec())); + + cookie = cookie_store->GetCookies(redirected_url); + EXPECT_EQ("cookie2", cookie); +} diff --git a/chrome/chrome.gyp b/chrome/chrome.gyp index 74b63eb..d47ac08 100755 --- a/chrome/chrome.gyp +++ b/chrome/chrome.gyp @@ -78,6 +78,7 @@ 'browser/extensions/page_action_apitest.cc', 'browser/extensions/stubs_apitest.cc', 'browser/gtk/bookmark_manager_browsertest.cc', + 'browser/net/cookie_policy_browsertest.cc', 'browser/net/ftp_browsertest.cc', 'browser/privacy_blacklist/blacklist_manager_browsertest.cc', 'browser/ssl/ssl_browser_tests.cc', |