Home | History | Annotate | Download | only in browsing_data 
      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 "chrome/browser/browsing_data/mock_browsing_data_cookie_helper.h"
      6 
      7 #include "base/logging.h"
      8 #include "net/cookies/canonical_cookie.h"
      9 #include "net/cookies/parsed_cookie.h"
     10 #include "testing/gtest/include/gtest/gtest.h"
     11 
     12 MockBrowsingDataCookieHelper::MockBrowsingDataCookieHelper(
     13     net::URLRequestContextGetter* request_context_getter)
     14     : BrowsingDataCookieHelper(request_context_getter) {
     15 }
     16 
     17 MockBrowsingDataCookieHelper::~MockBrowsingDataCookieHelper() {
     18 }
     19 
     20 void MockBrowsingDataCookieHelper::StartFetching(
     21     const net::CookieMonster::GetCookieListCallback &callback) {
     22   ASSERT_FALSE(callback.is_null());
     23   ASSERT_TRUE(callback_.is_null());
     24   callback_ = callback;
     25 }
     26 
     27 void MockBrowsingDataCookieHelper::DeleteCookie(
     28     const net::CanonicalCookie& cookie) {
     29   ASSERT_FALSE(callback_.is_null());
     30   std::string key = cookie.Name() + "=" + cookie.Value();
     31   ASSERT_TRUE(cookies_.find(key) != cookies_.end());
     32   cookies_[key] = false;
     33 }
     34 
     35 void MockBrowsingDataCookieHelper::AddCookieSamples(
     36     const GURL& url, const std::string& cookie_line) {
     37   typedef net::CookieList::const_iterator cookie_iterator;
     38   net::ParsedCookie pc(cookie_line);
     39   scoped_ptr<net::CanonicalCookie> cc(new net::CanonicalCookie(url, pc));
     40 
     41   if (cc.get()) {
     42     for (cookie_iterator cookie = cookie_list_.begin();
     43         cookie != cookie_list_.end(); ++cookie) {
     44       if (cookie->Name() == cc->Name() &&
     45           cookie->Domain() == cc->Domain() &&
     46           cookie->Path() == cc->Path()) {
     47         return;
     48       }
     49     }
     50     cookie_list_.push_back(*cc);
     51     cookies_[cookie_line] = true;
     52   }
     53 }
     54 
     55 void MockBrowsingDataCookieHelper::Notify() {
     56   if (!callback_.is_null())
     57     callback_.Run(cookie_list_);
     58 }
     59 
     60 void MockBrowsingDataCookieHelper::Reset() {
     61   for (std::map<const std::string, bool>::iterator i = cookies_.begin();
     62        i != cookies_.end(); ++i)
     63     i->second = true;
     64 }
     65 
     66 bool MockBrowsingDataCookieHelper::AllDeleted() {
     67   for (std::map<const std::string, bool>::const_iterator i = cookies_.begin();
     68        i != cookies_.end(); ++i)
     69     if (i->second)
     70       return false;
     71   return true;
     72 }
     73