| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "google_apis/gaia/oauth2_token_service_request.h" | 5 #include "google_apis/gaia/oauth2_token_service_request.h" |
| 6 | 6 |
| 7 #include <set> | 7 #include <set> |
| 8 #include <string> | 8 #include <string> |
| 9 #include <vector> | 9 #include <vector> |
| 10 #include "base/threading/thread.h" | 10 #include "base/threading/thread.h" |
| 11 #include "google_apis/gaia/fake_oauth2_token_service.h" | 11 #include "google_apis/gaia/fake_oauth2_token_service.h" |
| 12 #include "google_apis/gaia/google_service_auth_error.h" | 12 #include "google_apis/gaia/google_service_auth_error.h" |
| 13 #include "google_apis/gaia/oauth2_token_service.h" | 13 #include "google_apis/gaia/oauth2_token_service.h" |
| 14 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
| 15 | 15 |
| 16 namespace { | 16 namespace { |
| 17 | 17 |
| 18 const char kAccessToken[] = "access_token"; | 18 const char kAccessToken[] = "access_token"; |
| 19 const char kAccountId[] = "test_user@gmail.com"; | 19 const char kAccountId[] = "test_user@gmail.com"; |
| 20 const char kScope[] = "SCOPE"; | 20 const char kScope[] = "SCOPE"; |
| 21 | 21 |
| 22 class TestingOAuth2TokenServiceConsumer : public OAuth2TokenService::Consumer { | 22 class TestingOAuth2TokenServiceConsumer : public OAuth2TokenService::Consumer { |
| 23 public: | 23 public: |
| 24 TestingOAuth2TokenServiceConsumer(); | 24 TestingOAuth2TokenServiceConsumer(); |
| 25 virtual ~TestingOAuth2TokenServiceConsumer(); | 25 ~TestingOAuth2TokenServiceConsumer() override; |
| 26 | 26 |
| 27 virtual void OnGetTokenSuccess(const OAuth2TokenService::Request* request, | 27 void OnGetTokenSuccess(const OAuth2TokenService::Request* request, |
| 28 const std::string& access_token, | 28 const std::string& access_token, |
| 29 const base::Time& expiration_time) override; | 29 const base::Time& expiration_time) override; |
| 30 virtual void OnGetTokenFailure(const OAuth2TokenService::Request* request, | 30 void OnGetTokenFailure(const OAuth2TokenService::Request* request, |
| 31 const GoogleServiceAuthError& error) override; | 31 const GoogleServiceAuthError& error) override; |
| 32 | 32 |
| 33 int num_get_token_success_; | 33 int num_get_token_success_; |
| 34 int num_get_token_failure_; | 34 int num_get_token_failure_; |
| 35 std::string last_token_; | 35 std::string last_token_; |
| 36 GoogleServiceAuthError last_error_; | 36 GoogleServiceAuthError last_error_; |
| 37 }; | 37 }; |
| 38 | 38 |
| 39 TestingOAuth2TokenServiceConsumer::TestingOAuth2TokenServiceConsumer() | 39 TestingOAuth2TokenServiceConsumer::TestingOAuth2TokenServiceConsumer() |
| 40 : OAuth2TokenService::Consumer("test"), | 40 : OAuth2TokenService::Consumer("test"), |
| 41 num_get_token_success_(0), | 41 num_get_token_success_(0), |
| (...skipping 18 matching lines...) Expand all Loading... |
| 60 last_error_ = error; | 60 last_error_ = error; |
| 61 ++num_get_token_failure_; | 61 ++num_get_token_failure_; |
| 62 } | 62 } |
| 63 | 63 |
| 64 // A mock implementation of an OAuth2TokenService. | 64 // A mock implementation of an OAuth2TokenService. |
| 65 // | 65 // |
| 66 // Use SetResponse to vary the response to token requests. | 66 // Use SetResponse to vary the response to token requests. |
| 67 class MockOAuth2TokenService : public FakeOAuth2TokenService { | 67 class MockOAuth2TokenService : public FakeOAuth2TokenService { |
| 68 public: | 68 public: |
| 69 MockOAuth2TokenService(); | 69 MockOAuth2TokenService(); |
| 70 virtual ~MockOAuth2TokenService(); | 70 ~MockOAuth2TokenService() override; |
| 71 | 71 |
| 72 void SetResponse(const GoogleServiceAuthError& error, | 72 void SetResponse(const GoogleServiceAuthError& error, |
| 73 const std::string& access_token, | 73 const std::string& access_token, |
| 74 const base::Time& expiration); | 74 const base::Time& expiration); |
| 75 | 75 |
| 76 int num_invalidate_token() const { return num_invalidate_token_; } | 76 int num_invalidate_token() const { return num_invalidate_token_; } |
| 77 | 77 |
| 78 const std::string& last_token_invalidated() const { | 78 const std::string& last_token_invalidated() const { |
| 79 return last_token_invalidated_; | 79 return last_token_invalidated_; |
| 80 } | 80 } |
| 81 | 81 |
| 82 protected: | 82 protected: |
| 83 virtual void FetchOAuth2Token(RequestImpl* request, | 83 void FetchOAuth2Token(RequestImpl* request, |
| 84 const std::string& account_id, | 84 const std::string& account_id, |
| 85 net::URLRequestContextGetter* getter, | 85 net::URLRequestContextGetter* getter, |
| 86 const std::string& client_id, | 86 const std::string& client_id, |
| 87 const std::string& client_secret, | 87 const std::string& client_secret, |
| 88 const ScopeSet& scopes) override; | 88 const ScopeSet& scopes) override; |
| 89 | 89 |
| 90 virtual void InvalidateOAuth2Token(const std::string& account_id, | 90 void InvalidateOAuth2Token(const std::string& account_id, |
| 91 const std::string& client_id, | 91 const std::string& client_id, |
| 92 const ScopeSet& scopes, | 92 const ScopeSet& scopes, |
| 93 const std::string& access_token) override; | 93 const std::string& access_token) override; |
| 94 | 94 |
| 95 private: | 95 private: |
| 96 GoogleServiceAuthError response_error_; | 96 GoogleServiceAuthError response_error_; |
| 97 std::string response_access_token_; | 97 std::string response_access_token_; |
| 98 base::Time response_expiration_; | 98 base::Time response_expiration_; |
| 99 int num_invalidate_token_; | 99 int num_invalidate_token_; |
| 100 std::string last_token_invalidated_; | 100 std::string last_token_invalidated_; |
| 101 }; | 101 }; |
| 102 | 102 |
| 103 MockOAuth2TokenService::MockOAuth2TokenService() | 103 MockOAuth2TokenService::MockOAuth2TokenService() |
| (...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 147 public: | 147 public: |
| 148 virtual void SetUp() override; | 148 virtual void SetUp() override; |
| 149 virtual void TearDown() override; | 149 virtual void TearDown() override; |
| 150 | 150 |
| 151 protected: | 151 protected: |
| 152 class Provider : public OAuth2TokenServiceRequest::TokenServiceProvider { | 152 class Provider : public OAuth2TokenServiceRequest::TokenServiceProvider { |
| 153 public: | 153 public: |
| 154 Provider(const scoped_refptr<base::SingleThreadTaskRunner>& task_runner, | 154 Provider(const scoped_refptr<base::SingleThreadTaskRunner>& task_runner, |
| 155 OAuth2TokenService* token_service); | 155 OAuth2TokenService* token_service); |
| 156 | 156 |
| 157 virtual scoped_refptr<base::SingleThreadTaskRunner> | 157 scoped_refptr<base::SingleThreadTaskRunner> GetTokenServiceTaskRunner() |
| 158 GetTokenServiceTaskRunner() override; | 158 override; |
| 159 virtual OAuth2TokenService* GetTokenService() override; | 159 OAuth2TokenService* GetTokenService() override; |
| 160 | 160 |
| 161 private: | 161 private: |
| 162 virtual ~Provider(); | 162 ~Provider() override; |
| 163 | 163 |
| 164 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; | 164 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
| 165 OAuth2TokenService* token_service_; | 165 OAuth2TokenService* token_service_; |
| 166 }; | 166 }; |
| 167 | 167 |
| 168 base::MessageLoop ui_loop_; | 168 base::MessageLoop ui_loop_; |
| 169 OAuth2TokenService::ScopeSet scopes_; | 169 OAuth2TokenService::ScopeSet scopes_; |
| 170 scoped_ptr<MockOAuth2TokenService> oauth2_service_; | 170 scoped_ptr<MockOAuth2TokenService> oauth2_service_; |
| 171 scoped_refptr<OAuth2TokenServiceRequest::TokenServiceProvider> provider_; | 171 scoped_refptr<OAuth2TokenServiceRequest::TokenServiceProvider> provider_; |
| 172 TestingOAuth2TokenServiceConsumer consumer_; | 172 TestingOAuth2TokenServiceConsumer consumer_; |
| (...skipping 86 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 259 OAuth2TokenServiceRequest::InvalidateToken( | 259 OAuth2TokenServiceRequest::InvalidateToken( |
| 260 provider_.get(), kAccountId, scopes_, kAccessToken); | 260 provider_.get(), kAccountId, scopes_, kAccessToken); |
| 261 ui_loop_.RunUntilIdle(); | 261 ui_loop_.RunUntilIdle(); |
| 262 EXPECT_EQ(0, consumer_.num_get_token_success_); | 262 EXPECT_EQ(0, consumer_.num_get_token_success_); |
| 263 EXPECT_EQ(0, consumer_.num_get_token_failure_); | 263 EXPECT_EQ(0, consumer_.num_get_token_failure_); |
| 264 EXPECT_EQ(kAccessToken, oauth2_service_->last_token_invalidated()); | 264 EXPECT_EQ(kAccessToken, oauth2_service_->last_token_invalidated()); |
| 265 EXPECT_EQ(1, oauth2_service_->num_invalidate_token()); | 265 EXPECT_EQ(1, oauth2_service_->num_invalidate_token()); |
| 266 } | 266 } |
| 267 | 267 |
| 268 } // namespace | 268 } // namespace |
| OLD | NEW |