| OLD | NEW |
| 1 // Copyright 2012 The Chromium Authors. All rights reserved. | 1 // Copyright 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 #include "chrome/browser/signin/profile_oauth2_token_service_request.h" | 4 #include "chrome/browser/signin/profile_oauth2_token_service_request.h" |
| 5 | 5 |
| 6 #include <set> | 6 #include <set> |
| 7 #include <string> | 7 #include <string> |
| 8 #include <vector> | 8 #include <vector> |
| 9 #include "base/threading/thread.h" | 9 #include "base/threading/thread.h" |
| 10 #include "chrome/browser/signin/profile_oauth2_token_service.h" | 10 #include "chrome/browser/signin/profile_oauth2_token_service.h" |
| (...skipping 64 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 75 OAuth2TokenService::Consumer* consumer_; | 75 OAuth2TokenService::Consumer* consumer_; |
| 76 GoogleServiceAuthError error_; | 76 GoogleServiceAuthError error_; |
| 77 std::string access_token_; | 77 std::string access_token_; |
| 78 base::Time expiration_date_; | 78 base::Time expiration_date_; |
| 79 }; | 79 }; |
| 80 | 80 |
| 81 MockProfileOAuth2TokenService(); | 81 MockProfileOAuth2TokenService(); |
| 82 virtual ~MockProfileOAuth2TokenService(); | 82 virtual ~MockProfileOAuth2TokenService(); |
| 83 | 83 |
| 84 virtual scoped_ptr<OAuth2TokenService::Request> StartRequest( | 84 virtual scoped_ptr<OAuth2TokenService::Request> StartRequest( |
| 85 const std::set<std::string>& scopes, | 85 const std::string& account_id, |
| 86 const OAuth2TokenService::ScopeSet& scopes, |
| 86 OAuth2TokenService::Consumer* consumer) OVERRIDE; | 87 OAuth2TokenService::Consumer* consumer) OVERRIDE; |
| 87 | 88 |
| 88 void SetExpectation(bool success, std::string oauth2_access_token); | 89 void SetExpectation(bool success, std::string oauth2_access_token); |
| 89 | 90 |
| 90 private: | 91 private: |
| 91 static void InformConsumer( | 92 static void InformConsumer( |
| 92 base::WeakPtr<MockProfileOAuth2TokenService::Request> request); | 93 base::WeakPtr<MockProfileOAuth2TokenService::Request> request); |
| 93 | 94 |
| 94 bool success_; | 95 bool success_; |
| 95 std::string oauth2_access_token_; | 96 std::string oauth2_access_token_; |
| (...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 130 | 131 |
| 131 // static | 132 // static |
| 132 void MockProfileOAuth2TokenService::InformConsumer( | 133 void MockProfileOAuth2TokenService::InformConsumer( |
| 133 base::WeakPtr<MockProfileOAuth2TokenService::Request> request) { | 134 base::WeakPtr<MockProfileOAuth2TokenService::Request> request) { |
| 134 if (request.get()) | 135 if (request.get()) |
| 135 request->InformConsumer(); | 136 request->InformConsumer(); |
| 136 } | 137 } |
| 137 | 138 |
| 138 scoped_ptr<OAuth2TokenService::Request> | 139 scoped_ptr<OAuth2TokenService::Request> |
| 139 MockProfileOAuth2TokenService::StartRequest( | 140 MockProfileOAuth2TokenService::StartRequest( |
| 140 const std::set<std::string>& scopes, | 141 const std::string& account_id, |
| 142 const OAuth2TokenService::ScopeSet& scopes, |
| 141 OAuth2TokenService::Consumer* consumer) { | 143 OAuth2TokenService::Consumer* consumer) { |
| 142 scoped_ptr<Request> request; | 144 scoped_ptr<Request> request; |
| 143 if (success_) { | 145 if (success_) { |
| 144 request.reset(new MockProfileOAuth2TokenService::Request( | 146 request.reset(new MockProfileOAuth2TokenService::Request( |
| 145 consumer, | 147 consumer, |
| 146 GoogleServiceAuthError(GoogleServiceAuthError::NONE), | 148 GoogleServiceAuthError(GoogleServiceAuthError::NONE), |
| 147 oauth2_access_token_)); | 149 oauth2_access_token_)); |
| 148 } else { | 150 } else { |
| 149 request.reset(new MockProfileOAuth2TokenService::Request( | 151 request.reset(new MockProfileOAuth2TokenService::Request( |
| 150 consumer, | 152 consumer, |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 189 oauth2_service_ = (MockProfileOAuth2TokenService*) | 191 oauth2_service_ = (MockProfileOAuth2TokenService*) |
| 190 ProfileOAuth2TokenServiceFactory::GetForProfile(profile_.get()); | 192 ProfileOAuth2TokenServiceFactory::GetForProfile(profile_.get()); |
| 191 } | 193 } |
| 192 | 194 |
| 193 TEST_F(ProfileOAuth2TokenServiceRequestTest, | 195 TEST_F(ProfileOAuth2TokenServiceRequestTest, |
| 194 Failure) { | 196 Failure) { |
| 195 oauth2_service_->SetExpectation(false, std::string()); | 197 oauth2_service_->SetExpectation(false, std::string()); |
| 196 scoped_ptr<ProfileOAuth2TokenServiceRequest> request( | 198 scoped_ptr<ProfileOAuth2TokenServiceRequest> request( |
| 197 ProfileOAuth2TokenServiceRequest::CreateAndStart( | 199 ProfileOAuth2TokenServiceRequest::CreateAndStart( |
| 198 profile_.get(), | 200 profile_.get(), |
| 199 std::set<std::string>(), | 201 std::string(), |
| 202 OAuth2TokenService::ScopeSet(), |
| 200 &consumer_)); | 203 &consumer_)); |
| 201 ui_loop_.RunUntilIdle(); | 204 ui_loop_.RunUntilIdle(); |
| 202 EXPECT_EQ(0, consumer_.number_of_successful_tokens_); | 205 EXPECT_EQ(0, consumer_.number_of_successful_tokens_); |
| 203 EXPECT_EQ(1, consumer_.number_of_errors_); | 206 EXPECT_EQ(1, consumer_.number_of_errors_); |
| 204 } | 207 } |
| 205 | 208 |
| 206 TEST_F(ProfileOAuth2TokenServiceRequestTest, | 209 TEST_F(ProfileOAuth2TokenServiceRequestTest, |
| 207 Success) { | 210 Success) { |
| 208 scoped_ptr<ProfileOAuth2TokenServiceRequest> request( | 211 scoped_ptr<ProfileOAuth2TokenServiceRequest> request( |
| 209 ProfileOAuth2TokenServiceRequest::CreateAndStart( | 212 ProfileOAuth2TokenServiceRequest::CreateAndStart( |
| 210 profile_.get(), | 213 profile_.get(), |
| 211 std::set<std::string>(), | 214 std::string(), |
| 215 OAuth2TokenService::ScopeSet(), |
| 212 &consumer_)); | 216 &consumer_)); |
| 213 ui_loop_.RunUntilIdle(); | 217 ui_loop_.RunUntilIdle(); |
| 214 EXPECT_EQ(1, consumer_.number_of_successful_tokens_); | 218 EXPECT_EQ(1, consumer_.number_of_successful_tokens_); |
| 215 EXPECT_EQ("success token", consumer_.last_token_); | 219 EXPECT_EQ("success token", consumer_.last_token_); |
| 216 EXPECT_EQ(0, consumer_.number_of_errors_); | 220 EXPECT_EQ(0, consumer_.number_of_errors_); |
| 217 } | 221 } |
| 218 | 222 |
| 219 TEST_F(ProfileOAuth2TokenServiceRequestTest, | 223 TEST_F(ProfileOAuth2TokenServiceRequestTest, |
| 220 RequestDeletionBeforeServiceComplete) { | 224 RequestDeletionBeforeServiceComplete) { |
| 221 scoped_ptr<ProfileOAuth2TokenServiceRequest> request( | 225 scoped_ptr<ProfileOAuth2TokenServiceRequest> request( |
| 222 ProfileOAuth2TokenServiceRequest::CreateAndStart( | 226 ProfileOAuth2TokenServiceRequest::CreateAndStart( |
| 223 profile_.get(), | 227 profile_.get(), |
| 224 std::set<std::string>(), | 228 std::string(), |
| 229 OAuth2TokenService::ScopeSet(), |
| 225 &consumer_)); | 230 &consumer_)); |
| 226 request.reset(); | 231 request.reset(); |
| 227 ui_loop_.RunUntilIdle(); | 232 ui_loop_.RunUntilIdle(); |
| 228 EXPECT_EQ(0, consumer_.number_of_successful_tokens_); | 233 EXPECT_EQ(0, consumer_.number_of_successful_tokens_); |
| 229 EXPECT_EQ(0, consumer_.number_of_errors_); | 234 EXPECT_EQ(0, consumer_.number_of_errors_); |
| 230 } | 235 } |
| 231 | 236 |
| 232 TEST_F(ProfileOAuth2TokenServiceRequestTest, | 237 TEST_F(ProfileOAuth2TokenServiceRequestTest, |
| 233 RequestDeletionAfterServiceComplete) { | 238 RequestDeletionAfterServiceComplete) { |
| 234 scoped_ptr<ProfileOAuth2TokenServiceRequest> request( | 239 scoped_ptr<ProfileOAuth2TokenServiceRequest> request( |
| 235 ProfileOAuth2TokenServiceRequest::CreateAndStart( | 240 ProfileOAuth2TokenServiceRequest::CreateAndStart( |
| 236 profile_.get(), | 241 profile_.get(), |
| 237 std::set<std::string>(), | 242 std::string(), |
| 243 OAuth2TokenService::ScopeSet(), |
| 238 &consumer_)); | 244 &consumer_)); |
| 239 ui_loop_.RunUntilIdle(); | 245 ui_loop_.RunUntilIdle(); |
| 240 request.reset(); | 246 request.reset(); |
| 241 EXPECT_EQ(1, consumer_.number_of_successful_tokens_); | 247 EXPECT_EQ(1, consumer_.number_of_successful_tokens_); |
| 242 EXPECT_EQ(0, consumer_.number_of_errors_); | 248 EXPECT_EQ(0, consumer_.number_of_errors_); |
| 243 } | 249 } |
| 244 | 250 |
| 245 } // namespace | 251 } // namespace |
| OLD | NEW |