| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "net/test/url_request/url_request_mock_data_job.h" | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "base/message_loop/message_loop.h" | |
| 9 #include "base/strings/string_number_conversions.h" | |
| 10 #include "net/base/io_buffer.h" | |
| 11 #include "net/base/url_util.h" | |
| 12 #include "net/http/http_request_headers.h" | |
| 13 #include "net/url_request/url_request_filter.h" | |
| 14 | |
| 15 namespace net { | |
| 16 namespace { | |
| 17 | |
| 18 const char kMockHostname[] = "mock.data"; | |
| 19 | |
| 20 // Gets the data from URL of the form: | |
| 21 // scheme://kMockHostname/?data=abc&repeat_count=nnn. | |
| 22 std::string GetDataFromRequest(const net::URLRequest& request) { | |
| 23 std::string value; | |
| 24 if (!GetValueForKeyInQuery(request.url(), "data", &value)) | |
| 25 return "default_data"; | |
| 26 return value; | |
| 27 } | |
| 28 | |
| 29 // Gets the numeric repeat count from URL of the form: | |
| 30 // scheme://kMockHostname/?data=abc&repeat_count=nnn. | |
| 31 int GetRepeatCountFromRequest(const net::URLRequest& request) { | |
| 32 std::string value; | |
| 33 if (!GetValueForKeyInQuery(request.url(), "repeat", &value)) | |
| 34 return 1; | |
| 35 | |
| 36 int repeat_count; | |
| 37 if (!base::StringToInt(value, &repeat_count)) | |
| 38 return 1; | |
| 39 | |
| 40 DCHECK_GT(repeat_count, 0); | |
| 41 | |
| 42 return repeat_count; | |
| 43 } | |
| 44 | |
| 45 GURL GetMockUrl(const std::string& scheme, | |
| 46 const std::string& hostname, | |
| 47 const std::string& data, | |
| 48 int data_repeat_count) { | |
| 49 DCHECK_GT(data_repeat_count, 0); | |
| 50 std::string url(scheme + "://" + hostname + "/"); | |
| 51 url.append("?data="); | |
| 52 url.append(data); | |
| 53 url.append("&repeat="); | |
| 54 url.append(base::IntToString(data_repeat_count)); | |
| 55 return GURL(url); | |
| 56 } | |
| 57 | |
| 58 class MockJobInterceptor : public net::URLRequestInterceptor { | |
| 59 public: | |
| 60 MockJobInterceptor() {} | |
| 61 ~MockJobInterceptor() override {} | |
| 62 | |
| 63 // net::URLRequestInterceptor implementation | |
| 64 net::URLRequestJob* MaybeInterceptRequest( | |
| 65 net::URLRequest* request, | |
| 66 net::NetworkDelegate* network_delegate) const override { | |
| 67 return new URLRequestMockDataJob(request, network_delegate, | |
| 68 GetDataFromRequest(*request), | |
| 69 GetRepeatCountFromRequest(*request)); | |
| 70 } | |
| 71 | |
| 72 private: | |
| 73 DISALLOW_COPY_AND_ASSIGN(MockJobInterceptor); | |
| 74 }; | |
| 75 | |
| 76 } // namespace | |
| 77 | |
| 78 URLRequestMockDataJob::URLRequestMockDataJob(URLRequest* request, | |
| 79 NetworkDelegate* network_delegate, | |
| 80 const std::string& data, | |
| 81 int data_repeat_count) | |
| 82 : URLRequestJob(request, network_delegate), | |
| 83 data_offset_(0), | |
| 84 weak_factory_(this) { | |
| 85 DCHECK_GT(data_repeat_count, 0); | |
| 86 for (int i = 0; i < data_repeat_count; ++i) { | |
| 87 data_.append(data); | |
| 88 } | |
| 89 } | |
| 90 | |
| 91 void URLRequestMockDataJob::Start() { | |
| 92 // Start reading asynchronously so that all error reporting and data | |
| 93 // callbacks happen as they would for network requests. | |
| 94 base::MessageLoop::current()->PostTask( | |
| 95 FROM_HERE, base::Bind(&URLRequestMockDataJob::StartAsync, | |
| 96 weak_factory_.GetWeakPtr())); | |
| 97 } | |
| 98 | |
| 99 URLRequestMockDataJob::~URLRequestMockDataJob() { | |
| 100 } | |
| 101 | |
| 102 bool URLRequestMockDataJob::ReadRawData(IOBuffer* buf, | |
| 103 int buf_size, | |
| 104 int* bytes_read) { | |
| 105 DCHECK(bytes_read); | |
| 106 *bytes_read = static_cast<int>( | |
| 107 std::min(static_cast<size_t>(buf_size), data_.length() - data_offset_)); | |
| 108 memcpy(buf->data(), data_.c_str() + data_offset_, *bytes_read); | |
| 109 data_offset_ += *bytes_read; | |
| 110 return true; | |
| 111 } | |
| 112 | |
| 113 void URLRequestMockDataJob::StartAsync() { | |
| 114 if (!request_) | |
| 115 return; | |
| 116 | |
| 117 set_expected_content_size(data_.length()); | |
| 118 NotifyHeadersComplete(); | |
| 119 } | |
| 120 | |
| 121 // static | |
| 122 void URLRequestMockDataJob::AddUrlHandler() { | |
| 123 return AddUrlHandlerForHostname(kMockHostname); | |
| 124 } | |
| 125 | |
| 126 // static | |
| 127 void URLRequestMockDataJob::AddUrlHandlerForHostname( | |
| 128 const std::string& hostname) { | |
| 129 // Add |hostname| to net::URLRequestFilter for HTTP and HTTPS. | |
| 130 net::URLRequestFilter* filter = net::URLRequestFilter::GetInstance(); | |
| 131 filter->AddHostnameInterceptor("http", hostname, | |
| 132 make_scoped_ptr(new MockJobInterceptor())); | |
| 133 filter->AddHostnameInterceptor("https", hostname, | |
| 134 make_scoped_ptr(new MockJobInterceptor())); | |
| 135 } | |
| 136 | |
| 137 // static | |
| 138 GURL URLRequestMockDataJob::GetMockHttpUrl(const std::string& data, | |
| 139 int repeat_count) { | |
| 140 return GetMockHttpUrlForHostname(kMockHostname, data, repeat_count); | |
| 141 } | |
| 142 | |
| 143 // static | |
| 144 GURL URLRequestMockDataJob::GetMockHttpsUrl(const std::string& data, | |
| 145 int repeat_count) { | |
| 146 return GetMockHttpsUrlForHostname(kMockHostname, data, repeat_count); | |
| 147 } | |
| 148 | |
| 149 // static | |
| 150 GURL URLRequestMockDataJob::GetMockHttpUrlForHostname( | |
| 151 const std::string& hostname, | |
| 152 const std::string& data, | |
| 153 int repeat_count) { | |
| 154 return GetMockUrl("http", hostname, data, repeat_count); | |
| 155 } | |
| 156 | |
| 157 // static | |
| 158 GURL URLRequestMockDataJob::GetMockHttpsUrlForHostname( | |
| 159 const std::string& hostname, | |
| 160 const std::string& data, | |
| 161 int repeat_count) { | |
| 162 return GetMockUrl("https", hostname, data, repeat_count); | |
| 163 } | |
| 164 | |
| 165 } // namespace net | |
| OLD | NEW |