extensions/browser/value_store/test_value_store_factory.cc

// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "extensions/browser/value_store/test_value_store_factory.h"

#include "base/memory/ptr_util.h"
#include "extensions/browser/value_store/leveldb_value_store.h"
#include "extensions/browser/value_store/testing_value_store.h"
#include "extensions/browser/value_store/value_store_client_id.h"

namespace {

const char kUMAClientName[] = "Test";

}  // namespace

namespace extensions {

using SettingsNamespace = settings_namespace::Namespace;

TestValueStoreFactory::StorageHelper::StorageHelper() = default;

TestValueStoreFactory::StorageHelper::~StorageHelper() = default;

std::set<ValueStoreClientId>
TestValueStoreFactory::StorageHelper::GetKnownExtensionIDs(
    ModelType model_type) const {
  std::set<ValueStoreClientId> ids;
  switch (model_type) {
    case ValueStoreFactory::ModelType::APP:
      for (const auto& key : app_stores_)
        ids.insert(key.first);
      break;
    case ValueStoreFactory::ModelType::EXTENSION:
      for (const auto& key : extension_stores_)
        ids.insert(key.first);
      break;
  }
  return ids;
}

void TestValueStoreFactory::StorageHelper::Reset() {
  app_stores_.clear();
  extension_stores_.clear();
}

ValueStore* TestValueStoreFactory::StorageHelper::AddValueStore(
    const ValueStoreClientId& id,
    ValueStore* value_store,
    ModelType model_type) {
  if (model_type == ValueStoreFactory::ModelType::APP) {
    DCHECK(app_stores_.find(id) == app_stores_.end());
    app_stores_[id] = value_store;
  } else {
    DCHECK(extension_stores_.find(id) == extension_stores_.end());
    extension_stores_[id] = value_store;
  }
  return value_store;
}

void TestValueStoreFactory::StorageHelper::DeleteSettings(
    const ValueStoreClientId& id,
    ModelType model_type) {
  switch (model_type) {
    case ValueStoreFactory::ModelType::APP:
      app_stores_.erase(id);
      break;
    case ValueStoreFactory::ModelType::EXTENSION:
      extension_stores_.erase(id);
      break;
  }
}

bool TestValueStoreFactory::StorageHelper::HasSettings(
    const ValueStoreClientId& id,
    ModelType model_type) const {
  switch (model_type) {
    case ValueStoreFactory::ModelType::APP:
      return app_stores_.find(id) != app_stores_.end();
    case ValueStoreFactory::ModelType::EXTENSION:
      return extension_stores_.find(id) != extension_stores_.end();
  }
  NOTREACHED();
  return false;
}

ValueStore* TestValueStoreFactory::StorageHelper::GetExisting(
    const ValueStoreClientId& id) const {
  auto it = app_stores_.find(id);
  if (it != app_stores_.end())
    return it->second;
  it = extension_stores_.find(id);
  if (it != extension_stores_.end())
    return it->second;
  return nullptr;
}

TestValueStoreFactory::TestValueStoreFactory() = default;

TestValueStoreFactory::TestValueStoreFactory(const base::FilePath& db_path)
    : db_path_(db_path) {}

TestValueStoreFactory::~TestValueStoreFactory() = default;

std::unique_ptr<ValueStore> TestValueStoreFactory::CreateRulesStore() {
  if (db_path_.empty())
    last_created_store_ = new TestingValueStore();
  else
    last_created_store_ = new LeveldbValueStore(kUMAClientName, db_path_);
  return base::WrapUnique(last_created_store_);
}

std::unique_ptr<ValueStore> TestValueStoreFactory::CreateStateStore() {
  return CreateRulesStore();
}

TestValueStoreFactory::StorageHelper& TestValueStoreFactory::GetStorageHelper(
    SettingsNamespace settings_namespace) {
  switch (settings_namespace) {
    case settings_namespace::LOCAL:
      return local_helper_;
    case settings_namespace::SYNC:
      return sync_helper_;
    case settings_namespace::MANAGED:
      return managed_helper_;
    case settings_namespace::INVALID:
      break;
  }
  NOTREACHED();
  return local_helper_;
}

std::unique_ptr<ValueStore> TestValueStoreFactory::CreateSettingsStore(
    SettingsNamespace settings_namespace,
    ModelType model_type,
    const ValueStoreClientId& id) {
  std::unique_ptr<ValueStore> settings_store(CreateRulesStore());
  // Note: This factory is purposely keeping the raw pointers to each ValueStore
  //       created. Tests using TestValueStoreFactory must be careful to keep
  //       those ValueStore's alive for the duration of their test.
  GetStorageHelper(settings_namespace)
      .AddValueStore(id, settings_store.get(), model_type);
  return settings_store;
}

ValueStore* TestValueStoreFactory::LastCreatedStore() const {
  return last_created_store_;
}

void TestValueStoreFactory::DeleteSettings(SettingsNamespace settings_namespace,
                                           ModelType model_type,
                                           const ValueStoreClientId& id) {
  GetStorageHelper(settings_namespace).DeleteSettings(id, model_type);
}

bool TestValueStoreFactory::HasSettings(SettingsNamespace settings_namespace,
                                        ModelType model_type,
                                        const ValueStoreClientId& id) {
  return GetStorageHelper(settings_namespace).HasSettings(id, model_type);
}

std::set<ValueStoreClientId> TestValueStoreFactory::GetKnownExtensionIDs(
    SettingsNamespace settings_namespace,
    ModelType model_type) const {
  return const_cast<TestValueStoreFactory*>(this)
      ->GetStorageHelper(settings_namespace)
      .GetKnownExtensionIDs(model_type);
}

ValueStore* TestValueStoreFactory::GetExisting(
    const ValueStoreClientId& id) const {
  ValueStore* existing_store = local_helper_.GetExisting(id);
  if (existing_store)
    return existing_store;
  existing_store = sync_helper_.GetExisting(id);
  if (existing_store)
    return existing_store;
  existing_store = managed_helper_.GetExisting(id);
  DCHECK(existing_store != nullptr);
  return existing_store;
}

void TestValueStoreFactory::Reset() {
  last_created_store_ = nullptr;
  local_helper_.Reset();
  sync_helper_.Reset();
  managed_helper_.Reset();
}

}  // namespace extensions