Index: net/test/url_request/url_request_mock_http_job.h |
diff --git a/net/test/url_request/url_request_mock_http_job.h b/net/test/url_request/url_request_mock_http_job.h |
index 1a01a3084be2b9f6e847d2d2b0683ff080c14507..6eb3c72b86199aeeba02342344b0fbf39f8c1f50 100644 |
--- a/net/test/url_request/url_request_mock_http_job.h |
+++ b/net/test/url_request/url_request_mock_http_job.h |
@@ -28,6 +28,13 @@ namespace net { |
class URLRequestMockHTTPJob : public URLRequestFileJob { |
public: |
+ enum FailurePhase { |
+ START = 0, |
+ READ_ASYNC = 1, |
+ READ_SYNC = 2, |
+ MAX_FAILURE_PHASE = 3, |
+ }; |
+ |
// Note that all file IO is done using |worker_pool|. |
URLRequestMockHTTPJob(URLRequest* request, |
NetworkDelegate* network_delegate, |
@@ -35,6 +42,7 @@ class URLRequestMockHTTPJob : public URLRequestFileJob { |
const scoped_refptr<base::TaskRunner>& task_runner); |
void Start() override; |
+ bool ReadRawData(IOBuffer* buf, int buf_size, int* bytes_read) override; |
bool GetMimeType(std::string* mime_type) const override; |
int GetResponseCode() const override; |
bool GetCharset(std::string* charset) override; |
@@ -57,6 +65,13 @@ class URLRequestMockHTTPJob : public URLRequestFileJob { |
// construct a mock URL. |
static GURL GetMockUrl(const base::FilePath& path); |
+ // Given the path to a file relative to the path passed to AddUrlHandler(), |
+ // construct a mock URL that reports |net_error| at given |phase| of the |
+ // request. Reporting |net_error| ERR_IO_PENDING results in a hung request. |
+ static GURL GetMockUrlWithFailure(const base::FilePath& path, |
+ FailurePhase phase, |
+ int net_error); |
+ |
// Returns a URLRequestJobFactory::ProtocolHandler that serves |
// URLRequestMockHTTPJob's responding like an HTTP server. |base_path| is the |
// file path leading to the root of the directory to use as the root of the |
@@ -77,7 +92,13 @@ class URLRequestMockHTTPJob : public URLRequestFileJob { |
private: |
void GetResponseInfoConst(HttpResponseInfo* info) const; |
- void GetRawHeaders(std::string raw_headers); |
+ void SetHeadersAndStart(const std::string& raw_headers); |
+ // Checks query part of request url, and reports an error if it matches. |
+ // Error is parsed out from the query and is reported synchronously. |
+ // Reporting ERR_IO_PENDING results in a hung request. |
+ // The "readasync" error is posted asynchronously. |
+ bool MaybeReportErrorOnPhase(FailurePhase phase); |
+ |
std::string raw_headers_; |
const scoped_refptr<base::TaskRunner> task_runner_; |