1 // Copyright 2020 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chrome/browser/chromeos/login/screens/multidevice_setup_screen.h"
6
7 #include "base/bind.h"
8 #include "base/run_loop.h"
9 #include "chrome/browser/browser_process.h"
10 #include "chrome/browser/chromeos/login/screen_manager.h"
11 #include "chrome/browser/chromeos/login/test/js_checker.h"
12 #include "chrome/browser/chromeos/login/test/login_manager_mixin.h"
13 #include "chrome/browser/chromeos/login/test/oobe_base_test.h"
14 #include "chrome/browser/chromeos/login/test/oobe_screen_exit_waiter.h"
15 #include "chrome/browser/chromeos/login/test/oobe_screen_waiter.h"
16 #include "chrome/browser/chromeos/login/ui/login_display_host.h"
17 #include "chrome/browser/chromeos/login/wizard_controller.h"
18 #include "chrome/browser/ui/webui/chromeos/login/gaia_screen_handler.h"
19 #include "chrome/browser/ui/webui/chromeos/login/multidevice_setup_screen_handler.h"
20 #include "chromeos/services/multidevice_setup/public/cpp/fake_multidevice_setup_client.h"
21 #include "content/public/test/browser_test.h"
22
23 namespace chromeos {
24
25 class MultiDeviceSetupScreenTest : public OobeBaseTest {
26 public:
27 MultiDeviceSetupScreenTest() = default;
28 ~MultiDeviceSetupScreenTest() override = default;
29
SetUpOnMainThread()30 void SetUpOnMainThread() override {
31 MultiDeviceSetupScreen* screen = static_cast<MultiDeviceSetupScreen*>(
32 WizardController::default_controller()->screen_manager()->GetScreen(
33 MultiDeviceSetupScreenView::kScreenId));
34 screen->AddExitCallbackForTesting(base::BindRepeating(
35 &MultiDeviceSetupScreenTest::HandleScreenExit, base::Unretained(this)));
36
37 fake_multidevice_setup_client_ =
38 std::make_unique<multidevice_setup::FakeMultiDeviceSetupClient>();
39 screen->set_multidevice_setup_client_for_testing(
40 fake_multidevice_setup_client_.get());
41 OobeBaseTest::SetUpOnMainThread();
42 }
43
SimulateHostStatusChange()44 void SimulateHostStatusChange() {
45 multidevice_setup::MultiDeviceSetupClient::HostStatusWithDevice
46 host_status_with_device = multidevice_setup::MultiDeviceSetupClient::
47 GenerateDefaultHostStatusWithDevice();
48 host_status_with_device.first =
49 multidevice_setup::mojom::HostStatus::kEligibleHostExistsButNoHostSet;
50 fake_multidevice_setup_client_->SetHostStatusWithDevice(
51 host_status_with_device);
52 }
53
ShowMultiDeviceSetupScreen()54 void ShowMultiDeviceSetupScreen() {
55 login_manager_mixin_.LoginAsNewRegularUser();
56 OobeScreenExitWaiter(GetFirstSigninScreen()).Wait();
57 if (!screen_exited_) {
58 LoginDisplayHost::default_host()->StartWizard(
59 MultiDeviceSetupScreenView::kScreenId);
60 }
61 }
62
FinishDeviceSetup()63 void FinishDeviceSetup() {
64 test::OobeJS().Evaluate(
65 R"($('multidevice-setup-impl')
66 .$['multideviceSetup']
67 .fire('setup-exited', {didUserCompleteSetup: true});)");
68 }
69
CancelDeviceSetup()70 void CancelDeviceSetup() {
71 test::OobeJS().Evaluate(
72 R"($('multidevice-setup-impl')
73 .$['multideviceSetup']
74 .fire('setup-exited', {didUserCompleteSetup: false});)");
75 }
76
WaitForScreenShown()77 void WaitForScreenShown() {
78 OobeScreenWaiter(MultiDeviceSetupScreenView::kScreenId).Wait();
79 }
80
WaitForScreenExit()81 void WaitForScreenExit() {
82 if (screen_exited_)
83 return;
84 base::RunLoop run_loop;
85 screen_exit_callback_ = run_loop.QuitClosure();
86 run_loop.Run();
87 }
88
CheckUserChoice(bool Accepted)89 void CheckUserChoice(bool Accepted) {
90 histogram_tester_.ExpectBucketCount<
91 MultiDeviceSetupScreen::MultiDeviceSetupOOBEUserChoice>(
92 "MultiDeviceSetup.OOBE.UserChoice",
93 MultiDeviceSetupScreen::MultiDeviceSetupOOBEUserChoice::kAccepted,
94 Accepted);
95 histogram_tester_.ExpectBucketCount<
96 MultiDeviceSetupScreen::MultiDeviceSetupOOBEUserChoice>(
97 "MultiDeviceSetup.OOBE.UserChoice",
98 MultiDeviceSetupScreen::MultiDeviceSetupOOBEUserChoice::kDeclined,
99 !Accepted);
100 }
101
102 base::Optional<MultiDeviceSetupScreen::Result> screen_result_;
103 base::HistogramTester histogram_tester_;
104
105 private:
HandleScreenExit(MultiDeviceSetupScreen::Result result)106 void HandleScreenExit(MultiDeviceSetupScreen::Result result) {
107 ASSERT_FALSE(screen_exited_);
108 screen_exited_ = true;
109 screen_result_ = result;
110 if (screen_exit_callback_)
111 std::move(screen_exit_callback_).Run();
112 }
113
114 bool screen_exited_ = false;
115 base::RepeatingClosure screen_exit_callback_;
116 std::unique_ptr<multidevice_setup::FakeMultiDeviceSetupClient>
117 fake_multidevice_setup_client_;
118
119 LoginManagerMixin login_manager_mixin_{&mixin_host_};
120 };
121
IN_PROC_BROWSER_TEST_F(MultiDeviceSetupScreenTest,Accepted)122 IN_PROC_BROWSER_TEST_F(MultiDeviceSetupScreenTest, Accepted) {
123 SimulateHostStatusChange();
124 ShowMultiDeviceSetupScreen();
125 WaitForScreenShown();
126
127 FinishDeviceSetup();
128
129 WaitForScreenExit();
130 EXPECT_EQ(screen_result_.value(), MultiDeviceSetupScreen::Result::NEXT);
131 histogram_tester_.ExpectTotalCount(
132 "OOBE.StepCompletionTimeByExitReason.Multidevice-setup.Next", 1);
133 histogram_tester_.ExpectTotalCount(
134 "OOBE.StepCompletionTime.Multidevice-setup", 1);
135 CheckUserChoice(true);
136 }
137
IN_PROC_BROWSER_TEST_F(MultiDeviceSetupScreenTest,Declined)138 IN_PROC_BROWSER_TEST_F(MultiDeviceSetupScreenTest, Declined) {
139 SimulateHostStatusChange();
140 ShowMultiDeviceSetupScreen();
141 WaitForScreenShown();
142
143 CancelDeviceSetup();
144
145 WaitForScreenExit();
146 EXPECT_EQ(screen_result_.value(), MultiDeviceSetupScreen::Result::NEXT);
147 histogram_tester_.ExpectTotalCount(
148 "OOBE.StepCompletionTimeByExitReason.Multidevice-setup.Next", 1);
149 histogram_tester_.ExpectTotalCount(
150 "OOBE.StepCompletionTime.Multidevice-setup", 1);
151 CheckUserChoice(false);
152 }
153
IN_PROC_BROWSER_TEST_F(MultiDeviceSetupScreenTest,Skipped)154 IN_PROC_BROWSER_TEST_F(MultiDeviceSetupScreenTest, Skipped) {
155 ShowMultiDeviceSetupScreen();
156
157 WaitForScreenExit();
158 EXPECT_EQ(screen_result_.value(),
159 MultiDeviceSetupScreen::Result::NOT_APPLICABLE);
160 histogram_tester_.ExpectTotalCount(
161 "OOBE.StepCompletionTimeByExitReason.Multidevice-setup.Next", 0);
162 histogram_tester_.ExpectTotalCount(
163 "OOBE.StepCompletionTime.Multidevice-setup", 0);
164 }
165
166 } // namespace chromeos
167