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 |