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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
|
// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <string>
#include "base/bind.h"
#include "base/logging.h"
#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "base/memory/scoped_ptr.h"
#include "base/run_loop.h"
#include "chrome/browser/chromeos/login/test/oobe_base_test.h"
#include "chrome/browser/chromeos/login/test/oobe_screen_waiter.h"
#include "chrome/browser/chromeos/login/ui/login_display_host_impl.h"
#include "chrome/browser/chromeos/login/ui/oobe_display.h"
#include "chrome/browser/chromeos/login/ui/webui_login_view.h"
#include "chrome/browser/chromeos/login/wizard_controller.h"
#include "chrome/browser/chromeos/policy/device_policy_builder.h"
#include "chrome/browser/chromeos/policy/device_policy_cros_browser_test.h"
#include "chrome/browser/chromeos/profiles/profile_helper.h"
#include "chrome/browser/chromeos/settings/cros_settings.h"
#include "chrome/browser/ui/webui/chromeos/login/network_state_informer.h"
#include "chrome/browser/ui/webui/chromeos/login/oobe_ui.h"
#include "chrome/browser/ui/webui/chromeos/login/signin_screen_handler.h"
#include "chromeos/dbus/dbus_thread_manager.h"
#include "chromeos/dbus/fake_session_manager_client.h"
#include "chromeos/dbus/fake_shill_manager_client.h"
#include "chromeos/dbus/session_manager_client.h"
#include "chromeos/dbus/shill_manager_client.h"
#include "chromeos/dbus/shill_service_client.h"
#include "chromeos/settings/cros_settings_names.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/browser_test_utils.h"
#include "dbus/object_path.h"
#include "policy/proto/device_management_backend.pb.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace chromeos {
namespace system {
namespace {
void ErrorCallbackFunction(const std::string& error_name,
const std::string& error_message) {
LOG(ERROR) << "Shill Error: " << error_name << " : " << error_message;
}
} // namespace
class DeviceDisablingTest
: public OobeBaseTest,
public NetworkStateInformer::NetworkStateInformerObserver {
public:
DeviceDisablingTest();
// Sets up a device state blob that indicates the device is disabled, triggers
// a policy plus device state fetch and waits for it to succeed.
void MarkDisabledAndWaitForPolicyFetch();
std::string GetCurrentScreenName(content::WebContents* web_contents);
protected:
base::RunLoop network_state_change_wait_run_loop_;
private:
// OobeBaseTest:
void SetUpInProcessBrowserTestFixture() override;
void SetUpOnMainThread() override;
// NetworkStateInformer::NetworkStateInformerObserver:
void UpdateState(ErrorScreenActor::ErrorReason reason) override;
FakeSessionManagerClient* fake_session_manager_client_;
policy::DevicePolicyCrosTestHelper test_helper_;
DISALLOW_COPY_AND_ASSIGN(DeviceDisablingTest);
};
DeviceDisablingTest::DeviceDisablingTest()
: fake_session_manager_client_(new FakeSessionManagerClient) {
}
void DeviceDisablingTest::MarkDisabledAndWaitForPolicyFetch() {
base::RunLoop run_loop;
// Set up an |observer| that will wait for the disabled setting to change.
scoped_ptr<CrosSettings::ObserverSubscription> observer =
CrosSettings::Get()->AddSettingsObserver(
kDeviceDisabled,
run_loop.QuitClosure());
// Prepare a policy fetch response that indicates the device is disabled.
test_helper_.device_policy()->policy_data().mutable_device_state()->
set_device_mode(enterprise_management::DeviceState::DEVICE_MODE_DISABLED);
test_helper_.device_policy()->Build();
fake_session_manager_client_->set_device_policy(
test_helper_.device_policy()->GetBlob());
// Trigger a policy fetch.
fake_session_manager_client_->OnPropertyChangeComplete(true);
// Wait for the policy fetch to complete and the disabled setting to change.
run_loop.Run();
}
std::string DeviceDisablingTest::GetCurrentScreenName(
content::WebContents* web_contents ) {
std::string screen_name;
if (!content::ExecuteScriptAndExtractString(
web_contents,
"domAutomationController.send(Oobe.getInstance().currentScreen.id);",
&screen_name)) {
ADD_FAILURE();
}
return screen_name;
}
void DeviceDisablingTest::SetUpInProcessBrowserTestFixture() {
OobeBaseTest::SetUpInProcessBrowserTestFixture();
DBusThreadManager::GetSetterForTesting()->SetSessionManagerClient(
scoped_ptr<SessionManagerClient>(fake_session_manager_client_));
test_helper_.InstallOwnerKey();
test_helper_.MarkAsEnterpriseOwned();
}
void DeviceDisablingTest::SetUpOnMainThread() {
OobeBaseTest::SetUpOnMainThread();
// Set up fake networks.
DBusThreadManager::Get()->GetShillManagerClient()->GetTestInterface()->
SetupDefaultEnvironment();
}
void DeviceDisablingTest::UpdateState(ErrorScreenActor::ErrorReason reason) {
network_state_change_wait_run_loop_.Quit();
}
IN_PROC_BROWSER_TEST_F(DeviceDisablingTest, DisableDuringNormalOperation) {
// Mark the device as disabled and wait until cros settings update.
MarkDisabledAndWaitForPolicyFetch();
// Verify that the device disabled screen is being shown.
WizardController* const wizard_controller =
WizardController::default_controller();
ASSERT_TRUE(wizard_controller);
EXPECT_EQ(wizard_controller->GetScreen(
WizardController::kDeviceDisabledScreenName),
wizard_controller->current_screen());
}
// Verifies that device disabling works when the ephemeral users policy is
// enabled. This case warrants its own test because the UI behaves somewhat
// differently when the policy is set: A background job runs on startup that
// causes the UI to try and show the login screen after some delay. It must
// be ensured that the login screen does not show and does not clobber the
// disabled screen.
IN_PROC_BROWSER_TEST_F(DeviceDisablingTest, DisableWithEphemeralUsers) {
// Connect to the fake Ethernet network. This ensures that Chrome OS will not
// try to show the offline error screen.
base::RunLoop connect_run_loop;
DBusThreadManager::Get()->GetShillServiceClient()->Connect(
dbus::ObjectPath("/service/eth1"),
connect_run_loop.QuitClosure(),
base::Bind(&ErrorCallbackFunction));
connect_run_loop.Run();
// Skip to the login screen.
WizardController* wizard_controller = WizardController::default_controller();
ASSERT_TRUE(wizard_controller);
wizard_controller->SkipToLoginForTesting(LoginScreenContext());
OobeScreenWaiter(OobeDisplay::SCREEN_GAIA_SIGNIN).Wait();
// Mark the device as disabled and wait until cros settings update.
MarkDisabledAndWaitForPolicyFetch();
// When the ephemeral users policy is enabled, Chrome OS removes any non-owner
// cryptohomes on startup. At the end of that process, JavaScript attempts to
// show the login screen. Simulate this.
const LoginDisplayHostImpl* const host =
static_cast<LoginDisplayHostImpl*>(LoginDisplayHostImpl::default_host());
ASSERT_TRUE(host);
WebUILoginView* const webui_login_view = host->GetWebUILoginView();
ASSERT_TRUE(webui_login_view);
content::WebContents* const web_contents = webui_login_view->GetWebContents();
ASSERT_TRUE(web_contents);
ASSERT_TRUE(content::ExecuteScript(web_contents,
"Oobe.showAddUserForTesting();"));
// The login profile is scrubbed before attempting to show the login screen.
// Wait for the scrubbing to finish.
base::RunLoop run_loop;
ProfileHelper::Get()->ClearSigninProfile(run_loop.QuitClosure());
run_loop.Run();
base::RunLoop().RunUntilIdle();
// Verify that the login screen was not shown and the device disabled screen
// is still being shown instead.
EXPECT_EQ(OobeUI::kScreenDeviceDisabled, GetCurrentScreenName(web_contents));
// Disconnect from the fake Ethernet network.
OobeUI* const oobe_ui = host->GetOobeUI();
ASSERT_TRUE(oobe_ui);
const scoped_refptr<NetworkStateInformer> network_state_informer =
oobe_ui->network_state_informer_for_test();
ASSERT_TRUE(network_state_informer);
network_state_informer->AddObserver(this);
SigninScreenHandler* const signin_screen_handler =
oobe_ui->signin_screen_handler_for_test();
ASSERT_TRUE(signin_screen_handler);
signin_screen_handler->ZeroOfflineTimeoutForTesting();
SimulateNetworkOffline();
network_state_change_wait_run_loop_.Run();
network_state_informer->RemoveObserver(this);
base::RunLoop().RunUntilIdle();
// Verify that the offline error screen was not shown and the device disabled
// screen is still being shown instead.
EXPECT_EQ(OobeUI::kScreenDeviceDisabled, GetCurrentScreenName(web_contents));
}
} // namespace system
} // namespace chromeos
|
