Chromium Code Reviews| Index: content/browser/loader/upload_data_stream_builder_unittest.cc |
| diff --git a/content/browser/loader/upload_data_stream_builder_unittest.cc b/content/browser/loader/upload_data_stream_builder_unittest.cc |
| index a946e651e1f9700d673107635f070c2c6bf959be..2f15ac4a97c7a8d9f725a0f374ded871e8e05e02 100644 |
| --- a/content/browser/loader/upload_data_stream_builder_unittest.cc |
| +++ b/content/browser/loader/upload_data_stream_builder_unittest.cc |
| @@ -5,6 +5,7 @@ |
| #include "content/browser/loader/upload_data_stream_builder.h" |
| #include <algorithm> |
| +#include <string> |
| #include "base/files/file_path.h" |
| #include "base/files/file_util.h" |
| @@ -13,6 +14,9 @@ |
| #include "base/run_loop.h" |
| #include "base/time/time.h" |
| #include "content/common/resource_request_body.h" |
| +#include "net/base/io_buffer.h" |
| +#include "net/base/net_errors.h" |
| +#include "net/base/test_completion_callback.h" |
| #include "net/base/upload_bytes_element_reader.h" |
| #include "net/base/upload_data_stream.h" |
| #include "net/base/upload_file_element_reader.h" |
| @@ -108,8 +112,6 @@ TEST(UploadDataStreamBuilderTest, ResolveBlobAndCreateUploadDataStream) { |
| const std::string blob_id0("id-0"); |
| scoped_ptr<BlobDataBuilder> blob_data_builder( |
| new BlobDataBuilder(blob_id0)); |
| - scoped_ptr<BlobDataHandle> handle1 = |
| - blob_storage_context.AddFinishedBlob(blob_data_builder.get()); |
|
michaeln
2015/03/11 00:10:25
looks like this handle in the existing test actual
|
| const std::string blob_id1("id-1"); |
| const std::string kBlobData = "BlobData"; |
| @@ -310,4 +312,75 @@ TEST(UploadDataStreamBuilderTest, ResolveBlobAndCreateUploadDataStream) { |
| base::RunLoop().RunUntilIdle(); |
| } |
| +TEST(UploadDataStreamBuilderTest, |
| + WriteUploadDataStreamWithEmptyFileBackedBlob) { |
| + base::MessageLoop message_loop; |
| + { |
| + base::FilePath test_blob_path; |
| + ASSERT_TRUE(base::CreateTemporaryFile(&test_blob_path)); |
| + |
| + const uint64_t kZeroLength = 0; |
| + base::Time blob_time; |
| + base::Time::FromString("Tue, 15 Nov 1994, 12:45:26 GMT", &blob_time); |
| + ASSERT_TRUE(base::TouchFile(test_blob_path, blob_time, blob_time)); |
| + |
| + BlobStorageContext blob_storage_context; |
| + |
| + // First an empty blob |
| + scoped_ptr<BlobDataBuilder> blob_data_builder(new BlobDataBuilder("id-0")); |
| + scoped_ptr<BlobDataHandle> handle = |
| + blob_storage_context.AddFinishedBlob(blob_data_builder.get()); |
| + |
| + // Then a blob created from an empty file added several times. |
| + const std::string blob_id1("id-1"); |
| + blob_data_builder.reset(new BlobDataBuilder(blob_id1)); |
| + blob_data_builder->AppendFile(test_blob_path, 0, kZeroLength, blob_time); |
| + scoped_ptr<BlobDataHandle> handle2 = |
| + blob_storage_context.AddFinishedBlob(blob_data_builder.get()); |
| + |
| + ResourceRequestBody::Element blob_element; |
| + blob_element.SetToFilePathRange(test_blob_path, 0, kZeroLength, blob_time); |
| + |
| + scoped_refptr<ResourceRequestBody> request_body(new ResourceRequestBody()); |
| + scoped_ptr<net::UploadDataStream> upload(UploadDataStreamBuilder::Build( |
| + request_body.get(), &blob_storage_context, NULL, |
| + base::MessageLoopProxy::current().get())); |
| + |
| + request_body = new ResourceRequestBody(); |
| + request_body->AppendBlob(blob_id1); |
| + request_body->AppendBlob(blob_id1); |
| + request_body->AppendBlob(blob_id1); |
| + |
| + upload = UploadDataStreamBuilder::Build( |
| + request_body.get(), &blob_storage_context, NULL, |
| + base::MessageLoopProxy::current().get()); |
| + ASSERT_TRUE(upload->GetElementReaders()); |
| + const auto& readers = *upload->GetElementReaders(); |
| + ASSERT_EQ(3U, readers.size()); |
| + EXPECT_TRUE(AreElementsEqual(*readers[0], blob_element)); |
| + EXPECT_TRUE(AreElementsEqual(*readers[1], blob_element)); |
| + EXPECT_TRUE(AreElementsEqual(*readers[2], blob_element)); |
| + |
| + net::TestCompletionCallback init_callback; |
| + ASSERT_EQ(net::ERR_IO_PENDING, upload->Init(init_callback.callback())); |
| + EXPECT_EQ(net::OK, init_callback.WaitForResult()); |
| + |
| + EXPECT_EQ(kZeroLength, upload->size()); |
| + |
| + // Purposely (try to) read more than what is in the stream. If we try to |
| + // read zero bytes then UploadDataStream::Read will fail a DCHECK. |
| + int kBufferLength = kZeroLength + 1; |
| + scoped_ptr<char[]> buffer(new char[kBufferLength]); |
| + scoped_refptr<net::IOBuffer> io_buffer = |
| + new net::WrappedIOBuffer(buffer.get()); |
| + net::TestCompletionCallback read_callback; |
| + int result = |
| + upload->Read(io_buffer.get(), kBufferLength, read_callback.callback()); |
| + EXPECT_EQ(static_cast<int>(kZeroLength), read_callback.GetResult(result)); |
| + |
| + base::DeleteFile(test_blob_path, false); |
| + } |
| + // Clean up for ASAN. |
| + base::RunLoop().RunUntilIdle(); |
| +} |
| } // namespace content |