Index: net/test/url_request/url_request_mock_data_job.h |
diff --git a/net/test/url_request/url_request_mock_data_job.h b/net/test/url_request/url_request_mock_data_job.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..cea58d24c9ac7e8f9927f9fcc9780875c84455ee |
--- /dev/null |
+++ b/net/test/url_request/url_request_mock_data_job.h |
@@ -0,0 +1,61 @@ |
+// Copyright 2014 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#ifndef NET_URL_REQUEST_URL_REQUEST_MOCK_DATA_JOB_H_ |
+#define NET_URL_REQUEST_URL_REQUEST_MOCK_DATA_JOB_H_ |
+ |
+#include <string> |
+ |
+#include "base/memory/weak_ptr.h" |
+#include "net/base/net_export.h" |
+#include "net/url_request/url_request_job.h" |
+ |
+namespace net { |
+ |
+class URLRequest; |
+ |
+// Mock data job, which synchronously returns data repeated multiple times. |
+class NET_EXPORT URLRequestMockDataJob : public URLRequestJob { |
+ public: |
+ URLRequestMockDataJob(URLRequest* request, |
+ NetworkDelegate* network_delegate, |
+ const std::string& data, |
+ unsigned data_repeat_count); |
+ |
+ void Start() override; |
+ bool ReadRawData(IOBuffer* buf, int buf_size, int* bytes_read) override; |
+ |
+ // Adds the testing URLs to the URLRequestFilter. |
+ static void AddUrlHandler(); |
+ static void AddUrlHandlerForHostname(const std::string& hostname); |
+ |
+ // Given data and repeat cound, constructs a mock URL that will return that |
+ // data repeated |repeat_count| times when started. |data| must be safe for |
+ // URL. |
+ static GURL GetMockHttpUrl(const std::string& data, int repeat_count); |
+ static GURL GetMockHttpsUrl(const std::string& data, int repeat_count); |
+ |
+ // URLRequestFailedJob must be added as a handler for |hostname| for |
+ // the returned URL to return |net_error|. |
+ static GURL GetMockHttpUrlForHostname(const std::string& hostname, |
+ const std::string& data, |
+ int repeat_count); |
+ static GURL GetMockHttpsUrlForHostname(const std::string& hostname, |
+ const std::string& data, |
+ int repeat_count); |
+ |
+ protected: |
+ ~URLRequestMockDataJob() override; |
mmenke
2014/12/10 20:59:15
This doesn't need to be protected, does it? Fine
mef
2014/12/10 23:23:36
Done.
|
+ |
+ private: |
+ void StartAsync(); |
+ |
+ std::string data_; |
+ int64 data_offset_; |
+ base::WeakPtrFactory<URLRequestMockDataJob> weak_factory_; |
+}; |
+ |
+} // namespace net |
+ |
+#endif // NET_URL_REQUEST_URL_REQUEST_SIMPLE_JOB_H_ |