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 |