Chromium Code Reviews| Index: net/url_request/url_request_unittest.cc |
| diff --git a/net/url_request/url_request_unittest.cc b/net/url_request/url_request_unittest.cc |
| index 4ee326bf2eddd975296c53b8c1b429796ab29274..6dbe797e93a76f6ea196d5044510a48bd00b9fb8 100644 |
| --- a/net/url_request/url_request_unittest.cc |
| +++ b/net/url_request/url_request_unittest.cc |
| @@ -39,7 +39,9 @@ |
| #include "net/base/ssl_connection_status_flags.h" |
| #include "net/base/test_data_directory.h" |
| #include "net/base/test_root_certs.h" |
| -#include "net/base/upload_data.h" |
| +#include "net/base/upload_bytes_element_reader.h" |
| +#include "net/base/upload_data_stream.h" |
| +#include "net/base/upload_file_element_reader.h" |
| #include "net/cookies/cookie_monster.h" |
| #include "net/cookies/cookie_store_test_helpers.h" |
| #include "net/disk_cache/disk_cache.h" |
| @@ -110,10 +112,10 @@ void FillBuffer(char* buffer, size_t len) { |
| } |
| } |
| -scoped_refptr<UploadData> CreateSimpleUploadData(const char* data) { |
| - scoped_refptr<UploadData> upload(new UploadData); |
| - upload->AppendBytes(data, strlen(data)); |
| - return upload; |
| +UploadDataStream* CreateSimpleUploadData(const char* data) { |
| + ScopedVector<UploadElementReader> element_readers; |
| + element_readers.push_back(new UploadBytesElementReader(data, strlen(data))); |
| + return new UploadDataStream(&element_readers, 0); |
| } |
| // Verify that the SSLInfo of a successful SSL connection has valid values. |
| @@ -1909,7 +1911,7 @@ class URLRequestTestHTTP : public URLRequestTest { |
| URLRequest req(redirect_url, &d, &default_context_); |
| req.set_method(request_method); |
| if (include_data) { |
| - req.set_upload(CreateSimpleUploadData(kData).get()); |
| + req.set_upload(CreateSimpleUploadData(kData)); |
| HttpRequestHeaders headers; |
| headers.SetHeader(HttpRequestHeaders::kContentLength, |
| base::UintToString(arraysize(kData) - 1)); |
| @@ -1954,9 +1956,10 @@ class URLRequestTestHTTP : public URLRequestTest { |
| URLRequest r(test_server_.GetURL("echo"), &d, &default_context_); |
| r.set_method(method.c_str()); |
| - scoped_refptr<UploadData> upload_data(new UploadData()); |
| - upload_data->AppendBytes(uploadBytes, kMsgSize); |
| - r.set_upload(upload_data); |
| + ScopedVector<UploadElementReader> element_readers; |
| + element_readers.push_back(new UploadBytesElementReader( |
| + uploadBytes, kMsgSize)); |
| + r.set_upload(new UploadDataStream(&element_readers, 0)); |
|
mmenke
2012/12/11 18:53:21
Looks to me like you can just use CreateSimpleUplo
hashimoto
2012/12/12 10:29:07
Done.
|
| r.Start(); |
| EXPECT_TRUE(r.is_pending()); |
| @@ -2322,7 +2325,7 @@ TEST_F(URLRequestTestHTTP, NetworkDelegateRedirectRequestPost) { |
| GURL original_url(test_server_.GetURL("empty.html")); |
| URLRequest r(original_url, &d, &context); |
| r.set_method("POST"); |
| - r.set_upload(CreateSimpleUploadData(kData).get()); |
| + r.set_upload(CreateSimpleUploadData(kData)); |
| HttpRequestHeaders headers; |
| headers.SetHeader(HttpRequestHeaders::kContentLength, |
| base::UintToString(arraysize(kData) - 1)); |
| @@ -3128,7 +3131,7 @@ TEST_F(URLRequestTestHTTP, PostFileTest) { |
| PathService::Get(base::DIR_EXE, &dir); |
| file_util::SetCurrentDirectory(dir); |
| - scoped_refptr<UploadData> upload_data(new UploadData); |
| + ScopedVector<UploadElementReader> element_readers; |
| FilePath path; |
| PathService::Get(base::DIR_SOURCE_ROOT, &path); |
| @@ -3136,14 +3139,15 @@ TEST_F(URLRequestTestHTTP, PostFileTest) { |
| path = path.Append(FILE_PATH_LITERAL("data")); |
| path = path.Append(FILE_PATH_LITERAL("url_request_unittest")); |
| path = path.Append(FILE_PATH_LITERAL("with-headers.html")); |
| - upload_data->AppendFileRange(path, 0, kuint64max, base::Time()); |
| + element_readers.push_back(new UploadFileElementReader( |
| + path, 0, kuint64max, base::Time())); |
| // This file should just be ignored in the upload stream. |
| - upload_data->AppendFileRange( |
| + element_readers.push_back(new UploadFileElementReader( |
| FilePath(FILE_PATH_LITERAL( |
| "c:\\path\\to\\non\\existant\\file.randomness.12345")), |
| - 0, kuint64max, base::Time()); |
| - r.set_upload(upload_data); |
| + 0, kuint64max, base::Time())); |
| + r.set_upload(new UploadDataStream(&element_readers, 0)); |
| r.Start(); |
| EXPECT_TRUE(r.is_pending()); |
| @@ -3677,7 +3681,7 @@ TEST_F(URLRequestTestHTTP, InterceptPost302RedirectGet) { |
| TestDelegate d; |
| URLRequest req(test_server_.GetURL("empty.html"), &d, &default_context_); |
| req.set_method("POST"); |
| - req.set_upload(CreateSimpleUploadData(kData).get()); |
| + req.set_upload(CreateSimpleUploadData(kData)); |
| HttpRequestHeaders headers; |
| headers.SetHeader(HttpRequestHeaders::kContentLength, |
| base::UintToString(arraysize(kData) - 1)); |
| @@ -3701,7 +3705,7 @@ TEST_F(URLRequestTestHTTP, InterceptPost307RedirectPost) { |
| TestDelegate d; |
| URLRequest req(test_server_.GetURL("empty.html"), &d, &default_context_); |
| req.set_method("POST"); |
| - req.set_upload(CreateSimpleUploadData(kData).get()); |
| + req.set_upload(CreateSimpleUploadData(kData)); |
| HttpRequestHeaders headers; |
| headers.SetHeader(HttpRequestHeaders::kContentLength, |
| base::UintToString(arraysize(kData) - 1)); |
| @@ -4221,7 +4225,7 @@ TEST_F(HTTPSRequestTest, HSTSPreservesPosts) { |
| &d, |
| &context); |
| req.set_method("POST"); |
| - req.set_upload(CreateSimpleUploadData(kData).get()); |
| + req.set_upload(CreateSimpleUploadData(kData)); |
| req.Start(); |
| MessageLoop::current()->Run(); |