OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2017 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/enrollment/enrollment_screen.h" |
| 6 #include "base/bind.h" |
| 7 #include "base/command_line.h" |
| 8 #include "base/message_loop/message_loop.h" |
| 9 #include "base/test/scoped_mock_time_message_loop_task_runner.h" |
| 10 #include "base/threading/thread_task_runner_handle.h" |
| 11 #include "chrome/browser/chromeos/login/enrollment/enterprise_enrollment_helper.
h" |
| 12 #include "chrome/browser/chromeos/login/enrollment/enterprise_enrollment_helper_
mock.h" |
| 13 #include "chrome/browser/chromeos/login/enrollment/mock_enrollment_screen.h" |
| 14 #include "chrome/browser/chromeos/login/screens/mock_base_screen_delegate.h" |
| 15 #include "chrome/browser/chromeos/policy/device_cloud_policy_manager_chromeos.h" |
| 16 #include "chrome/browser/chromeos/policy/enrollment_config.h" |
| 17 #include "chrome/test/base/testing_browser_process.h" |
| 18 #include "chromeos/chromeos_switches.h" |
| 19 #include "chromeos/dbus/dbus_thread_manager.h" |
| 20 #include "components/pairing/fake_controller_pairing_controller.h" |
| 21 #include "testing/gtest/include/gtest/gtest.h" |
| 22 |
| 23 using testing::AnyNumber; |
| 24 using testing::Invoke; |
| 25 |
| 26 namespace chromeos { |
| 27 |
| 28 class EnrollmentScreenUnitTest : public testing::Test { |
| 29 public: |
| 30 EnrollmentScreenUnitTest() : fake_controller_("") { |
| 31 enrollment_config_.mode = policy::EnrollmentConfig::MODE_ATTESTATION_FORCED; |
| 32 enrollment_config_.auth_mechanism = |
| 33 policy::EnrollmentConfig::AUTH_MECHANISM_ATTESTATION; |
| 34 } |
| 35 |
| 36 // Closure passed to EnterpriseEnrollmentHelper::SetupEnrollmentHelperMock |
| 37 // which creates the EnterpriseEnrollmentHelperMock object that will |
| 38 // eventually be tied to the EnrollmentScreen. It also sets up the |
| 39 // appropriate expectations for testing with the Google Mock framework. |
| 40 // The template parameter should_enroll indicates whether or not |
| 41 // the EnterpriseEnrollmentHelper should be mocked to successfully enroll. |
| 42 template <bool should_enroll> |
| 43 static EnterpriseEnrollmentHelper* MockEnrollmentHelperCreator( |
| 44 EnterpriseEnrollmentHelper::EnrollmentStatusConsumer* status_consumer, |
| 45 const policy::EnrollmentConfig& enrollment_config, |
| 46 const std::string& enrolling_user_domain) { |
| 47 EnterpriseEnrollmentHelperMock* mock = |
| 48 new EnterpriseEnrollmentHelperMock(status_consumer); |
| 49 if (should_enroll) { |
| 50 EXPECT_CALL(*mock, EnrollUsingAttestation()) |
| 51 .Times(AnyNumber()) |
| 52 .WillRepeatedly(Invoke([mock]() { |
| 53 static_cast<EnrollmentScreen*>(mock->status_consumer()) |
| 54 ->ShowEnrollmentStatusOnSuccess(); |
| 55 })); |
| 56 } else { |
| 57 EXPECT_CALL(*mock, EnrollUsingAttestation()) |
| 58 .Times(AnyNumber()) |
| 59 .WillRepeatedly(Invoke([mock]() { |
| 60 mock->status_consumer()->OnEnrollmentError( |
| 61 policy::EnrollmentStatus::ForStatus( |
| 62 policy::EnrollmentStatus::REGISTRATION_FAILED)); |
| 63 })); |
| 64 } |
| 65 return mock; |
| 66 } |
| 67 |
| 68 // Creates the EnrollmentScreen and sets required parameters. |
| 69 void SetUpEnrollmentScreen() { |
| 70 enrollment_screen_.reset( |
| 71 new EnrollmentScreen(&mock_delegate_, &mock_actor_)); |
| 72 enrollment_screen_->SetParameters(enrollment_config_, &fake_controller_); |
| 73 } |
| 74 |
| 75 // Fast forwards time by the specified amount. |
| 76 void FastForwardTime(base::TimeDelta time) { |
| 77 runner_.task_runner()->FastForwardBy(time); |
| 78 } |
| 79 |
| 80 // testing::Test: |
| 81 void SetUp() override { |
| 82 // Initialize the thread manager. |
| 83 DBusThreadManager::Initialize(); |
| 84 |
| 85 // Configure the browser to use Hands-Off Enrollment. |
| 86 base::CommandLine::ForCurrentProcess()->AppendSwitchASCII( |
| 87 switches::kEnterpriseEnableZeroTouchEnrollment, "hands-off"); |
| 88 } |
| 89 |
| 90 void TearDown() override { |
| 91 TestingBrowserProcess::GetGlobal()->SetShuttingDown(true); |
| 92 DBusThreadManager::Shutdown(); |
| 93 } |
| 94 |
| 95 protected: |
| 96 // A pointer to the EnrollmentScreen. |
| 97 std::unique_ptr<EnrollmentScreen> enrollment_screen_; |
| 98 |
| 99 private: |
| 100 // Replace main thread's task runner with a mock for duration of test. |
| 101 base::MessageLoop loop_; |
| 102 base::ScopedMockTimeMessageLoopTaskRunner runner_; |
| 103 |
| 104 // Objects required by the EnrollmentScreen that can be re-used. |
| 105 policy::EnrollmentConfig enrollment_config_; |
| 106 pairing_chromeos::FakeControllerPairingController fake_controller_; |
| 107 MockBaseScreenDelegate mock_delegate_; |
| 108 MockEnrollmentScreenActor mock_actor_; |
| 109 |
| 110 DISALLOW_COPY_AND_ASSIGN(EnrollmentScreenUnitTest); |
| 111 }; |
| 112 |
| 113 TEST_F(EnrollmentScreenUnitTest, Retries) { |
| 114 // Define behavior of EnterpriseEnrollmentHelperMock to always fail |
| 115 // enrollment. |
| 116 EnterpriseEnrollmentHelper::SetupEnrollmentHelperMock( |
| 117 &EnrollmentScreenUnitTest::MockEnrollmentHelperCreator<false>); |
| 118 |
| 119 SetUpEnrollmentScreen(); |
| 120 |
| 121 // Remove jitter to enable deterministic testing. |
| 122 enrollment_screen_->retry_policy_.jitter_factor = 0; |
| 123 |
| 124 // Start zero-touch enrollment. |
| 125 enrollment_screen_->Show(); |
| 126 |
| 127 // Fast forward time by 1 minute. |
| 128 FastForwardTime(base::TimeDelta::FromMinutes(1)); |
| 129 |
| 130 // Check that we have retried 4 times. |
| 131 EXPECT_EQ(enrollment_screen_->num_retries_, 4); |
| 132 } |
| 133 |
| 134 TEST_F(EnrollmentScreenUnitTest, DoesNotRetryOnTopOfUser) { |
| 135 // Define behavior of EnterpriseEnrollmentHelperMock to always fail |
| 136 // enrollment. |
| 137 EnterpriseEnrollmentHelper::SetupEnrollmentHelperMock( |
| 138 &EnrollmentScreenUnitTest::MockEnrollmentHelperCreator<false>); |
| 139 |
| 140 SetUpEnrollmentScreen(); |
| 141 |
| 142 // Remove jitter to enable deterministic testing. |
| 143 enrollment_screen_->retry_policy_.jitter_factor = 0; |
| 144 |
| 145 // Start zero-touch enrollment. |
| 146 enrollment_screen_->Show(); |
| 147 |
| 148 // Schedule user retry button click after 30 sec. |
| 149 base::ThreadTaskRunnerHandle::Get()->PostDelayedTask( |
| 150 FROM_HERE, base::Bind(&EnrollmentScreen::OnRetry, |
| 151 enrollment_screen_->weak_ptr_factory_.GetWeakPtr()), |
| 152 base::TimeDelta::FromSeconds(30)); |
| 153 |
| 154 // Fast forward time by 1 minute. |
| 155 FastForwardTime(base::TimeDelta::FromMinutes(1)); |
| 156 |
| 157 // Check that the number of retries is still 4. |
| 158 EXPECT_EQ(enrollment_screen_->num_retries_, 4); |
| 159 } |
| 160 |
| 161 TEST_F(EnrollmentScreenUnitTest, DoesNotRetryAfterSuccess) { |
| 162 // Define behavior of EnterpriseEnrollmentHelperMock to successfully enroll. |
| 163 EnterpriseEnrollmentHelper::SetupEnrollmentHelperMock( |
| 164 &EnrollmentScreenUnitTest::MockEnrollmentHelperCreator<true>); |
| 165 |
| 166 SetUpEnrollmentScreen(); |
| 167 |
| 168 // Start zero-touch enrollment. |
| 169 enrollment_screen_->Show(); |
| 170 |
| 171 // Fast forward time by 1 minute. |
| 172 FastForwardTime(base::TimeDelta::FromMinutes(1)); |
| 173 |
| 174 // Check that we do not retry. |
| 175 EXPECT_EQ(enrollment_screen_->num_retries_, 0); |
| 176 } |
| 177 } // namespace chromeos |
OLD | NEW |