1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85


ash / session / test_pref_service_provider.cc [blame]

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

#include "ash/session/test_pref_service_provider.h"

#include <algorithm>

#include "ash/public/cpp/ash_prefs.h"
#include "base/check.h"
#include "components/account_id/account_id.h"
#include "components/prefs/testing_pref_service.h"

namespace ash {

TestPrefServiceProvider::TestPrefServiceProvider() = default;
TestPrefServiceProvider::~TestPrefServiceProvider() = default;

void TestPrefServiceProvider::CreateSigninPrefsIfNeeded() {
  if (signin_prefs_)
    return;

  auto pref_service = std::make_unique<TestingPrefServiceSimple>();
  RegisterSigninProfilePrefs(pref_service->registry(), /*country=*/"",
                             /**for_test=*/true);
  signin_prefs_ = std::move(pref_service);
}

void TestPrefServiceProvider::SetSigninPrefs(
    std::unique_ptr<PrefService> signin_prefs) {
  DCHECK(!signin_prefs_);
  signin_prefs_ = std::move(signin_prefs);
}

PrefService* TestPrefServiceProvider::GetSigninPrefs() {
  return signin_prefs_.get();
}

void TestPrefServiceProvider::CreateUserPrefs(const AccountId& account_id) {
  auto pref_service = std::make_unique<TestingPrefServiceSimple>();
  RegisterUserProfilePrefs(pref_service->registry(), /*country=*/"",
                           /*for_test=*/true);
  SetUserPrefs(account_id, std::move(pref_service));
}

void TestPrefServiceProvider::SetUserPrefs(
    const AccountId& account_id,
    std::unique_ptr<PrefService> pref_service) {
  CHECK_EQ(GetUserPrefs(account_id), nullptr);

  user_prefs_map_.emplace(account_id, std::move(pref_service));
}

void TestPrefServiceProvider::SetUnownedUserPrefs(
    const AccountId& account_id,
    raw_ptr<PrefService> unowned_pref_service) {
  CHECK_EQ(GetUserPrefs(account_id), nullptr);

  unowned_user_prefs_map_.emplace(account_id, std::move(unowned_pref_service));
}

PrefService* TestPrefServiceProvider::GetUserPrefs(
    const AccountId& account_id) {
  auto it = user_prefs_map_.find(account_id);
  if (it != user_prefs_map_.end()) {
    return it->second.get();
  }

  auto unowned_it = unowned_user_prefs_map_.find(account_id);
  if (unowned_it != unowned_user_prefs_map_.end()) {
    return unowned_it->second.get();
  }

  return nullptr;
}

void TestPrefServiceProvider::ClearUnownedUserPrefs(
    const AccountId& account_id) {
  auto unowned_it = unowned_user_prefs_map_.find(account_id);
  CHECK(unowned_it != unowned_user_prefs_map_.end());

  unowned_user_prefs_map_.erase(unowned_it);
}

}  // namespace ash