| 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 // A URLRequestJob class that pulls the net and http headers from disk. | 5 // A URLRequestJob class that pulls the net and http headers from disk. |
| 6 | 6 |
| 7 #ifndef NET_TEST_URL_REQUEST_URL_REQUEST_MOCK_HTTP_JOB_H_ | 7 #ifndef NET_TEST_URL_REQUEST_URL_REQUEST_MOCK_HTTP_JOB_H_ |
| 8 #define NET_TEST_URL_REQUEST_URL_REQUEST_MOCK_HTTP_JOB_H_ | 8 #define NET_TEST_URL_REQUEST_URL_REQUEST_MOCK_HTTP_JOB_H_ |
| 9 | 9 |
| 10 #include <string> | 10 #include <string> |
| (...skipping 10 matching lines...) Expand all Loading... |
| 21 } | 21 } |
| 22 | 22 |
| 23 namespace net { | 23 namespace net { |
| 24 class URLRequestInterceptor; | 24 class URLRequestInterceptor; |
| 25 } | 25 } |
| 26 | 26 |
| 27 namespace net { | 27 namespace net { |
| 28 | 28 |
| 29 class URLRequestMockHTTPJob : public URLRequestFileJob { | 29 class URLRequestMockHTTPJob : public URLRequestFileJob { |
| 30 public: | 30 public: |
| 31 enum FailurePhase { |
| 32 START = 0, |
| 33 READ_ASYNC = 1, |
| 34 READ_SYNC = 2, |
| 35 MAX_FAILURE_PHASE = 3, |
| 36 }; |
| 37 |
| 31 // Note that all file IO is done using |worker_pool|. | 38 // Note that all file IO is done using |worker_pool|. |
| 32 URLRequestMockHTTPJob(URLRequest* request, | 39 URLRequestMockHTTPJob(URLRequest* request, |
| 33 NetworkDelegate* network_delegate, | 40 NetworkDelegate* network_delegate, |
| 34 const base::FilePath& file_path, | 41 const base::FilePath& file_path, |
| 35 const scoped_refptr<base::TaskRunner>& task_runner); | 42 const scoped_refptr<base::TaskRunner>& task_runner); |
| 36 | 43 |
| 37 void Start() override; | 44 void Start() override; |
| 45 bool ReadRawData(IOBuffer* buf, int buf_size, int* bytes_read) override; |
| 38 bool GetMimeType(std::string* mime_type) const override; | 46 bool GetMimeType(std::string* mime_type) const override; |
| 39 int GetResponseCode() const override; | 47 int GetResponseCode() const override; |
| 40 bool GetCharset(std::string* charset) override; | 48 bool GetCharset(std::string* charset) override; |
| 41 void GetResponseInfo(HttpResponseInfo* info) override; | 49 void GetResponseInfo(HttpResponseInfo* info) override; |
| 42 bool IsRedirectResponse(GURL* location, int* http_status_code) override; | 50 bool IsRedirectResponse(GURL* location, int* http_status_code) override; |
| 43 | 51 |
| 44 // Adds the testing URLs to the URLRequestFilter. | 52 // Adds the testing URLs to the URLRequestFilter. |
| 45 static void AddUrlHandler( | 53 static void AddUrlHandler( |
| 46 const base::FilePath& base_path, | 54 const base::FilePath& base_path, |
| 47 const scoped_refptr<base::SequencedWorkerPool>& worker_pool); | 55 const scoped_refptr<base::SequencedWorkerPool>& worker_pool); |
| 48 | 56 |
| 49 // Respond to all HTTP requests of |hostname| with contents of the file | 57 // Respond to all HTTP requests of |hostname| with contents of the file |
| 50 // located at |file_path|. | 58 // located at |file_path|. |
| 51 static void AddHostnameToFileHandler( | 59 static void AddHostnameToFileHandler( |
| 52 const std::string& hostname, | 60 const std::string& hostname, |
| 53 const base::FilePath& file, | 61 const base::FilePath& file, |
| 54 const scoped_refptr<base::SequencedWorkerPool>& worker_pool); | 62 const scoped_refptr<base::SequencedWorkerPool>& worker_pool); |
| 55 | 63 |
| 56 // Given the path to a file relative to the path passed to AddUrlHandler(), | 64 // Given the path to a file relative to the path passed to AddUrlHandler(), |
| 57 // construct a mock URL. | 65 // construct a mock URL. |
| 58 static GURL GetMockUrl(const base::FilePath& path); | 66 static GURL GetMockUrl(const base::FilePath& path); |
| 59 | 67 |
| 68 // Given the path to a file relative to the path passed to AddUrlHandler(), |
| 69 // construct a mock URL that reports |net_error| at given |phase| of the |
| 70 // request. Reporting |net_error| ERR_IO_PENDING results in a hung request. |
| 71 static GURL GetMockUrlWithFailure(const base::FilePath& path, |
| 72 FailurePhase phase, |
| 73 int net_error); |
| 74 |
| 60 // Returns a URLRequestJobFactory::ProtocolHandler that serves | 75 // Returns a URLRequestJobFactory::ProtocolHandler that serves |
| 61 // URLRequestMockHTTPJob's responding like an HTTP server. |base_path| is the | 76 // URLRequestMockHTTPJob's responding like an HTTP server. |base_path| is the |
| 62 // file path leading to the root of the directory to use as the root of the | 77 // file path leading to the root of the directory to use as the root of the |
| 63 // HTTP server. | 78 // HTTP server. |
| 64 static scoped_ptr<URLRequestInterceptor> CreateInterceptor( | 79 static scoped_ptr<URLRequestInterceptor> CreateInterceptor( |
| 65 const base::FilePath& base_path, | 80 const base::FilePath& base_path, |
| 66 const scoped_refptr<base::SequencedWorkerPool>& worker_pool); | 81 const scoped_refptr<base::SequencedWorkerPool>& worker_pool); |
| 67 | 82 |
| 68 // Returns a URLRequestJobFactory::ProtocolHandler that serves | 83 // Returns a URLRequestJobFactory::ProtocolHandler that serves |
| 69 // URLRequestMockHTTPJob's responding like an HTTP server. It responds to all | 84 // URLRequestMockHTTPJob's responding like an HTTP server. It responds to all |
| 70 // requests with the contents of |file|. | 85 // requests with the contents of |file|. |
| 71 static scoped_ptr<URLRequestInterceptor> CreateInterceptorForSingleFile( | 86 static scoped_ptr<URLRequestInterceptor> CreateInterceptorForSingleFile( |
| 72 const base::FilePath& file, | 87 const base::FilePath& file, |
| 73 const scoped_refptr<base::SequencedWorkerPool>& worker_pool); | 88 const scoped_refptr<base::SequencedWorkerPool>& worker_pool); |
| 74 | 89 |
| 75 protected: | 90 protected: |
| 76 ~URLRequestMockHTTPJob() override; | 91 ~URLRequestMockHTTPJob() override; |
| 77 | 92 |
| 78 private: | 93 private: |
| 79 void GetResponseInfoConst(HttpResponseInfo* info) const; | 94 void GetResponseInfoConst(HttpResponseInfo* info) const; |
| 80 void GetRawHeaders(std::string raw_headers); | 95 void SetHeadersAndStart(const std::string& raw_headers); |
| 96 // Checks query part of request url, and reports an error if it matches. |
| 97 // Error is parsed out from the query and is reported synchronously. |
| 98 // Reporting ERR_IO_PENDING results in a hung request. |
| 99 // The "readasync" error is posted asynchronously. |
| 100 bool MaybeReportErrorOnPhase(FailurePhase phase); |
| 101 |
| 81 std::string raw_headers_; | 102 std::string raw_headers_; |
| 82 const scoped_refptr<base::TaskRunner> task_runner_; | 103 const scoped_refptr<base::TaskRunner> task_runner_; |
| 83 | 104 |
| 84 base::WeakPtrFactory<URLRequestMockHTTPJob> weak_ptr_factory_; | 105 base::WeakPtrFactory<URLRequestMockHTTPJob> weak_ptr_factory_; |
| 85 | 106 |
| 86 DISALLOW_COPY_AND_ASSIGN(URLRequestMockHTTPJob); | 107 DISALLOW_COPY_AND_ASSIGN(URLRequestMockHTTPJob); |
| 87 }; | 108 }; |
| 88 | 109 |
| 89 } // namespace net | 110 } // namespace net |
| 90 | 111 |
| 91 #endif // NET_TEST_URL_REQUEST_URL_REQUEST_MOCK_HTTP_JOB_H_ | 112 #endif // NET_TEST_URL_REQUEST_URL_REQUEST_MOCK_HTTP_JOB_H_ |
| OLD | NEW |