// 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