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 |