Index: chrome/browser/net/cookie_policy_browsertest.cc |
=================================================================== |
--- chrome/browser/net/cookie_policy_browsertest.cc (revision 153070) |
+++ chrome/browser/net/cookie_policy_browsertest.cc (working copy) |
@@ -4,62 +4,27 @@ |
#include "base/bind.h" |
#include "base/bind_helpers.h" |
-#include "base/synchronization/waitable_event.h" |
#include "chrome/browser/content_settings/host_content_settings_map.h" |
#include "chrome/browser/prefs/pref_service.h" |
#include "chrome/browser/profiles/profile.h" |
#include "chrome/browser/ui/browser.h" |
+#include "chrome/browser/ui/browser_tabstrip.h" |
+#include "chrome/browser/ui/tab_contents/tab_contents.h" |
#include "chrome/common/pref_names.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" |
#include "net/base/mock_host_resolver.h" |
-#include "net/cookies/cookie_store.h" |
#include "net/test/test_server.h" |
-#include "net/url_request/url_request_context.h" |
-#include "net/url_request/url_request_context_getter.h" |
using content::BrowserThread; |
namespace { |
-void GetCookiesCallback(std::string* cookies_out, |
- base::WaitableEvent* event, |
- const std::string& cookies) { |
- *cookies_out = cookies; |
- event->Signal(); |
-} |
- |
-void GetCookiesOnIOThread(const GURL& url, |
- net::URLRequestContextGetter* context_getter, |
- base::WaitableEvent* event, |
- std::string* cookies) { |
- net::CookieStore* cookie_store = |
- context_getter->GetURLRequestContext()->cookie_store(); |
- cookie_store->GetCookiesWithOptionsAsync( |
- url, net::CookieOptions(), |
- base::Bind(&GetCookiesCallback, |
- base::Unretained(cookies), base::Unretained(event))); |
-} |
- |
class CookiePolicyBrowserTest : public InProcessBrowserTest { |
protected: |
CookiePolicyBrowserTest() {} |
- std::string GetCookies(const GURL& url) { |
- std::string cookies; |
- base::WaitableEvent event(true /* manual reset */, |
- false /* not initially signaled */); |
- net::URLRequestContextGetter* context_getter = |
- browser()->profile()->GetRequestContext(); |
- EXPECT_TRUE( |
- BrowserThread::PostTask( |
- BrowserThread::IO, FROM_HERE, |
- base::Bind(&GetCookiesOnIOThread, url, |
- make_scoped_refptr(context_getter), &event, &cookies))); |
- event.Wait(); |
- return cookies; |
- } |
- |
private: |
DISALLOW_COPY_AND_ASSIGN(CookiePolicyBrowserTest); |
}; |
@@ -73,12 +38,13 @@ |
GURL url(test_server()->GetURL("set-cookie?cookie1")); |
- std::string cookie = GetCookies(url); |
+ TabContents* tab = chrome::GetActiveTabContents(browser()); |
+ std::string cookie = content::GetCookies(tab->profile(), url); |
ASSERT_EQ("", cookie); |
ui_test_utils::NavigateToURL(browser(), url); |
- cookie = GetCookies(url); |
+ cookie = content::GetCookies(tab->profile(), url); |
EXPECT_EQ("cookie1", cookie); |
} |
@@ -103,7 +69,8 @@ |
replacements.SetHostStr(new_host); |
redirected_url = redirected_url.ReplaceComponents(replacements); |
- std::string cookie = GetCookies(redirected_url); |
+ TabContents* tab = chrome::GetActiveTabContents(browser()); |
+ std::string cookie = content::GetCookies(tab->profile(), redirected_url); |
ASSERT_EQ("", cookie); |
host_resolver()->AddRule("www.example.com", "127.0.0.1"); |
@@ -111,7 +78,7 @@ |
ui_test_utils::NavigateToURL(browser(), |
GURL(url.spec() + redirected_url.spec())); |
- cookie = GetCookies(redirected_url); |
+ cookie = content::GetCookies(tab->profile(), redirected_url); |
EXPECT_EQ("cookie2", cookie); |
} |