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