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
86
87
|
// 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 {
// static
std::unique_ptr<TestingPrefServiceSimple>
TestPrefServiceProvider::CreateUserPrefServiceSimple() {
auto pref_service = std::make_unique<TestingPrefServiceSimple>();
RegisterUserProfilePrefs(pref_service->registry(), /*country=*/"",
/*for_test=*/true);
return pref_service;
}
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::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
|
