components/dom_distiller/core/fake_db.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 "components/dom_distiller/core/fake_db.h"
   6
   7 #include "base/bind.h"
   8
   9 namespace dom_distiller {
  10 namespace test {
  11
  12 FakeDB::FakeDB(EntryMap* db) : db_(db) {}
  13
  14 FakeDB::~FakeDB() {}
  15
  16 void FakeDB::Init(const base::FilePath& database_dir,
  17                   DomDistillerDatabaseInterface::InitCallback callback) {
  18  dir_ = database_dir;
  19   init_callback_ = callback;
  20 }
  21
  22 void FakeDB::UpdateEntries(
  23     scoped_ptr<EntryVector> entries_to_save,
  24     scoped_ptr<EntryVector> entries_to_remove,
  25     DomDistillerDatabaseInterface::UpdateCallback callback) {
  26   for (EntryVector::iterator it = entries_to_save->begin();
  27        it != entries_to_save->end();
  28        ++it) {
  29     (*db_)[it->entry_id()] = *it;
  30   }
  31   for (EntryVector::iterator it = entries_to_remove->begin();
  32        it != entries_to_remove->end();
  33        ++it) {
  34     (*db_).erase(it->entry_id());
  35   }
  36   update_callback_ = callback;
  37 }
  38
  39 void FakeDB::LoadEntries(DomDistillerDatabaseInterface::LoadCallback callback) {
  40   scoped_ptr<EntryVector> entries(new EntryVector());
  41   for (EntryMap::iterator it = db_->begin(); it != db_->end(); ++it) {
  42     entries->push_back(it->second);
  43   }
  44   load_callback_ =
  45       base::Bind(RunLoadCallback, callback, base::Passed(&entries));
  46 }
  47
  48 base::FilePath& FakeDB::GetDirectory() { return dir_; }
  49
  50 void FakeDB::InitCallback(bool success) {
  51   init_callback_.Run(success);
  52   init_callback_.Reset();
  53 }
  54
  55 void FakeDB::LoadCallback(bool success) {
  56   load_callback_.Run(success);
  57   load_callback_.Reset();
  58 }
  59
  60 void FakeDB::UpdateCallback(bool success) {
  61   update_callback_.Run(success);
  62   update_callback_.Reset();
  63 }
  64
  65 // static
  66 void FakeDB::RunLoadCallback(
  67     DomDistillerDatabaseInterface::LoadCallback callback,
  68     scoped_ptr<EntryVector> entries,
  69     bool success) {
  70   callback.Run(success, entries.Pass());
  71 }
  72
  73 // static
  74 base::FilePath FakeDB::DirectoryForTestDB() {
  75   return base::FilePath(FILE_PATH_LITERAL("/fake/path"));
  76 }
  77
  78 }  // namespace test
  79 }  // namespace dom_distiller