Chromium Code Reviews| 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_ |