OLD | NEW |
1 // Copyright (c) 2015 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
| 5 #include "chrome/browser/chromeos/policy/status_uploader.h" |
| 6 |
| 7 #include <utility> |
| 8 |
5 #include "base/prefs/testing_pref_service.h" | 9 #include "base/prefs/testing_pref_service.h" |
6 #include "base/test/test_simple_task_runner.h" | 10 #include "base/test/test_simple_task_runner.h" |
7 #include "base/time/time.h" | 11 #include "base/time/time.h" |
8 #include "chrome/browser/chromeos/policy/device_local_account.h" | 12 #include "chrome/browser/chromeos/policy/device_local_account.h" |
9 #include "chrome/browser/chromeos/policy/device_status_collector.h" | 13 #include "chrome/browser/chromeos/policy/device_status_collector.h" |
10 #include "chrome/browser/chromeos/policy/status_uploader.h" | |
11 #include "chrome/browser/chromeos/settings/scoped_cros_settings_test_helper.h" | 14 #include "chrome/browser/chromeos/settings/scoped_cros_settings_test_helper.h" |
12 #include "chromeos/settings/cros_settings_names.h" | 15 #include "chromeos/settings/cros_settings_names.h" |
13 #include "components/policy/core/common/cloud/cloud_policy_client.h" | 16 #include "components/policy/core/common/cloud/cloud_policy_client.h" |
14 #include "components/policy/core/common/cloud/mock_cloud_policy_client.h" | 17 #include "components/policy/core/common/cloud/mock_cloud_policy_client.h" |
15 #include "components/policy/core/common/cloud/mock_device_management_service.h" | 18 #include "components/policy/core/common/cloud/mock_device_management_service.h" |
16 #include "content/public/test/test_browser_thread_bundle.h" | 19 #include "content/public/test/test_browser_thread_bundle.h" |
17 #include "content/public/test/test_utils.h" | 20 #include "content/public/test/test_utils.h" |
18 #include "net/url_request/url_request_context_getter.h" | 21 #include "net/url_request/url_request_context_getter.h" |
19 #include "testing/gmock/include/gmock/gmock.h" | 22 #include "testing/gmock/include/gmock/gmock.h" |
20 #include "testing/gtest/include/gtest/gtest.h" | 23 #include "testing/gtest/include/gtest/gtest.h" |
(...skipping 108 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
129 chromeos::ScopedCrosSettingsTestHelper settings_helper_; | 132 chromeos::ScopedCrosSettingsTestHelper settings_helper_; |
130 scoped_ptr<MockDeviceStatusCollector> collector_; | 133 scoped_ptr<MockDeviceStatusCollector> collector_; |
131 ui::UserActivityDetector detector_; | 134 ui::UserActivityDetector detector_; |
132 MockCloudPolicyClient client_; | 135 MockCloudPolicyClient client_; |
133 MockDeviceManagementService device_management_service_; | 136 MockDeviceManagementService device_management_service_; |
134 TestingPrefServiceSimple prefs_; | 137 TestingPrefServiceSimple prefs_; |
135 }; | 138 }; |
136 | 139 |
137 TEST_F(StatusUploaderTest, BasicTest) { | 140 TEST_F(StatusUploaderTest, BasicTest) { |
138 EXPECT_TRUE(task_runner_->GetPendingTasks().empty()); | 141 EXPECT_TRUE(task_runner_->GetPendingTasks().empty()); |
139 StatusUploader uploader(&client_, collector_.Pass(), task_runner_); | 142 StatusUploader uploader(&client_, std::move(collector_), task_runner_); |
140 EXPECT_EQ(1U, task_runner_->GetPendingTasks().size()); | 143 EXPECT_EQ(1U, task_runner_->GetPendingTasks().size()); |
141 // On startup, first update should happen immediately. | 144 // On startup, first update should happen immediately. |
142 EXPECT_EQ(base::TimeDelta(), task_runner_->NextPendingTaskDelay()); | 145 EXPECT_EQ(base::TimeDelta(), task_runner_->NextPendingTaskDelay()); |
143 } | 146 } |
144 | 147 |
145 TEST_F(StatusUploaderTest, DifferentFrequencyAtStart) { | 148 TEST_F(StatusUploaderTest, DifferentFrequencyAtStart) { |
146 // Keep a pointer to the mock collector because collector_ gets cleared | 149 // Keep a pointer to the mock collector because collector_ gets cleared |
147 // when it is passed to the StatusUploader constructor below. | 150 // when it is passed to the StatusUploader constructor below. |
148 MockDeviceStatusCollector* const mock_collector = collector_.get(); | 151 MockDeviceStatusCollector* const mock_collector = collector_.get(); |
149 const int new_delay = StatusUploader::kDefaultUploadDelayMs * 2; | 152 const int new_delay = StatusUploader::kDefaultUploadDelayMs * 2; |
150 settings_helper_.SetInteger(chromeos::kReportUploadFrequency, new_delay); | 153 settings_helper_.SetInteger(chromeos::kReportUploadFrequency, new_delay); |
151 const base::TimeDelta expected_delay = base::TimeDelta::FromMilliseconds( | 154 const base::TimeDelta expected_delay = base::TimeDelta::FromMilliseconds( |
152 new_delay); | 155 new_delay); |
153 EXPECT_TRUE(task_runner_->GetPendingTasks().empty()); | 156 EXPECT_TRUE(task_runner_->GetPendingTasks().empty()); |
154 StatusUploader uploader(&client_, collector_.Pass(), task_runner_); | 157 StatusUploader uploader(&client_, std::move(collector_), task_runner_); |
155 ASSERT_EQ(1U, task_runner_->GetPendingTasks().size()); | 158 ASSERT_EQ(1U, task_runner_->GetPendingTasks().size()); |
156 // On startup, first update should happen immediately. | 159 // On startup, first update should happen immediately. |
157 EXPECT_EQ(base::TimeDelta(), task_runner_->NextPendingTaskDelay()); | 160 EXPECT_EQ(base::TimeDelta(), task_runner_->NextPendingTaskDelay()); |
158 | 161 |
159 EXPECT_CALL(*mock_collector, GetDeviceStatus(_)).WillRepeatedly(Return(true)); | 162 EXPECT_CALL(*mock_collector, GetDeviceStatus(_)).WillRepeatedly(Return(true)); |
160 EXPECT_CALL(*mock_collector, GetDeviceSessionStatus(_)).WillRepeatedly( | 163 EXPECT_CALL(*mock_collector, GetDeviceSessionStatus(_)).WillRepeatedly( |
161 Return(true)); | 164 Return(true)); |
162 // Second update should use the delay specified in settings. | 165 // Second update should use the delay specified in settings. |
163 RunPendingUploadTaskAndCheckNext(uploader, expected_delay); | 166 RunPendingUploadTaskAndCheckNext(uploader, expected_delay); |
164 } | 167 } |
165 | 168 |
166 TEST_F(StatusUploaderTest, ResetTimerAfterStatusCollection) { | 169 TEST_F(StatusUploaderTest, ResetTimerAfterStatusCollection) { |
167 // Keep a pointer to the mock collector because collector_ gets cleared | 170 // Keep a pointer to the mock collector because collector_ gets cleared |
168 // when it is passed to the StatusUploader constructor below. | 171 // when it is passed to the StatusUploader constructor below. |
169 MockDeviceStatusCollector* const mock_collector = collector_.get(); | 172 MockDeviceStatusCollector* const mock_collector = collector_.get(); |
170 StatusUploader uploader(&client_, collector_.Pass(), task_runner_); | 173 StatusUploader uploader(&client_, std::move(collector_), task_runner_); |
171 EXPECT_CALL(*mock_collector, GetDeviceStatus(_)).WillRepeatedly(Return(true)); | 174 EXPECT_CALL(*mock_collector, GetDeviceStatus(_)).WillRepeatedly(Return(true)); |
172 EXPECT_CALL(*mock_collector, GetDeviceSessionStatus(_)).WillRepeatedly( | 175 EXPECT_CALL(*mock_collector, GetDeviceSessionStatus(_)).WillRepeatedly( |
173 Return(true)); | 176 Return(true)); |
174 const base::TimeDelta expected_delay = base::TimeDelta::FromMilliseconds( | 177 const base::TimeDelta expected_delay = base::TimeDelta::FromMilliseconds( |
175 StatusUploader::kDefaultUploadDelayMs); | 178 StatusUploader::kDefaultUploadDelayMs); |
176 RunPendingUploadTaskAndCheckNext(uploader, expected_delay); | 179 RunPendingUploadTaskAndCheckNext(uploader, expected_delay); |
177 | 180 |
178 // Handle this response also, and ensure new task is queued. | 181 // Handle this response also, and ensure new task is queued. |
179 RunPendingUploadTaskAndCheckNext(uploader, expected_delay); | 182 RunPendingUploadTaskAndCheckNext(uploader, expected_delay); |
180 | 183 |
181 // Now that the previous request was satisfied, a task to do the next | 184 // Now that the previous request was satisfied, a task to do the next |
182 // upload should be queued again. | 185 // upload should be queued again. |
183 EXPECT_EQ(1U, task_runner_->GetPendingTasks().size()); | 186 EXPECT_EQ(1U, task_runner_->GetPendingTasks().size()); |
184 } | 187 } |
185 | 188 |
186 TEST_F(StatusUploaderTest, ResetTimerAfterFailedStatusCollection) { | 189 TEST_F(StatusUploaderTest, ResetTimerAfterFailedStatusCollection) { |
187 // Keep a pointer to the mock collector because collector_ gets cleared | 190 // Keep a pointer to the mock collector because collector_ gets cleared |
188 // when it is passed to the StatusUploader constructor below. | 191 // when it is passed to the StatusUploader constructor below. |
189 MockDeviceStatusCollector* mock_collector = collector_.get(); | 192 MockDeviceStatusCollector* mock_collector = collector_.get(); |
190 StatusUploader uploader(&client_, collector_.Pass(), task_runner_); | 193 StatusUploader uploader(&client_, std::move(collector_), task_runner_); |
191 EXPECT_CALL(*mock_collector, GetDeviceStatus(_)).WillOnce(Return(false)); | 194 EXPECT_CALL(*mock_collector, GetDeviceStatus(_)).WillOnce(Return(false)); |
192 EXPECT_CALL(*mock_collector, GetDeviceSessionStatus(_)).WillOnce( | 195 EXPECT_CALL(*mock_collector, GetDeviceSessionStatus(_)).WillOnce( |
193 Return(false)); | 196 Return(false)); |
194 task_runner_->RunPendingTasks(); | 197 task_runner_->RunPendingTasks(); |
195 | 198 |
196 // Make sure the next status upload is queued up. | 199 // Make sure the next status upload is queued up. |
197 EXPECT_EQ(1U, task_runner_->GetPendingTasks().size()); | 200 EXPECT_EQ(1U, task_runner_->GetPendingTasks().size()); |
198 const base::TimeDelta expected_delay = base::TimeDelta::FromMilliseconds( | 201 const base::TimeDelta expected_delay = base::TimeDelta::FromMilliseconds( |
199 StatusUploader::kDefaultUploadDelayMs); | 202 StatusUploader::kDefaultUploadDelayMs); |
200 CheckPendingTaskDelay(uploader, expected_delay); | 203 CheckPendingTaskDelay(uploader, expected_delay); |
201 } | 204 } |
202 | 205 |
203 TEST_F(StatusUploaderTest, ChangeFrequency) { | 206 TEST_F(StatusUploaderTest, ChangeFrequency) { |
204 // Keep a pointer to the mock collector because collector_ gets cleared | 207 // Keep a pointer to the mock collector because collector_ gets cleared |
205 // when it is passed to the StatusUploader constructor below. | 208 // when it is passed to the StatusUploader constructor below. |
206 MockDeviceStatusCollector* const mock_collector = collector_.get(); | 209 MockDeviceStatusCollector* const mock_collector = collector_.get(); |
207 StatusUploader uploader(&client_, collector_.Pass(), task_runner_); | 210 StatusUploader uploader(&client_, std::move(collector_), task_runner_); |
208 EXPECT_CALL(*mock_collector, GetDeviceStatus(_)).WillRepeatedly(Return(true)); | 211 EXPECT_CALL(*mock_collector, GetDeviceStatus(_)).WillRepeatedly(Return(true)); |
209 EXPECT_CALL(*mock_collector, GetDeviceSessionStatus(_)).WillRepeatedly( | 212 EXPECT_CALL(*mock_collector, GetDeviceSessionStatus(_)).WillRepeatedly( |
210 Return(true)); | 213 Return(true)); |
211 // Change the frequency. The new frequency should be reflected in the timing | 214 // Change the frequency. The new frequency should be reflected in the timing |
212 // used for the next callback. | 215 // used for the next callback. |
213 const int new_delay = StatusUploader::kDefaultUploadDelayMs * 2; | 216 const int new_delay = StatusUploader::kDefaultUploadDelayMs * 2; |
214 settings_helper_.SetInteger(chromeos::kReportUploadFrequency, new_delay); | 217 settings_helper_.SetInteger(chromeos::kReportUploadFrequency, new_delay); |
215 const base::TimeDelta expected_delay = base::TimeDelta::FromMilliseconds( | 218 const base::TimeDelta expected_delay = base::TimeDelta::FromMilliseconds( |
216 new_delay); | 219 new_delay); |
217 RunPendingUploadTaskAndCheckNext(uploader, expected_delay); | 220 RunPendingUploadTaskAndCheckNext(uploader, expected_delay); |
218 } | 221 } |
219 | 222 |
220 #if defined(USE_X11) || defined(USE_OZONE) | 223 #if defined(USE_X11) || defined(USE_OZONE) |
221 TEST_F(StatusUploaderTest, NoUploadAfterUserInput) { | 224 TEST_F(StatusUploaderTest, NoUploadAfterUserInput) { |
222 StatusUploader uploader(&client_, collector_.Pass(), task_runner_); | 225 StatusUploader uploader(&client_, std::move(collector_), task_runner_); |
223 // Should allow data upload before there is user input. | 226 // Should allow data upload before there is user input. |
224 EXPECT_TRUE(uploader.IsSessionDataUploadAllowed()); | 227 EXPECT_TRUE(uploader.IsSessionDataUploadAllowed()); |
225 | 228 |
226 // Now mock user input, and no session data should be allowed. | 229 // Now mock user input, and no session data should be allowed. |
227 #if defined(USE_X11) | 230 #if defined(USE_X11) |
228 ui::ScopedXI2Event native_event; | 231 ui::ScopedXI2Event native_event; |
229 const int kPointerDeviceId = 10; | 232 const int kPointerDeviceId = 10; |
230 std::vector<int> device_list; | 233 std::vector<int> device_list; |
231 device_list.push_back(kPointerDeviceId); | 234 device_list.push_back(kPointerDeviceId); |
232 ui::TouchFactory::GetInstance()->SetPointerDeviceForTest(device_list); | 235 ui::TouchFactory::GetInstance()->SetPointerDeviceForTest(device_list); |
233 native_event.InitGenericButtonEvent( | 236 native_event.InitGenericButtonEvent( |
234 kPointerDeviceId, ui::ET_MOUSE_PRESSED, gfx::Point(), | 237 kPointerDeviceId, ui::ET_MOUSE_PRESSED, gfx::Point(), |
235 ui::EF_LEFT_MOUSE_BUTTON | ui::EF_CONTROL_DOWN); | 238 ui::EF_LEFT_MOUSE_BUTTON | ui::EF_CONTROL_DOWN); |
236 #elif defined(USE_OZONE) | 239 #elif defined(USE_OZONE) |
237 ui::MouseEvent e(ui::ET_MOUSE_PRESSED, gfx::Point(), gfx::Point(), | 240 ui::MouseEvent e(ui::ET_MOUSE_PRESSED, gfx::Point(), gfx::Point(), |
238 ui::EventTimeForNow(), 0, 0); | 241 ui::EventTimeForNow(), 0, 0); |
239 const ui::PlatformEvent& native_event = &e; | 242 const ui::PlatformEvent& native_event = &e; |
240 #endif | 243 #endif |
241 ui::UserActivityDetector::Get()->DidProcessEvent(native_event); | 244 ui::UserActivityDetector::Get()->DidProcessEvent(native_event); |
242 EXPECT_FALSE(uploader.IsSessionDataUploadAllowed()); | 245 EXPECT_FALSE(uploader.IsSessionDataUploadAllowed()); |
243 } | 246 } |
244 #endif | 247 #endif |
245 | 248 |
246 TEST_F(StatusUploaderTest, NoUploadAfterVideoCapture) { | 249 TEST_F(StatusUploaderTest, NoUploadAfterVideoCapture) { |
247 StatusUploader uploader(&client_, collector_.Pass(), task_runner_); | 250 StatusUploader uploader(&client_, std::move(collector_), task_runner_); |
248 // Should allow data upload before there is video capture. | 251 // Should allow data upload before there is video capture. |
249 EXPECT_TRUE(uploader.IsSessionDataUploadAllowed()); | 252 EXPECT_TRUE(uploader.IsSessionDataUploadAllowed()); |
250 | 253 |
251 // Now mock video capture, and no session data should be allowed. | 254 // Now mock video capture, and no session data should be allowed. |
252 MediaCaptureDevicesDispatcher::GetInstance()->OnMediaRequestStateChanged( | 255 MediaCaptureDevicesDispatcher::GetInstance()->OnMediaRequestStateChanged( |
253 0, 0, 0, GURL("http://www.google.com"), | 256 0, 0, 0, GURL("http://www.google.com"), |
254 content::MEDIA_DEVICE_VIDEO_CAPTURE, | 257 content::MEDIA_DEVICE_VIDEO_CAPTURE, |
255 content::MEDIA_REQUEST_STATE_OPENING); | 258 content::MEDIA_REQUEST_STATE_OPENING); |
256 base::RunLoop().RunUntilIdle(); | 259 base::RunLoop().RunUntilIdle(); |
257 EXPECT_FALSE(uploader.IsSessionDataUploadAllowed()); | 260 EXPECT_FALSE(uploader.IsSessionDataUploadAllowed()); |
258 } | 261 } |
259 | 262 |
260 } // namespace policy | 263 } // namespace policy |
OLD | NEW |