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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/ash/app_restore/full_restore_prefs.h"
#include <memory>
#include "ash/constants/ash_pref_names.h"
#include "ash/wm/window_restore/window_restore_util.h"
#include "chrome/browser/ash/login/users/fake_chrome_user_manager.h"
#include "chrome/browser/prefs/session_startup_pref.h"
#include "chrome/common/pref_names.h"
#include "components/pref_registry/pref_registry_syncable.h"
#include "components/sync_preferences/testing_pref_service_syncable.h"
#include "components/user_manager/scoped_user_manager.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace ash::full_restore {
// Unit tests for full_restore_prefs.
class FullRestorePrefsTest : public testing::Test {
public:
FullRestorePrefsTest() = default;
void SetUp() override {
pref_service_ =
std::make_unique<sync_preferences::TestingPrefServiceSyncable>();
}
user_prefs::PrefRegistrySyncable* registry() {
return pref_service_->registry();
}
FakeChromeUserManager* GetFakeUserManager() const {
return fake_user_manager_.Get();
}
RestoreOption GetRestoreOption() const {
return static_cast<RestoreOption>(
pref_service_->GetInteger(prefs::kRestoreAppsAndPagesPrefName));
}
std::unique_ptr<sync_preferences::TestingPrefServiceSyncable> pref_service_;
user_manager::TypedScopedUserManager<ash::FakeChromeUserManager>
fake_user_manager_{std::make_unique<ash::FakeChromeUserManager>()};
};
// For a brand new user, set 'ask every time' as the default value.
TEST_F(FullRestorePrefsTest, NewUser) {
GetFakeUserManager()->SetIsCurrentUserNew(true);
RegisterProfilePrefs(registry());
SetDefaultRestorePrefIfNecessary(pref_service_.get());
EXPECT_EQ(RestoreOption::kAskEveryTime, GetRestoreOption());
EXPECT_TRUE(CanPerformRestore(pref_service_.get()));
}
// When a user upgrades to the full restore release, set 'ask every time' as the
// default value if the browser setting is 'continue where you left off'.
TEST_F(FullRestorePrefsTest, UpgradingFromRestore) {
SessionStartupPref::RegisterProfilePrefs(registry());
pref_service_->SetInteger(
::prefs::kRestoreOnStartup,
static_cast<int>(SessionStartupPref::kPrefValueLast));
RegisterProfilePrefs(registry());
SetDefaultRestorePrefIfNecessary(pref_service_.get());
EXPECT_EQ(RestoreOption::kAskEveryTime, GetRestoreOption());
EXPECT_TRUE(CanPerformRestore(pref_service_.get()));
}
// When a user upgrades to the full restore release, set 'do not restore' as the
// default value if the browser setting is 'new tab'.
TEST_F(FullRestorePrefsTest, UpgradingFromNotRestore) {
SessionStartupPref::RegisterProfilePrefs(registry());
pref_service_->SetInteger(
::prefs::kRestoreOnStartup,
static_cast<int>(SessionStartupPref::kPrefValueNewTab));
RegisterProfilePrefs(registry());
SetDefaultRestorePrefIfNecessary(pref_service_.get());
EXPECT_EQ(RestoreOption::kDoNotRestore, GetRestoreOption());
EXPECT_FALSE(CanPerformRestore(pref_service_.get()));
}
// For a new Chrome OS user, set 'always restore' as the default value if the
// browser setting is 'continue where you left off'.
TEST_F(FullRestorePrefsTest, NewChromeOSUserFromRestore) {
GetFakeUserManager()->SetIsCurrentUserNew(true);
RegisterProfilePrefs(registry());
SetDefaultRestorePrefIfNecessary(pref_service_.get());
EXPECT_EQ(RestoreOption::kAskEveryTime, GetRestoreOption());
EXPECT_TRUE(CanPerformRestore(pref_service_.get()));
SessionStartupPref::RegisterProfilePrefs(registry());
pref_service_->SetInteger(
::prefs::kRestoreOnStartup,
static_cast<int>(SessionStartupPref::kPrefValueLast));
UpdateRestorePrefIfNecessary(pref_service_.get());
EXPECT_EQ(RestoreOption::kAlways, GetRestoreOption());
EXPECT_TRUE(CanPerformRestore(pref_service_.get()));
}
// For a new Chrome OS user, set 'ask every time' as the default value if the
// browser setting is 'new tab'.
TEST_F(FullRestorePrefsTest, NewChromeOSUserFromNotRestore) {
GetFakeUserManager()->SetIsCurrentUserNew(true);
RegisterProfilePrefs(registry());
SetDefaultRestorePrefIfNecessary(pref_service_.get());
EXPECT_EQ(RestoreOption::kAskEveryTime, GetRestoreOption());
EXPECT_TRUE(CanPerformRestore(pref_service_.get()));
SessionStartupPref::RegisterProfilePrefs(registry());
pref_service_->SetInteger(
::prefs::kRestoreOnStartup,
static_cast<int>(SessionStartupPref::kPrefValueNewTab));
UpdateRestorePrefIfNecessary(pref_service_.get());
EXPECT_EQ(RestoreOption::kAskEveryTime, GetRestoreOption());
EXPECT_TRUE(CanPerformRestore(pref_service_.get()));
}
} // namespace ash::full_restore
|
