| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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/auto_enrollment_client.h" | 5 #include "chrome/browser/chromeos/policy/auto_enrollment_client.h" |
| 6 | 6 |
| 7 #include <stdint.h> | 7 #include <stdint.h> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/bind_helpers.h" | 10 #include "base/bind_helpers.h" |
| 11 #include "base/macros.h" |
| 11 #include "base/message_loop/message_loop.h" | 12 #include "base/message_loop/message_loop.h" |
| 12 #include "base/prefs/pref_service.h" | 13 #include "base/prefs/pref_service.h" |
| 13 #include "base/prefs/testing_pref_service.h" | 14 #include "base/prefs/testing_pref_service.h" |
| 14 #include "base/run_loop.h" | 15 #include "base/run_loop.h" |
| 15 #include "base/strings/stringprintf.h" | 16 #include "base/strings/stringprintf.h" |
| 16 #include "base/values.h" | 17 #include "base/values.h" |
| 17 #include "chrome/browser/browser_process.h" | 18 #include "chrome/browser/browser_process.h" |
| 18 #include "chrome/browser/chromeos/policy/server_backed_device_state.h" | 19 #include "chrome/browser/chromeos/policy/server_backed_device_state.h" |
| 19 #include "chrome/common/pref_names.h" | 20 #include "chrome/common/pref_names.h" |
| 20 #include "chrome/test/base/scoped_testing_local_state.h" | 21 #include "chrome/test/base/scoped_testing_local_state.h" |
| (...skipping 62 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 83 state_ = state; | 84 state_ = state; |
| 84 } | 85 } |
| 85 | 86 |
| 86 void ServerWillFail(DeviceManagementStatus error) { | 87 void ServerWillFail(DeviceManagementStatus error) { |
| 87 em::DeviceManagementResponse dummy_response; | 88 em::DeviceManagementResponse dummy_response; |
| 88 EXPECT_CALL(*service_, | 89 EXPECT_CALL(*service_, |
| 89 CreateJob(DeviceManagementRequestJob::TYPE_AUTO_ENROLLMENT, _)) | 90 CreateJob(DeviceManagementRequestJob::TYPE_AUTO_ENROLLMENT, _)) |
| 90 .WillOnce(service_->FailJob(error)); | 91 .WillOnce(service_->FailJob(error)); |
| 91 } | 92 } |
| 92 | 93 |
| 93 void ServerWillReply(int64 modulus, bool with_hashes, bool with_id_hash) { | 94 void ServerWillReply(int64_t modulus, bool with_hashes, bool with_id_hash) { |
| 94 em::DeviceManagementResponse response; | 95 em::DeviceManagementResponse response; |
| 95 em::DeviceAutoEnrollmentResponse* enrollment_response = | 96 em::DeviceAutoEnrollmentResponse* enrollment_response = |
| 96 response.mutable_auto_enrollment_response(); | 97 response.mutable_auto_enrollment_response(); |
| 97 if (modulus >= 0) | 98 if (modulus >= 0) |
| 98 enrollment_response->set_expected_modulus(modulus); | 99 enrollment_response->set_expected_modulus(modulus); |
| 99 if (with_hashes) { | 100 if (with_hashes) { |
| 100 for (int i = 0; i < 10; ++i) { | 101 for (int i = 0; i < 10; ++i) { |
| 101 std::string state_key = base::StringPrintf("state_key %d", i); | 102 std::string state_key = base::StringPrintf("state_key %d", i); |
| 102 std::string hash = crypto::SHA256HashString(state_key); | 103 std::string hash = crypto::SHA256HashString(state_key); |
| 103 enrollment_response->mutable_hash()->Add()->assign(hash); | 104 enrollment_response->mutable_hash()->Add()->assign(hash); |
| (...skipping 276 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 380 EXPECT_EQ(AUTO_ENROLLMENT_STATE_NO_ENROLLMENT, state_); | 381 EXPECT_EQ(AUTO_ENROLLMENT_STATE_NO_ENROLLMENT, state_); |
| 381 EXPECT_TRUE(auto_enrollment_request().has_remainder()); | 382 EXPECT_TRUE(auto_enrollment_request().has_remainder()); |
| 382 EXPECT_TRUE(auto_enrollment_request().has_modulus()); | 383 EXPECT_TRUE(auto_enrollment_request().has_modulus()); |
| 383 EXPECT_EQ(1, auto_enrollment_request().modulus()); | 384 EXPECT_EQ(1, auto_enrollment_request().modulus()); |
| 384 EXPECT_EQ(0, auto_enrollment_request().remainder()); | 385 EXPECT_EQ(0, auto_enrollment_request().remainder()); |
| 385 VerifyCachedResult(false, 0); | 386 VerifyCachedResult(false, 0); |
| 386 EXPECT_FALSE(HasServerBackedState()); | 387 EXPECT_FALSE(HasServerBackedState()); |
| 387 } | 388 } |
| 388 | 389 |
| 389 TEST_F(AutoEnrollmentClientTest, ManyBitsUploaded) { | 390 TEST_F(AutoEnrollmentClientTest, ManyBitsUploaded) { |
| 390 int64 bottom62 = INT64_C(0x386e7244d097c3e6); | 391 int64_t bottom62 = INT64_C(0x386e7244d097c3e6); |
| 391 for (int i = 0; i <= 62; ++i) { | 392 for (int i = 0; i <= 62; ++i) { |
| 392 CreateClient(kStateKey, i, i); | 393 CreateClient(kStateKey, i, i); |
| 393 ServerWillReply(-1, false, false); | 394 ServerWillReply(-1, false, false); |
| 394 client_->Start(); | 395 client_->Start(); |
| 395 EXPECT_EQ(AUTO_ENROLLMENT_STATE_NO_ENROLLMENT, state_); | 396 EXPECT_EQ(AUTO_ENROLLMENT_STATE_NO_ENROLLMENT, state_); |
| 396 EXPECT_TRUE(auto_enrollment_request().has_remainder()); | 397 EXPECT_TRUE(auto_enrollment_request().has_remainder()); |
| 397 EXPECT_TRUE(auto_enrollment_request().has_modulus()); | 398 EXPECT_TRUE(auto_enrollment_request().has_modulus()); |
| 398 EXPECT_EQ(INT64_C(1) << i, auto_enrollment_request().modulus()); | 399 EXPECT_EQ(INT64_C(1) << i, auto_enrollment_request().modulus()); |
| 399 EXPECT_EQ(bottom62 % (INT64_C(1) << i), | 400 EXPECT_EQ(bottom62 % (INT64_C(1) << i), |
| 400 auto_enrollment_request().remainder()); | 401 auto_enrollment_request().remainder()); |
| (...skipping 213 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 614 EXPECT_EQ(AUTO_ENROLLMENT_STATE_TRIGGER_ENROLLMENT, state_); | 615 EXPECT_EQ(AUTO_ENROLLMENT_STATE_TRIGGER_ENROLLMENT, state_); |
| 615 EXPECT_TRUE(HasCachedDecision()); | 616 EXPECT_TRUE(HasCachedDecision()); |
| 616 VerifyServerBackedState("example.com", | 617 VerifyServerBackedState("example.com", |
| 617 kDeviceStateRestoreModeReEnrollmentEnforced, | 618 kDeviceStateRestoreModeReEnrollmentEnforced, |
| 618 kDisabledMessage); | 619 kDisabledMessage); |
| 619 Mock::VerifyAndClearExpectations(service_.get()); | 620 Mock::VerifyAndClearExpectations(service_.get()); |
| 620 } | 621 } |
| 621 | 622 |
| 622 } // namespace | 623 } // namespace |
| 623 } // namespace policy | 624 } // namespace policy |
| OLD | NEW |