| 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..57161eaf4caf1a281d91b2c5fd020e2c192f5f39 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"
|
| @@ -310,4 +314,70 @@ TEST(UploadDataStreamBuilderTest, ResolveBlobAndCreateUploadDataStream) {
|
| base::RunLoop().RunUntilIdle();
|
| }
|
|
|
| +TEST(UploadDataStreamBuilderTest,
|
| + WriteUploadDataStreamWithEmptyFileBackedBlob) {
|
| + base::MessageLoopForIO 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;
|
| +
|
| + // A blob created from an empty file added several times.
|
| + const std::string blob_id("id-0");
|
| + scoped_ptr<BlobDataBuilder> blob_data_builder(new BlobDataBuilder(blob_id));
|
| + blob_data_builder->AppendFile(test_blob_path, 0, kZeroLength, blob_time);
|
| + scoped_ptr<BlobDataHandle> handle =
|
| + 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_id);
|
| + request_body->AppendBlob(blob_id);
|
| + request_body->AppendBlob(blob_id);
|
| +
|
| + 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
|
|
|