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
|
// Copyright 2018 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/system/user_removal_manager.h"
#include <memory>
#include "base/functional/bind.h"
#include "base/functional/callback.h"
#include "base/memory/scoped_refptr.h"
#include "base/test/test_mock_time_task_runner.h"
#include "chrome/browser/ash/login/users/fake_chrome_user_manager.h"
#include "chrome/browser/ash/settings/scoped_testing_cros_settings.h"
#include "chrome/common/pref_names.h"
#include "chrome/test/base/testing_browser_process.h"
#include "chromeos/ash/components/install_attributes/stub_install_attributes.h"
#include "components/account_id/account_id.h"
#include "components/prefs/pref_service.h"
#include "components/prefs/testing_pref_service.h"
#include "components/user_manager/fake_user_manager_delegate.h"
#include "components/user_manager/scoped_user_manager.h"
#include "components/user_manager/test_helper.h"
#include "components/user_manager/user_manager.h"
#include "components/user_manager/user_manager_impl.h"
#include "google_apis/gaia/gaia_id.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace ash {
namespace {
class UserRemovalManagerTest : public testing::Test {
public:
UserRemovalManagerTest(const UserRemovalManagerTest&) = delete;
UserRemovalManagerTest& operator=(const UserRemovalManagerTest&) = delete;
protected:
UserRemovalManagerTest();
~UserRemovalManagerTest() override;
void SetUp() override {
testing::Test::SetUp();
// Emulate managed device.
user_manager_->SetOwnerId(EmptyAccountId());
user_manager::TestHelper test_helper(user_manager_.Get());
ASSERT_TRUE(test_helper.AddRegularUser(
AccountId::FromUserEmailGaiaId("user1", GaiaId("1"))));
ASSERT_TRUE(test_helper.AddRegularUser(
AccountId::FromUserEmailGaiaId("user2", GaiaId("2"))));
ASSERT_TRUE(test_helper.AddRegularUser(
AccountId::FromUserEmailGaiaId("user3", GaiaId("3"))));
}
PrefService* local_state() {
return TestingBrowserProcess::GetGlobal()->GetTestingLocalState();
}
scoped_refptr<base::TestMockTimeTaskRunner> task_runner_;
ash::ScopedStubInstallAttributes install_attributes_{
ash::StubInstallAttributes::CreateCloudManaged("test.domain",
"device_id")};
user_manager::ScopedUserManager user_manager_{
std::make_unique<user_manager::UserManagerImpl>(
std::make_unique<user_manager::FakeUserManagerDelegate>(),
TestingBrowserProcess::GetGlobal()->GetTestingLocalState())};
};
UserRemovalManagerTest::UserRemovalManagerTest()
: task_runner_(base::MakeRefCounted<base::TestMockTimeTaskRunner>(
base::TestMockTimeTaskRunner::Type::kBoundToThread)) {}
UserRemovalManagerTest::~UserRemovalManagerTest() = default;
} // namespace
// Test that the InitiateUserRemoval/RemoveUsersIfNeeded sequence results in
// users being removed from the device.
TEST_F(UserRemovalManagerTest, TestUserRemovingWorks) {
user_removal_manager::InitiateUserRemoval(base::OnceClosure());
EXPECT_TRUE(user_removal_manager::RemoveUsersIfNeeded());
EXPECT_TRUE(user_manager::UserManager::Get()->GetPersistedUsers().empty());
EXPECT_TRUE(local_state()
->FindPreference(prefs::kRemoveUsersRemoteCommand)
->IsDefaultValue());
}
// Test that if Chrome crashes in the middle of the users removal - it does not
// try again.
TEST_F(UserRemovalManagerTest, TestUserRemovingDoNotRetryOnFailure) {
// If explicitly set to false - it means chrome might've crashed during the
// previous removal.
local_state()->SetBoolean(prefs::kRemoveUsersRemoteCommand, false);
EXPECT_FALSE(user_removal_manager::RemoveUsersIfNeeded());
EXPECT_FALSE(user_manager::UserManager::Get()->GetPersistedUsers().empty());
}
// Test that the failsafe timer runs LogOut after 60 seconds.
TEST_F(UserRemovalManagerTest, TestFailsafeTimer) {
bool log_out_called = false;
user_removal_manager::OverrideLogOutForTesting(base::BindOnce(
[](bool* log_out_called) { *log_out_called = true; }, &log_out_called));
// This call starts the timer.
user_removal_manager::InitiateUserRemoval(base::OnceClosure());
// After 55s the timer is not run yet.
task_runner_->FastForwardBy(base::Seconds(55));
EXPECT_FALSE(log_out_called);
// After 60s the timer is run.
task_runner_->FastForwardBy(base::Seconds(5));
EXPECT_TRUE(log_out_called);
}
} // namespace ash
|
