chrome/browser/browsing_data/mock_browsing_data_indexed_db_helper.cc

   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_indexed_db_helper.h"
   6
   7 #include "base/callback.h"
   8 #include "base/logging.h"
   9 #include "base/stl_util.h"
  10 #include "chrome/browser/profiles/profile.h"
  11 #include "content/public/browser/browser_context.h"
  12 #include "content/public/browser/storage_partition.h"
  13 #include "testing/gtest/include/gtest/gtest.h"
  14
  15 MockBrowsingDataIndexedDBHelper::MockBrowsingDataIndexedDBHelper(
  16     Profile* profile)
  17     : BrowsingDataIndexedDBHelper(
  18         content::BrowserContext::GetDefaultStoragePartition(profile)->
  19             GetIndexedDBContext()) {
  20 }
  21
  22 MockBrowsingDataIndexedDBHelper::~MockBrowsingDataIndexedDBHelper() {
  23 }
  24
  25 void MockBrowsingDataIndexedDBHelper::StartFetching(
  26     const FetchCallback& callback) {
  27   ASSERT_FALSE(callback.is_null());
  28   ASSERT_TRUE(callback_.is_null());
  29   callback_ = callback;
  30 }
  31
  32 void MockBrowsingDataIndexedDBHelper::DeleteIndexedDB(
  33     const GURL& origin) {
  34   ASSERT_FALSE(callback_.is_null());
  35   ASSERT_TRUE(ContainsKey(origins_, origin));
  36   origins_[origin] = false;
  37 }
  38
  39 void MockBrowsingDataIndexedDBHelper::AddIndexedDBSamples() {
  40   const GURL kOrigin1("http://idbhost1:1/");
  41   const GURL kOrigin2("http://idbhost2:2/");
  42   content::IndexedDBInfo info1(kOrigin1, 1, base::Time(), 0);
  43   response_.push_back(info1);
  44   origins_[kOrigin1] = true;
  45   content::IndexedDBInfo info2(kOrigin2, 2, base::Time(), 0);
  46   response_.push_back(info2);
  47   origins_[kOrigin2] = true;
  48 }
  49
  50 void MockBrowsingDataIndexedDBHelper::Notify() {
  51   callback_.Run(response_);
  52 }
  53
  54 void MockBrowsingDataIndexedDBHelper::Reset() {
  55   for (auto& pair : origins_)
  56     pair.second = true;
  57 }
  58
  59 bool MockBrowsingDataIndexedDBHelper::AllDeleted() {
  60   for (const auto& pair : origins_) {
  61     if (pair.second)
  62       return false;
  63   }
  64   return true;
  65 }