| 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 <ostream> | 5 #include <ostream> |
| 6 #include <vector> | 6 #include <vector> |
| 7 | 7 |
| 8 #include "base/bind.h" | 8 #include "base/bind.h" |
| 9 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
| 10 #include "base/message_loop/message_loop_proxy.h" | 10 #include "base/message_loop/message_loop_proxy.h" |
| (...skipping 59 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 70 void InitializeService() { | 70 void InitializeService() { |
| 71 service_->ScheduleInitialization(0); | 71 service_->ScheduleInitialization(0); |
| 72 base::RunLoop().RunUntilIdle(); | 72 base::RunLoop().RunUntilIdle(); |
| 73 } | 73 } |
| 74 | 74 |
| 75 net::TestURLFetcher* GetFetcher() { | 75 net::TestURLFetcher* GetFetcher() { |
| 76 return factory_.GetFetcherByID(DeviceManagementService::kURLFetcherID); | 76 return factory_.GetFetcherByID(DeviceManagementService::kURLFetcherID); |
| 77 } | 77 } |
| 78 | 78 |
| 79 DeviceManagementRequestJob* StartRegistrationJob() { | 79 DeviceManagementRequestJob* StartRegistrationJob() { |
| 80 DeviceManagementRequestJob* job = | 80 DeviceManagementRequestJob* job = service_->CreateJob( |
| 81 service_->CreateJob(DeviceManagementRequestJob::TYPE_REGISTRATION, | 81 DeviceManagementRequestJob::TYPE_REGISTRATION, request_context_.get()); |
| 82 request_context_); | |
| 83 job->SetGaiaToken(kGaiaAuthToken); | 82 job->SetGaiaToken(kGaiaAuthToken); |
| 84 job->SetOAuthToken(kOAuthToken); | 83 job->SetOAuthToken(kOAuthToken); |
| 85 job->SetClientID(kClientID); | 84 job->SetClientID(kClientID); |
| 86 job->GetRequest()->mutable_register_request(); | 85 job->GetRequest()->mutable_register_request(); |
| 87 job->SetRetryCallback(base::Bind( | 86 job->SetRetryCallback(base::Bind( |
| 88 &DeviceManagementServiceTestBase::OnJobRetry, base::Unretained(this))); | 87 &DeviceManagementServiceTestBase::OnJobRetry, base::Unretained(this))); |
| 89 job->Start(base::Bind(&DeviceManagementServiceTestBase::OnJobDone, | 88 job->Start(base::Bind(&DeviceManagementServiceTestBase::OnJobDone, |
| 90 base::Unretained(this))); | 89 base::Unretained(this))); |
| 91 return job; | 90 return job; |
| 92 } | 91 } |
| 93 | 92 |
| 94 DeviceManagementRequestJob* StartApiAuthCodeFetchJob() { | 93 DeviceManagementRequestJob* StartApiAuthCodeFetchJob() { |
| 95 DeviceManagementRequestJob* job = service_->CreateJob( | 94 DeviceManagementRequestJob* job = service_->CreateJob( |
| 96 DeviceManagementRequestJob::TYPE_API_AUTH_CODE_FETCH, | 95 DeviceManagementRequestJob::TYPE_API_AUTH_CODE_FETCH, |
| 97 request_context_); | 96 request_context_.get()); |
| 98 job->SetGaiaToken(kGaiaAuthToken); | 97 job->SetGaiaToken(kGaiaAuthToken); |
| 99 job->SetOAuthToken(kOAuthToken); | 98 job->SetOAuthToken(kOAuthToken); |
| 100 job->SetClientID(kClientID); | 99 job->SetClientID(kClientID); |
| 101 job->GetRequest()->mutable_service_api_access_request(); | 100 job->GetRequest()->mutable_service_api_access_request(); |
| 102 job->SetRetryCallback(base::Bind( | 101 job->SetRetryCallback(base::Bind( |
| 103 &DeviceManagementServiceTestBase::OnJobRetry, base::Unretained(this))); | 102 &DeviceManagementServiceTestBase::OnJobRetry, base::Unretained(this))); |
| 104 job->Start(base::Bind(&DeviceManagementServiceTestBase::OnJobDone, | 103 job->Start(base::Bind(&DeviceManagementServiceTestBase::OnJobDone, |
| 105 base::Unretained(this))); | 104 base::Unretained(this))); |
| 106 return job; | 105 return job; |
| 107 } | 106 } |
| 108 | 107 |
| 109 DeviceManagementRequestJob* StartUnregistrationJob() { | 108 DeviceManagementRequestJob* StartUnregistrationJob() { |
| 110 DeviceManagementRequestJob* job = | 109 DeviceManagementRequestJob* job = |
| 111 service_->CreateJob(DeviceManagementRequestJob::TYPE_UNREGISTRATION, | 110 service_->CreateJob(DeviceManagementRequestJob::TYPE_UNREGISTRATION, |
| 112 request_context_); | 111 request_context_.get()); |
| 113 job->SetDMToken(kDMToken); | 112 job->SetDMToken(kDMToken); |
| 114 job->SetClientID(kClientID); | 113 job->SetClientID(kClientID); |
| 115 job->GetRequest()->mutable_unregister_request(); | 114 job->GetRequest()->mutable_unregister_request(); |
| 116 job->SetRetryCallback(base::Bind( | 115 job->SetRetryCallback(base::Bind( |
| 117 &DeviceManagementServiceTestBase::OnJobRetry, base::Unretained(this))); | 116 &DeviceManagementServiceTestBase::OnJobRetry, base::Unretained(this))); |
| 118 job->Start(base::Bind(&DeviceManagementServiceTestBase::OnJobDone, | 117 job->Start(base::Bind(&DeviceManagementServiceTestBase::OnJobDone, |
| 119 base::Unretained(this))); | 118 base::Unretained(this))); |
| 120 return job; | 119 return job; |
| 121 } | 120 } |
| 122 | 121 |
| 123 DeviceManagementRequestJob* StartPolicyFetchJob() { | 122 DeviceManagementRequestJob* StartPolicyFetchJob() { |
| 124 DeviceManagementRequestJob* job = | 123 DeviceManagementRequestJob* job = service_->CreateJob( |
| 125 service_->CreateJob(DeviceManagementRequestJob::TYPE_POLICY_FETCH, | 124 DeviceManagementRequestJob::TYPE_POLICY_FETCH, request_context_.get()); |
| 126 request_context_); | |
| 127 job->SetGaiaToken(kGaiaAuthToken); | 125 job->SetGaiaToken(kGaiaAuthToken); |
| 128 job->SetOAuthToken(kOAuthToken); | 126 job->SetOAuthToken(kOAuthToken); |
| 129 job->SetClientID(kClientID); | 127 job->SetClientID(kClientID); |
| 130 em::PolicyFetchRequest* fetch_request = | 128 em::PolicyFetchRequest* fetch_request = |
| 131 job->GetRequest()->mutable_policy_request()->add_request(); | 129 job->GetRequest()->mutable_policy_request()->add_request(); |
| 132 fetch_request->set_policy_type(dm_protocol::kChromeUserPolicyType); | 130 fetch_request->set_policy_type(dm_protocol::kChromeUserPolicyType); |
| 133 job->SetRetryCallback(base::Bind( | 131 job->SetRetryCallback(base::Bind( |
| 134 &DeviceManagementServiceTestBase::OnJobRetry, base::Unretained(this))); | 132 &DeviceManagementServiceTestBase::OnJobRetry, base::Unretained(this))); |
| 135 job->Start(base::Bind(&DeviceManagementServiceTestBase::OnJobDone, | 133 job->Start(base::Bind(&DeviceManagementServiceTestBase::OnJobDone, |
| 136 base::Unretained(this))); | 134 base::Unretained(this))); |
| 137 return job; | 135 return job; |
| 138 } | 136 } |
| 139 | 137 |
| 140 DeviceManagementRequestJob* StartAutoEnrollmentJob() { | 138 DeviceManagementRequestJob* StartAutoEnrollmentJob() { |
| 141 DeviceManagementRequestJob* job = | 139 DeviceManagementRequestJob* job = |
| 142 service_->CreateJob(DeviceManagementRequestJob::TYPE_AUTO_ENROLLMENT, | 140 service_->CreateJob(DeviceManagementRequestJob::TYPE_AUTO_ENROLLMENT, |
| 143 request_context_); | 141 request_context_.get()); |
| 144 job->SetClientID(kClientID); | 142 job->SetClientID(kClientID); |
| 145 em::DeviceAutoEnrollmentRequest* request = | 143 em::DeviceAutoEnrollmentRequest* request = |
| 146 job->GetRequest()->mutable_auto_enrollment_request(); | 144 job->GetRequest()->mutable_auto_enrollment_request(); |
| 147 request->set_modulus(1); | 145 request->set_modulus(1); |
| 148 request->set_remainder(0); | 146 request->set_remainder(0); |
| 149 job->SetRetryCallback(base::Bind( | 147 job->SetRetryCallback(base::Bind( |
| 150 &DeviceManagementServiceTestBase::OnJobRetry, base::Unretained(this))); | 148 &DeviceManagementServiceTestBase::OnJobRetry, base::Unretained(this))); |
| 151 job->Start(base::Bind(&DeviceManagementServiceTestBase::OnJobDone, | 149 job->Start(base::Bind(&DeviceManagementServiceTestBase::OnJobDone, |
| 152 base::Unretained(this))); | 150 base::Unretained(this))); |
| 153 return job; | 151 return job; |
| (...skipping 541 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 695 EXPECT_CALL(*this, OnJobDone(DM_STATUS_REQUEST_FAILED, _, _)); | 693 EXPECT_CALL(*this, OnJobDone(DM_STATUS_REQUEST_FAILED, _, _)); |
| 696 EXPECT_CALL(*this, OnJobRetry(_)).Times(0); | 694 EXPECT_CALL(*this, OnJobRetry(_)).Times(0); |
| 697 fetcher->set_status(net::URLRequestStatus(net::URLRequestStatus::FAILED, | 695 fetcher->set_status(net::URLRequestStatus(net::URLRequestStatus::FAILED, |
| 698 net::ERR_NETWORK_CHANGED)); | 696 net::ERR_NETWORK_CHANGED)); |
| 699 fetcher->set_url(GURL(kServiceUrl)); | 697 fetcher->set_url(GURL(kServiceUrl)); |
| 700 fetcher->delegate()->OnURLFetchComplete(fetcher); | 698 fetcher->delegate()->OnURLFetchComplete(fetcher); |
| 701 Mock::VerifyAndClearExpectations(this); | 699 Mock::VerifyAndClearExpectations(this); |
| 702 } | 700 } |
| 703 | 701 |
| 704 } // namespace policy | 702 } // namespace policy |
| OLD | NEW |