| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 // A collection of classes that are useful when testing things that use a | 5 // A collection of classes that are useful when testing things that use a |
| 6 // GaiaAuthenticator2. | 6 // GaiaAuthenticator2. |
| 7 | 7 |
| 8 #ifndef CHROME_COMMON_NET_GAIA_GAIA_AUTHENTICATOR2_UNITTEST_H_ | 8 #ifndef CHROME_COMMON_NET_GAIA_GAIA_AUTHENTICATOR2_UNITTEST_H_ |
| 9 #define CHROME_COMMON_NET_GAIA_GAIA_AUTHENTICATOR2_UNITTEST_H_ | 9 #define CHROME_COMMON_NET_GAIA_GAIA_AUTHENTICATOR2_UNITTEST_H_ |
| 10 | 10 |
| 11 #include <string> |
| 12 |
| 11 #include "chrome/common/net/gaia/gaia_authenticator2.h" | 13 #include "chrome/common/net/gaia/gaia_authenticator2.h" |
| 12 #include "chrome/common/net/url_fetcher.h" | 14 #include "chrome/common/net/url_fetcher.h" |
| 13 #include "chrome/common/net/http_return.h" | 15 #include "chrome/common/net/http_return.h" |
| 14 #include "net/url_request/url_request_status.h" | 16 #include "net/url_request/url_request_status.h" |
| 15 | 17 |
| 16 // Responds as though ClientLogin returned from the server. | 18 // Responds as though ClientLogin returned from the server. |
| 17 class MockFetcher : public URLFetcher { | 19 class MockFetcher : public URLFetcher { |
| 18 public: | 20 public: |
| 19 MockFetcher(bool success, | 21 MockFetcher(bool success, |
| 20 const GURL& url, | 22 const GURL& url, |
| 23 const std::string & results, |
| 21 URLFetcher::RequestType request_type, | 24 URLFetcher::RequestType request_type, |
| 22 URLFetcher::Delegate* d) | 25 URLFetcher::Delegate* d) |
| 23 : URLFetcher(url, request_type, d), | 26 : URLFetcher(url, request_type, d), |
| 24 success_(success), | 27 success_(success), |
| 25 url_(url) {} | 28 url_(url), |
| 29 results_(results) {} |
| 26 ~MockFetcher() {} | 30 ~MockFetcher() {} |
| 27 void Start() { | 31 void Start() { |
| 28 URLRequestStatus::Status code; | 32 URLRequestStatus::Status code; |
| 29 int http_code; | 33 int http_code; |
| 30 if (success_) { | 34 if (success_) { |
| 31 http_code = RC_REQUEST_OK; | 35 http_code = RC_REQUEST_OK; |
| 32 code = URLRequestStatus::SUCCESS; | 36 code = URLRequestStatus::SUCCESS; |
| 33 } else { | 37 } else { |
| 34 http_code = RC_FORBIDDEN; | 38 http_code = RC_FORBIDDEN; |
| 35 code = URLRequestStatus::FAILED; | 39 code = URLRequestStatus::FAILED; |
| 36 } | 40 } |
| 37 | 41 |
| 38 URLRequestStatus status(code, 0); | 42 URLRequestStatus status(code, 0); |
| 39 delegate()->OnURLFetchComplete(NULL, | 43 delegate()->OnURLFetchComplete(NULL, |
| 40 url_, | 44 url_, |
| 41 status, | 45 status, |
| 42 http_code, | 46 http_code, |
| 43 ResponseCookies(), | 47 ResponseCookies(), |
| 44 std::string()); | 48 results_); |
| 45 } | 49 } |
| 46 private: | 50 private: |
| 47 bool success_; | 51 bool success_; |
| 48 GURL url_; | 52 GURL url_; |
| 53 std::string results_; |
| 49 DISALLOW_COPY_AND_ASSIGN(MockFetcher); | 54 DISALLOW_COPY_AND_ASSIGN(MockFetcher); |
| 50 }; | 55 }; |
| 51 | 56 |
| 52 class MockFactory : public URLFetcher::Factory { | 57 class MockFactory : public URLFetcher::Factory { |
| 53 public: | 58 public: |
| 54 MockFactory() | 59 MockFactory() |
| 55 : success_(true) {} | 60 : success_(true) {} |
| 56 ~MockFactory() {} | 61 ~MockFactory() {} |
| 57 URLFetcher* CreateURLFetcher(int id, | 62 URLFetcher* CreateURLFetcher(int id, |
| 58 const GURL& url, | 63 const GURL& url, |
| 59 URLFetcher::RequestType request_type, | 64 URLFetcher::RequestType request_type, |
| 60 URLFetcher::Delegate* d) { | 65 URLFetcher::Delegate* d) { |
| 61 return new MockFetcher(success_, url, request_type, d); | 66 return new MockFetcher(success_, url, results_, request_type, d); |
| 62 } | 67 } |
| 63 void set_success(bool success) { | 68 void set_success(bool success) { |
| 64 success_ = success; | 69 success_ = success; |
| 65 } | 70 } |
| 71 void set_results(const std::string& results) { |
| 72 results_ = results; |
| 73 } |
| 66 private: | 74 private: |
| 67 bool success_; | 75 bool success_; |
| 76 std::string results_; |
| 68 DISALLOW_COPY_AND_ASSIGN(MockFactory); | 77 DISALLOW_COPY_AND_ASSIGN(MockFactory); |
| 69 }; | 78 }; |
| 70 | 79 |
| 71 #endif // CHROME_COMMON_NET_GAIA_GAIA_AUTHENTICATOR2_UNITTEST_H_ | 80 #endif // CHROME_COMMON_NET_GAIA_GAIA_AUTHENTICATOR2_UNITTEST_H_ |
| OLD | NEW |