| 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 #ifndef CONTENT_TEST_TEST_URL_FETCHER_FACTORY_H_ | 5 #ifndef CONTENT_TEST_TEST_URL_FETCHER_FACTORY_H_ |
| 6 #define CONTENT_TEST_TEST_URL_FETCHER_FACTORY_H_ | 6 #define CONTENT_TEST_TEST_URL_FETCHER_FACTORY_H_ |
| 7 #pragma once | 7 #pragma once |
| 8 | 8 |
| 9 #include <list> | 9 #include <list> |
| 10 #include <map> | 10 #include <map> |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 51 // ... | 51 // ... |
| 52 // | 52 // |
| 53 // Note: if you don't know when your request objects will be created you | 53 // Note: if you don't know when your request objects will be created you |
| 54 // might want to use the FakeURLFetcher and FakeURLFetcherFactory classes | 54 // might want to use the FakeURLFetcher and FakeURLFetcherFactory classes |
| 55 // below. | 55 // below. |
| 56 | 56 |
| 57 class TestURLFetcher : public content::URLFetcher { | 57 class TestURLFetcher : public content::URLFetcher { |
| 58 public: | 58 public: |
| 59 TestURLFetcher(int id, | 59 TestURLFetcher(int id, |
| 60 const GURL& url, | 60 const GURL& url, |
| 61 content::URLFetcherDelegate* d); | 61 net::URLFetcherDelegate* d); |
| 62 virtual ~TestURLFetcher(); | 62 virtual ~TestURLFetcher(); |
| 63 | 63 |
| 64 // content::URLFetcher implementation | 64 // content::URLFetcher implementation |
| 65 virtual void SetUploadData(const std::string& upload_content_type, | 65 virtual void SetUploadData(const std::string& upload_content_type, |
| 66 const std::string& upload_content) OVERRIDE; | 66 const std::string& upload_content) OVERRIDE; |
| 67 virtual void SetChunkedUpload( | 67 virtual void SetChunkedUpload( |
| 68 const std::string& upload_content_type) OVERRIDE; | 68 const std::string& upload_content_type) OVERRIDE; |
| 69 // Overriden to cache the chunks uploaded. Caller can read back the uploaded | 69 // Overriden to cache the chunks uploaded. Caller can read back the uploaded |
| 70 // chunks with the upload_chunks() accessor. | 70 // chunks with the upload_chunks() accessor. |
| 71 virtual void AppendChunkToUpload(const std::string& data, | 71 virtual void AppendChunkToUpload(const std::string& data, |
| (...skipping 47 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 119 // Unique ID in our factory. | 119 // Unique ID in our factory. |
| 120 int id() const { return id_; } | 120 int id() const { return id_; } |
| 121 | 121 |
| 122 // Returns the data uploaded on this URLFetcher. | 122 // Returns the data uploaded on this URLFetcher. |
| 123 const std::string& upload_data() const { return upload_data_; } | 123 const std::string& upload_data() const { return upload_data_; } |
| 124 | 124 |
| 125 // Returns the chunks of data uploaded on this URLFetcher. | 125 // Returns the chunks of data uploaded on this URLFetcher. |
| 126 const std::list<std::string>& upload_chunks() const { return chunks_; } | 126 const std::list<std::string>& upload_chunks() const { return chunks_; } |
| 127 | 127 |
| 128 // Returns the delegate installed on the URLFetcher. | 128 // Returns the delegate installed on the URLFetcher. |
| 129 content::URLFetcherDelegate* delegate() const { return delegate_; } | 129 net::URLFetcherDelegate* delegate() const { return delegate_; } |
| 130 | 130 |
| 131 void set_url(const GURL& url) { fake_url_ = url; } | 131 void set_url(const GURL& url) { fake_url_ = url; } |
| 132 void set_status(const net::URLRequestStatus& status); | 132 void set_status(const net::URLRequestStatus& status); |
| 133 void set_response_code(int response_code) { | 133 void set_response_code(int response_code) { |
| 134 fake_response_code_ = response_code; | 134 fake_response_code_ = response_code; |
| 135 } | 135 } |
| 136 void set_cookies(const net::ResponseCookies& c) { fake_cookies_ = c; } | 136 void set_cookies(const net::ResponseCookies& c) { fake_cookies_ = c; } |
| 137 void set_was_fetched_via_proxy(bool flag); | 137 void set_was_fetched_via_proxy(bool flag); |
| 138 void set_response_headers(scoped_refptr<net::HttpResponseHeaders> headers); | 138 void set_response_headers(scoped_refptr<net::HttpResponseHeaders> headers); |
| 139 void set_backoff_delay(base::TimeDelta backoff_delay); | 139 void set_backoff_delay(base::TimeDelta backoff_delay); |
| 140 | 140 |
| 141 // Set string data. | 141 // Set string data. |
| 142 void SetResponseString(const std::string& response); | 142 void SetResponseString(const std::string& response); |
| 143 | 143 |
| 144 // Set File data. | 144 // Set File data. |
| 145 void SetResponseFilePath(const FilePath& path); | 145 void SetResponseFilePath(const FilePath& path); |
| 146 | 146 |
| 147 private: | 147 private: |
| 148 enum ResponseDestinationType { | 148 enum ResponseDestinationType { |
| 149 STRING, // Default: In a std::string | 149 STRING, // Default: In a std::string |
| 150 TEMP_FILE // Write to a temp file | 150 TEMP_FILE // Write to a temp file |
| 151 }; | 151 }; |
| 152 | 152 |
| 153 const int id_; | 153 const int id_; |
| 154 const GURL original_url_; | 154 const GURL original_url_; |
| 155 content::URLFetcherDelegate* delegate_; | 155 net::URLFetcherDelegate* delegate_; |
| 156 std::string upload_data_; | 156 std::string upload_data_; |
| 157 std::list<std::string> chunks_; | 157 std::list<std::string> chunks_; |
| 158 bool did_receive_last_chunk_; | 158 bool did_receive_last_chunk_; |
| 159 | 159 |
| 160 // User can use set_* methods to provide values returned by getters. | 160 // User can use set_* methods to provide values returned by getters. |
| 161 // Setting the real values is not possible, because the real class | 161 // Setting the real values is not possible, because the real class |
| 162 // has no setters. The data is a private member of a class defined | 162 // has no setters. The data is a private member of a class defined |
| 163 // in a .cc file, so we can't get at it with friendship. | 163 // in a .cc file, so we can't get at it with friendship. |
| 164 int fake_load_flags_; | 164 int fake_load_flags_; |
| 165 GURL fake_url_; | 165 GURL fake_url_; |
| (...skipping 17 matching lines...) Expand all Loading... |
| 183 class TestURLFetcherFactory : public content::URLFetcherFactory, | 183 class TestURLFetcherFactory : public content::URLFetcherFactory, |
| 184 public ScopedURLFetcherFactory { | 184 public ScopedURLFetcherFactory { |
| 185 public: | 185 public: |
| 186 TestURLFetcherFactory(); | 186 TestURLFetcherFactory(); |
| 187 virtual ~TestURLFetcherFactory(); | 187 virtual ~TestURLFetcherFactory(); |
| 188 | 188 |
| 189 virtual content::URLFetcher* CreateURLFetcher( | 189 virtual content::URLFetcher* CreateURLFetcher( |
| 190 int id, | 190 int id, |
| 191 const GURL& url, | 191 const GURL& url, |
| 192 content::URLFetcher::RequestType request_type, | 192 content::URLFetcher::RequestType request_type, |
| 193 content::URLFetcherDelegate* d) OVERRIDE; | 193 net::URLFetcherDelegate* d) OVERRIDE; |
| 194 TestURLFetcher* GetFetcherByID(int id) const; | 194 TestURLFetcher* GetFetcherByID(int id) const; |
| 195 void RemoveFetcherFromMap(int id); | 195 void RemoveFetcherFromMap(int id); |
| 196 | 196 |
| 197 private: | 197 private: |
| 198 // Maps from id passed to create to the returned URLFetcher. | 198 // Maps from id passed to create to the returned URLFetcher. |
| 199 typedef std::map<int, TestURLFetcher*> Fetchers; | 199 typedef std::map<int, TestURLFetcher*> Fetchers; |
| 200 Fetchers fetchers_; | 200 Fetchers fetchers_; |
| 201 | 201 |
| 202 DISALLOW_COPY_AND_ASSIGN(TestURLFetcherFactory); | 202 DISALLOW_COPY_AND_ASSIGN(TestURLFetcherFactory); |
| 203 }; | 203 }; |
| (...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 245 | 245 |
| 246 // If no fake response is set for the given URL this method will delegate the | 246 // If no fake response is set for the given URL this method will delegate the |
| 247 // call to |default_factory_| if it is not NULL, or return NULL if it is | 247 // call to |default_factory_| if it is not NULL, or return NULL if it is |
| 248 // NULL. | 248 // NULL. |
| 249 // Otherwise, it will return a URLFetcher object which will respond with the | 249 // Otherwise, it will return a URLFetcher object which will respond with the |
| 250 // pre-baked response that the client has set by calling SetFakeResponse(). | 250 // pre-baked response that the client has set by calling SetFakeResponse(). |
| 251 virtual content::URLFetcher* CreateURLFetcher( | 251 virtual content::URLFetcher* CreateURLFetcher( |
| 252 int id, | 252 int id, |
| 253 const GURL& url, | 253 const GURL& url, |
| 254 content::URLFetcher::RequestType request_type, | 254 content::URLFetcher::RequestType request_type, |
| 255 content::URLFetcherDelegate* d) OVERRIDE; | 255 net::URLFetcherDelegate* d) OVERRIDE; |
| 256 | 256 |
| 257 // Sets the fake response for a given URL. If success is true we will serve | 257 // Sets the fake response for a given URL. If success is true we will serve |
| 258 // an HTTP/200 and an HTTP/500 otherwise. The |response_data| may be empty. | 258 // an HTTP/200 and an HTTP/500 otherwise. The |response_data| may be empty. |
| 259 void SetFakeResponse(const std::string& url, | 259 void SetFakeResponse(const std::string& url, |
| 260 const std::string& response_data, | 260 const std::string& response_data, |
| 261 bool success); | 261 bool success); |
| 262 | 262 |
| 263 // Clear all the fake responses that were previously set via | 263 // Clear all the fake responses that were previously set via |
| 264 // SetFakeResponse(). | 264 // SetFakeResponse(). |
| 265 void ClearFakeResponses(); | 265 void ClearFakeResponses(); |
| (...skipping 13 matching lines...) Expand all Loading... |
| 279 class URLFetcherImplFactory : public content::URLFetcherFactory { | 279 class URLFetcherImplFactory : public content::URLFetcherFactory { |
| 280 public: | 280 public: |
| 281 URLFetcherImplFactory(); | 281 URLFetcherImplFactory(); |
| 282 virtual ~URLFetcherImplFactory(); | 282 virtual ~URLFetcherImplFactory(); |
| 283 | 283 |
| 284 // This method will create a real URLFetcher. | 284 // This method will create a real URLFetcher. |
| 285 virtual content::URLFetcher* CreateURLFetcher( | 285 virtual content::URLFetcher* CreateURLFetcher( |
| 286 int id, | 286 int id, |
| 287 const GURL& url, | 287 const GURL& url, |
| 288 content::URLFetcher::RequestType request_type, | 288 content::URLFetcher::RequestType request_type, |
| 289 content::URLFetcherDelegate* d) OVERRIDE; | 289 net::URLFetcherDelegate* d) OVERRIDE; |
| 290 | 290 |
| 291 }; | 291 }; |
| 292 | 292 |
| 293 #endif // CONTENT_TEST_TEST_URL_FETCHER_FACTORY_H_ | 293 #endif // CONTENT_TEST_TEST_URL_FETCHER_FACTORY_H_ |
| OLD | NEW |