Create blobs from Disk Cache entries.

content/browser/loader/upload_data_stream_builder_unittest.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #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"
 #include "base/run_loop.h"
 #include "base/thread_task_runner_handle.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_disk_cache_entry_element_reader.h"
 #include "net/base/upload_file_element_reader.h"
+#include "net/disk_cache/disk_cache.h"
 #include "storage/browser/blob/blob_data_builder.h"
 #include "storage/browser/blob/blob_storage_context.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
 
 using storage::BlobDataBuilder;
 using storage::BlobDataHandle;
 using storage::BlobStorageContext;
 
 namespace content {
 namespace {
 
+const int kTestDiskCacheStreamIndex = 0;
+
+// Our disk cache tests don't need a real data handle since the tests themselves
+// scope the disk cache and entries.
+class EmptyDataHandle : public storage::BlobDataBuilder::DataHandle {
+ private:
+  ~EmptyDataHandle() override {}
+};
+
+scoped_ptr<disk_cache::Backend> CreateInMemoryDiskCache() {
+  scoped_ptr<disk_cache::Backend> cache;
+  net::TestCompletionCallback cb;
+  int rv = disk_cache::CreateCacheBackend(net::MEMORY_CACHE,
+                                          net::CACHE_BACKEND_DEFAULT,
+                                          base::FilePath(), 0,
+                                          false, NULL, NULL, &cache,
+                                          cb.callback());
+  EXPECT_EQ(net::OK, cb.GetResult(rv));
+
+  return cache.Pass();
+}
+
+disk_cache::ScopedEntryPtr CreateDiskCacheEntry(disk_cache::Backend* cache,
+                                                const char* key,
+                                                const std::string& data) {
+  disk_cache::Entry* tmp_entry = nullptr;
+  net::TestCompletionCallback cb;
+  int rv = cache->CreateEntry(key, &tmp_entry, cb.callback());
+  if (cb.GetResult(rv) != net::OK)
+    return nullptr;
+  disk_cache::ScopedEntryPtr entry(tmp_entry);
+
+  scoped_refptr<net::StringIOBuffer> iobuffer = new net::StringIOBuffer(data);
+  rv = entry->WriteData(kTestDiskCacheStreamIndex, 0, iobuffer.get(),
+                        iobuffer->size(), cb.callback(), false);
+  EXPECT_EQ(static_cast<int>(data.size()), cb.GetResult(rv));
+  return entry.Pass();
+}
+
 bool AreElementsEqual(const net::UploadElementReader& reader,
                       const ResourceRequestBody::Element& element) {
   switch(element.type()) {
     case ResourceRequestBody::Element::TYPE_BYTES: {
       const net::UploadBytesElementReader* bytes_reader =
           reader.AsBytesReader();
       return bytes_reader &&
           element.length() == bytes_reader->length() &&
           std::equal(element.bytes(), element.bytes() + element.length(),
                      bytes_reader->bytes());
     }
     case ResourceRequestBody::Element::TYPE_FILE: {
       const net::UploadFileElementReader* file_reader = reader.AsFileReader();
       return file_reader &&
           file_reader->path() == element.path() &&
           file_reader->range_offset() == element.offset() &&
           file_reader->range_length() == element.length() &&
           file_reader->expected_modification_time() ==
           element.expected_modification_time();
       break;
     }
+    case ResourceRequestBody::Element::TYPE_DISK_CACHE_ENTRY: {
+      // TODO(gavinp): Should we be comparing a higher level structure
+      // such as the BlobDataItem so that we can do stronger equality
+      // comparisons?
+      const net::UploadDiskCacheEntryElementReader* disk_cache_entry_reader =
+          reader.AsDiskCacheEntryReaderForTests();
+      return disk_cache_entry_reader &&
+          disk_cache_entry_reader->range_offset_for_tests() ==
+              static_cast<int>(element.offset()) &&
+          disk_cache_entry_reader->range_length_for_tests() ==
+              static_cast<int>(element.length());
+      break;
+    }
     default:
       NOTREACHED();
   }
   return false;
 }
 
 }  // namespace
 
 TEST(UploadDataStreamBuilderTest, CreateUploadDataStreamWithoutBlob) {
   base::MessageLoop message_loop;
@@ skipping 52 matching lines @@
 
     const std::string blob_id1("id-1");
     const std::string kBlobData = "BlobData";
     blob_data_builder.reset(new BlobDataBuilder(blob_id1));
     blob_data_builder->AppendData(kBlobData);
     blob_data_builder->AppendFile(
         base::FilePath(FILE_PATH_LITERAL("BlobFile.txt")), 0, 20, time1);
     scoped_ptr<BlobDataHandle> handle2 =
         blob_storage_context.AddFinishedBlob(blob_data_builder.get());
 
+    const std::string blob_id2("id-2");

[michaeln, 2015/06/17 00:57:37]

nit: should two adjacent lines be consistent with one another :)

kBlobId2 = "id";
disk_cache_data("x");

[gavinp, 2015/06/17 15:14:21]

Awkwardly, this usage is cut and paste from line 172 above, so it's already
consistent in its inconsistent way.... Awkward.

+    const std::string kDiskCacheData = "DiskCacheData";
+    scoped_ptr<disk_cache::Backend> disk_cache_backend =
+        CreateInMemoryDiskCache();
+    ASSERT_TRUE(disk_cache_backend);
+    disk_cache::ScopedEntryPtr disk_cache_entry =
+        CreateDiskCacheEntry(disk_cache_backend.get(), "a key", kDiskCacheData);
+    ASSERT_TRUE(disk_cache_entry);
+    blob_data_builder.reset(new BlobDataBuilder(blob_id2));
+    blob_data_builder->AppendDiskCacheEntry(
+        new EmptyDataHandle(), disk_cache_entry.get(),
+        kTestDiskCacheStreamIndex);
+    scoped_ptr<BlobDataHandle> handle3 =
+        blob_storage_context.AddFinishedBlob(blob_data_builder.get());
+
     // Setup upload data elements for comparison.
-    ResourceRequestBody::Element blob_element1, blob_element2;
+    ResourceRequestBody::Element blob_element1, blob_element2, blob_element3;
     blob_element1.SetToBytes(kBlobData.c_str(), kBlobData.size());
     blob_element2.SetToFilePathRange(
         base::FilePath(FILE_PATH_LITERAL("BlobFile.txt")), 0, 20, time1);
+    blob_element3.SetToDiskCacheEntryRange(0, kDiskCacheData.size());
 
     ResourceRequestBody::Element upload_element1, upload_element2;
     upload_element1.SetToBytes("Hello", 5);
     upload_element2.SetToFilePathRange(
         base::FilePath(FILE_PATH_LITERAL("foo1.txt")), 0, 20, time2);
 
     // Test no blob reference.
     scoped_refptr<ResourceRequestBody> request_body(new ResourceRequestBody());
     request_body->AppendBytes(
         upload_element1.bytes(),
@@ skipping 32 matching lines @@
     upload = UploadDataStreamBuilder::Build(
         request_body.get(), &blob_storage_context, NULL,
         base::ThreadTaskRunnerHandle::Get().get());
     ASSERT_TRUE(upload->GetElementReaders());
     ASSERT_EQ(2U, upload->GetElementReaders()->size());
     EXPECT_TRUE(AreElementsEqual(
         *(*upload->GetElementReaders())[0], blob_element1));
     EXPECT_TRUE(AreElementsEqual(
         *(*upload->GetElementReaders())[1], blob_element2));
 
+    // Test having one blob reference which refers to a disk cache entry.
+    request_body = new ResourceRequestBody();
+    request_body->AppendBlob(blob_id2);
+
+    upload = UploadDataStreamBuilder::Build(
+        request_body.get(), &blob_storage_context, NULL,
+        base::ThreadTaskRunnerHandle::Get().get());
+    ASSERT_TRUE(upload->GetElementReaders());
+    ASSERT_EQ(1U, upload->GetElementReaders()->size());
+    EXPECT_TRUE(AreElementsEqual(
+        *(*upload->GetElementReaders())[0], blob_element3));
+
     // Test having one blob reference at the beginning.
     request_body = new ResourceRequestBody();
     request_body->AppendBlob(blob_id1);
     request_body->AppendBytes(
         upload_element1.bytes(),
         upload_element1.length());
     request_body->AppendFileRange(
         upload_element2.path(),
         upload_element2.offset(),
         upload_element2.length(),
@@ skipping 165 matching lines @@
     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