src/chrome/browser/profile_resetter/profile_resetter_browsertest.cc

   1 // Copyright 2013 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 "chrome/browser/profile_resetter/profile_resetter.h"
   6
   7 #include "base/bind.h"
   8 #include "chrome/browser/profile_resetter/profile_resetter_test_base.h"
   9 #include "chrome/browser/profiles/profile.h"
  10 #include "chrome/browser/ui/browser.h"
  11 #include "chrome/test/base/in_process_browser_test.h"
  12 #include "content/public/test/test_utils.h"
  13 #include "net/cookies/cookie_store.h"
  14 #include "net/url_request/url_request_context.h"
  15 #include "net/url_request/url_request_context_getter.h"
  16
  17 namespace {
  18
  19 const char kCookieDefinition[] = "A=1";
  20 const char kCookieHostname[] = "http://host1";
  21
  22 using content::BrowserThread;
  23
  24 // RemoveCookieTester provides the user with the ability to set and get a
  25 // cookie for given profile.
  26 class RemoveCookieTester {
  27  public:
  28   explicit RemoveCookieTester(Profile* profile);
  29   ~RemoveCookieTester();
  30
  31   std::string GetCookie(const std::string& host);
  32   void AddCookie(const std::string& host, const std::string& definition);
  33
  34  private:
  35   void GetCookieOnIOThread(net::URLRequestContextGetter* context_getter,
  36                            const std::string& host);
  37   void SetCookieOnIOThread(net::URLRequestContextGetter* context_getter,
  38                            const std::string& host,
  39                            const std::string& definition);
  40   void GetCookieCallback(const std::string& cookies);
  41   void SetCookieCallback(bool result);
  42
  43   void BlockUntilNotified();
  44   void Notify();
  45
  46   std::string last_cookies_;
  47   bool waiting_callback_;
  48   Profile* profile_;
  49   scoped_refptr<content::MessageLoopRunner> runner_;
  50
  51   DISALLOW_COPY_AND_ASSIGN(RemoveCookieTester);
  52 };
  53
  54 RemoveCookieTester::RemoveCookieTester(Profile* profile)
  55     : waiting_callback_(false),
  56       profile_(profile) {
  57 }
  58
  59 RemoveCookieTester::~RemoveCookieTester() {}
  60
  61 // Returns true, if the given cookie exists in the cookie store.
  62 std::string RemoveCookieTester::GetCookie(const std::string& host) {
  63   last_cookies_.clear();
  64   DCHECK(!waiting_callback_);
  65   waiting_callback_ = true;
  66   BrowserThread::PostTask(
  67       BrowserThread::IO, FROM_HERE,
  68       base::Bind(&RemoveCookieTester::GetCookieOnIOThread,
  69                  base::Unretained(this),
  70                  base::Unretained(profile_->GetRequestContext()),
  71                  host));
  72   BlockUntilNotified();
  73   return last_cookies_;
  74 }
  75
  76 void RemoveCookieTester::AddCookie(const std::string& host,
  77                                    const std::string& definition) {
  78   DCHECK(!waiting_callback_);
  79   waiting_callback_ = true;
  80   BrowserThread::PostTask(
  81       BrowserThread::IO, FROM_HERE,
  82       base::Bind(&RemoveCookieTester::SetCookieOnIOThread,
  83                  base::Unretained(this),
  84                  base::Unretained(profile_->GetRequestContext()),
  85                  host,
  86                  definition));
  87   BlockUntilNotified();
  88 }
  89
  90 void RemoveCookieTester::GetCookieOnIOThread(
  91     net::URLRequestContextGetter* context_getter,
  92     const std::string& host) {
  93   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
  94   net::CookieStore* cookie_store = context_getter->
  95       GetURLRequestContext()->cookie_store();
  96   cookie_store->GetCookiesWithOptionsAsync(
  97       GURL(host), net::CookieOptions(),
  98       base::Bind(&RemoveCookieTester::GetCookieCallback,
  99                  base::Unretained(this)));
 100 }
 101
 102 void RemoveCookieTester::SetCookieOnIOThread(
 103     net::URLRequestContextGetter* context_getter,
 104     const std::string& host,
 105     const std::string& definition) {
 106   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
 107   net::CookieStore* cookie_store = context_getter->
 108       GetURLRequestContext()->cookie_store();
 109   cookie_store->SetCookieWithOptionsAsync(
 110       GURL(host), definition, net::CookieOptions(),
 111       base::Bind(&RemoveCookieTester::SetCookieCallback,
 112                  base::Unretained(this)));
 113 }
 114
 115 void RemoveCookieTester::GetCookieCallback(const std::string& cookies) {
 116   if (!BrowserThread::CurrentlyOn(BrowserThread::UI)) {
 117     BrowserThread::PostTask(
 118         BrowserThread::UI, FROM_HERE,
 119         base::Bind(&RemoveCookieTester::GetCookieCallback,
 120                    base::Unretained(this), cookies));
 121     return;
 122   }
 123   last_cookies_ = cookies;
 124   Notify();
 125 }
 126
 127 void RemoveCookieTester::SetCookieCallback(bool result) {
 128   if (!BrowserThread::CurrentlyOn(BrowserThread::UI)) {
 129     BrowserThread::PostTask(
 130         BrowserThread::UI, FROM_HERE,
 131         base::Bind(&RemoveCookieTester::SetCookieCallback,
 132                    base::Unretained(this), result));
 133     return;
 134   }
 135   ASSERT_TRUE(result);
 136   Notify();
 137 }
 138
 139 void RemoveCookieTester::BlockUntilNotified() {
 140   DCHECK(!runner_.get());
 141   if (waiting_callback_) {
 142     runner_ = new content::MessageLoopRunner;
 143     runner_->Run();
 144     runner_ = NULL;
 145   }
 146 }
 147
 148 void RemoveCookieTester::Notify() {
 149   DCHECK(waiting_callback_);
 150   waiting_callback_ = false;
 151   if (runner_.get())
 152     runner_->Quit();
 153 }
 154
 155 class ProfileResetTest : public InProcessBrowserTest,
 156                          public ProfileResetterTestBase {
 157  protected:
 158   virtual void SetUpOnMainThread() OVERRIDE {
 159     resetter_.reset(new ProfileResetter(browser()->profile()));
 160   }
 161 };
 162
 163
 164 IN_PROC_BROWSER_TEST_F(ProfileResetTest, ResetCookiesAndSiteData) {
 165   RemoveCookieTester tester(browser()->profile());
 166   tester.AddCookie(kCookieHostname, kCookieDefinition);
 167   ASSERT_EQ(kCookieDefinition, tester.GetCookie(kCookieHostname));
 168
 169   ResetAndWait(ProfileResetter::COOKIES_AND_SITE_DATA);
 170
 171   EXPECT_EQ("", tester.GetCookie(kCookieHostname));
 172 }
 173
 174 }  // namespace