Break large blobs into multiple ipcs during creation.

webkit/blob/blob_storage_controller_unittest.cc

 // Copyright (c) 2011 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 "base/file_path.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/time.h"
 #include "net/base/upload_data.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "webkit/blob/blob_data.h"
 #include "webkit/blob/blob_storage_controller.h"
 
 using net::UploadData;
 
 namespace webkit_blob {
 
+namespace {
+
+void RegisterBlobUrl(BlobStorageController* controller,
+                     const GURL& url, const BlobData* data) {
+  controller->RegisterUnfinalizedBlobUrl(url);
+  for (std::vector<BlobData::Item>::const_iterator iter =
+           data->items().begin();
+       iter != data->items().end(); ++iter) {
+    controller->AppendBlobDataItem(url, *iter);
+  }
+  controller->FinalizeBlob(url, data->content_type());
+}
+
+}
+
 TEST(BlobStorageControllerTest, RegisterBlobUrl) {
   // Setup a set of blob data for testing.
   base::Time time1, time2;
   base::Time::FromString("Tue, 15 Nov 1994, 12:45:26 GMT", &time1);
   base::Time::FromString("Mon, 14 Nov 1994, 11:30:49 GMT", &time2);
 
   scoped_refptr<BlobData> blob_data1(new BlobData());
   blob_data1->AppendData("Data1");
   blob_data1->AppendData("Data2");
   blob_data1->AppendFile(FilePath(FILE_PATH_LITERAL("File1.txt")),
     10, 1024, time1);
 
   scoped_refptr<BlobData> blob_data2(new BlobData());
   blob_data2->AppendData("Data3");
   blob_data2->AppendBlob(GURL("blob://url_1"), 8, 100);
   blob_data2->AppendFile(FilePath(FILE_PATH_LITERAL("File2.txt")),
       0, 20, time2);
 
   scoped_refptr<BlobData> canonicalized_blob_data2(new BlobData());
   canonicalized_blob_data2->AppendData("Data3");
   canonicalized_blob_data2->AppendData("Data2", 3, 2);
   canonicalized_blob_data2->AppendFile(FilePath(FILE_PATH_LITERAL("File1.txt")),
       10, 98, time1);
   canonicalized_blob_data2->AppendFile(FilePath(FILE_PATH_LITERAL("File2.txt")),
       0, 20, time2);
 
-  scoped_ptr<BlobStorageController> blob_storage_controller(
-      new BlobStorageController());
+  BlobStorageController blob_storage_controller;
 
   // Test registering a blob URL referring to the blob data containing only
   // data and file.
   GURL blob_url1("blob://url_1");
-  blob_storage_controller->RegisterBlobUrl(blob_url1, blob_data1);
+  RegisterBlobUrl(&blob_storage_controller, blob_url1, blob_data1);
 
   BlobData* blob_data_found =
-      blob_storage_controller->GetBlobDataFromUrl(blob_url1);
+      blob_storage_controller.GetBlobDataFromUrl(blob_url1);
   ASSERT_TRUE(blob_data_found != NULL);
   EXPECT_TRUE(*blob_data_found == *blob_data1);
 
   // Test registering a blob URL referring to the blob data containing data,
   // file and blob.
   GURL blob_url2("blob://url_2");
-  blob_storage_controller->RegisterBlobUrl(blob_url2, blob_data2);
+  RegisterBlobUrl(&blob_storage_controller, blob_url2, blob_data2);
 
-  blob_data_found = blob_storage_controller->GetBlobDataFromUrl(blob_url2);
+  blob_data_found = blob_storage_controller.GetBlobDataFromUrl(blob_url2);
   ASSERT_TRUE(blob_data_found != NULL);
   EXPECT_TRUE(*blob_data_found == *canonicalized_blob_data2);
 
   // Test registering a blob URL referring to existent blob URL.
   GURL blob_url3("blob://url_3");
-  blob_storage_controller->RegisterBlobUrlFrom(blob_url3, blob_url1);
+  blob_storage_controller.RegisterBlobUrlFrom(blob_url3, blob_url1);
 
-  blob_data_found = blob_storage_controller->GetBlobDataFromUrl(blob_url3);
+  blob_data_found = blob_storage_controller.GetBlobDataFromUrl(blob_url3);
   ASSERT_TRUE(blob_data_found != NULL);
   EXPECT_TRUE(*blob_data_found == *blob_data1);
 
   // Test unregistering a blob URL.
-  blob_storage_controller->UnregisterBlobUrl(blob_url3);
-  blob_data_found = blob_storage_controller->GetBlobDataFromUrl(blob_url3);
+  blob_storage_controller.UnregisterBlobUrl(blob_url3);
+  blob_data_found = blob_storage_controller.GetBlobDataFromUrl(blob_url3);
   EXPECT_TRUE(!blob_data_found);
 }
 
 TEST(BlobStorageControllerTest, ResolveBlobReferencesInUploadData) {
   // Setup blob data for testing.
   base::Time time1, time2;
   base::Time::FromString("Tue, 15 Nov 1994, 12:45:26 GMT", &time1);
   base::Time::FromString("Mon, 14 Nov 1994, 11:30:49 GMT", &time2);
 
-  scoped_ptr<BlobStorageController> blob_storage_controller(
-      new BlobStorageController());
-
+  BlobStorageController blob_storage_controller;
   scoped_refptr<BlobData> blob_data(new BlobData());
 
   GURL blob_url0("blob://url_0");
-  blob_storage_controller->RegisterBlobUrl(blob_url0, blob_data);
+  RegisterBlobUrl(&blob_storage_controller, blob_url0, blob_data);
 
   blob_data->AppendData("BlobData");
   blob_data->AppendFile(
       FilePath(FILE_PATH_LITERAL("BlobFile.txt")), 0, 20, time1);
 
   GURL blob_url1("blob://url_1");
-  blob_storage_controller->RegisterBlobUrl(blob_url1, blob_data);
+  RegisterBlobUrl(&blob_storage_controller, blob_url1, blob_data);
 
   GURL blob_url2("blob://url_2");
-  blob_storage_controller->RegisterBlobUrlFrom(blob_url2, blob_url1);
+  blob_storage_controller.RegisterBlobUrlFrom(blob_url2, blob_url1);
 
   GURL blob_url3("blob://url_3");
-  blob_storage_controller->RegisterBlobUrlFrom(blob_url3, blob_url2);
+  blob_storage_controller.RegisterBlobUrlFrom(blob_url3, blob_url2);
 
   // Setup upload data elements for comparison.
   UploadData::Element blob_element1, blob_element2;
   blob_element1.SetToBytes(
       blob_data->items().at(0).data().c_str() +
           static_cast<int>(blob_data->items().at(0).offset()),
       static_cast<int>(blob_data->items().at(0).length()));
   blob_element2.SetToFilePathRange(
       blob_data->items().at(1).file_path(),
       blob_data->items().at(1).offset(),
       blob_data->items().at(1).length(),
       blob_data->items().at(1).expected_modification_time());
 
   UploadData::Element upload_element1, upload_element2;
   upload_element1.SetToBytes("Hello", 5);
   upload_element2.SetToFilePathRange(
       FilePath(FILE_PATH_LITERAL("foo1.txt")), 0, 20, time2);
 
   // Test no blob reference.
   scoped_refptr<UploadData> upload_data(new UploadData());
   upload_data->AppendBytes(
       &upload_element1.bytes().at(0),
       upload_element1.bytes().size());
   upload_data->AppendFileRange(
       upload_element2.file_path(),
       upload_element2.file_range_offset(),
       upload_element2.file_range_length(),
       upload_element2.expected_file_modification_time());
 
-  blob_storage_controller->ResolveBlobReferencesInUploadData(upload_data.get());
+  blob_storage_controller.ResolveBlobReferencesInUploadData(upload_data.get());
   ASSERT_EQ(upload_data->elements()->size(), 2U);
   EXPECT_TRUE(upload_data->elements()->at(0) == upload_element1);
   EXPECT_TRUE(upload_data->elements()->at(1) == upload_element2);
 
   // Test having only one blob reference that refers to empty blob data.
   upload_data = new UploadData();
   upload_data->AppendBlob(blob_url0);
 
-  blob_storage_controller->ResolveBlobReferencesInUploadData(upload_data.get());
+  blob_storage_controller.ResolveBlobReferencesInUploadData(upload_data.get());
   ASSERT_EQ(upload_data->elements()->size(), 0U);
 
   // Test having only one blob reference.
   upload_data = new UploadData();
   upload_data->AppendBlob(blob_url1);
 
-  blob_storage_controller->ResolveBlobReferencesInUploadData(upload_data.get());
+  blob_storage_controller.ResolveBlobReferencesInUploadData(upload_data.get());
   ASSERT_EQ(upload_data->elements()->size(), 2U);
   EXPECT_TRUE(upload_data->elements()->at(0) == blob_element1);
   EXPECT_TRUE(upload_data->elements()->at(1) == blob_element2);
 
   // Test having one blob reference at the beginning.
   upload_data = new UploadData();
   upload_data->AppendBlob(blob_url1);
   upload_data->AppendBytes(
       &upload_element1.bytes().at(0),
       upload_element1.bytes().size());
   upload_data->AppendFileRange(
       upload_element2.file_path(),
       upload_element2.file_range_offset(),
       upload_element2.file_range_length(),
       upload_element2.expected_file_modification_time());
 
-  blob_storage_controller->ResolveBlobReferencesInUploadData(upload_data.get());
+  blob_storage_controller.ResolveBlobReferencesInUploadData(upload_data.get());
   ASSERT_EQ(upload_data->elements()->size(), 4U);
   EXPECT_TRUE(upload_data->elements()->at(0) == blob_element1);
   EXPECT_TRUE(upload_data->elements()->at(1) == blob_element2);
   EXPECT_TRUE(upload_data->elements()->at(2) == upload_element1);
   EXPECT_TRUE(upload_data->elements()->at(3) == upload_element2);
 
   // Test having one blob reference at the end.
   upload_data = new UploadData();
   upload_data->AppendBytes(
       &upload_element1.bytes().at(0),
       upload_element1.bytes().size());
   upload_data->AppendFileRange(
       upload_element2.file_path(),
       upload_element2.file_range_offset(),
       upload_element2.file_range_length(),
       upload_element2.expected_file_modification_time());
   upload_data->AppendBlob(blob_url1);
 
-  blob_storage_controller->ResolveBlobReferencesInUploadData(upload_data.get());
+  blob_storage_controller.ResolveBlobReferencesInUploadData(upload_data.get());
   ASSERT_EQ(upload_data->elements()->size(), 4U);
   EXPECT_TRUE(upload_data->elements()->at(0) == upload_element1);
   EXPECT_TRUE(upload_data->elements()->at(1) == upload_element2);
   EXPECT_TRUE(upload_data->elements()->at(2) == blob_element1);
   EXPECT_TRUE(upload_data->elements()->at(3) == blob_element2);
 
   // Test having one blob reference in the middle.
   upload_data = new UploadData();
   upload_data->AppendBytes(
       &upload_element1.bytes().at(0),
       upload_element1.bytes().size());
   upload_data->AppendBlob(blob_url1);
   upload_data->AppendFileRange(
       upload_element2.file_path(),
       upload_element2.file_range_offset(),
       upload_element2.file_range_length(),
       upload_element2.expected_file_modification_time());
 
-  blob_storage_controller->ResolveBlobReferencesInUploadData(upload_data.get());
+  blob_storage_controller.ResolveBlobReferencesInUploadData(upload_data.get());
   ASSERT_EQ(upload_data->elements()->size(), 4U);
   EXPECT_TRUE(upload_data->elements()->at(0) == upload_element1);
   EXPECT_TRUE(upload_data->elements()->at(1) == blob_element1);
   EXPECT_TRUE(upload_data->elements()->at(2) == blob_element2);
   EXPECT_TRUE(upload_data->elements()->at(3) == upload_element2);
 
   // Test having multiple blob references.
   upload_data = new UploadData();
   upload_data->AppendBlob(blob_url1);
   upload_data->AppendBytes(
       &upload_element1.bytes().at(0),
       upload_element1.bytes().size());
   upload_data->AppendBlob(blob_url2);
   upload_data->AppendBlob(blob_url3);
   upload_data->AppendFileRange(
       upload_element2.file_path(),
       upload_element2.file_range_offset(),
       upload_element2.file_range_length(),
       upload_element2.expected_file_modification_time());
 
-  blob_storage_controller->ResolveBlobReferencesInUploadData(upload_data.get());
+  blob_storage_controller.ResolveBlobReferencesInUploadData(upload_data.get());
   ASSERT_EQ(upload_data->elements()->size(), 8U);
   EXPECT_TRUE(upload_data->elements()->at(0) == blob_element1);
   EXPECT_TRUE(upload_data->elements()->at(1) == blob_element2);
   EXPECT_TRUE(upload_data->elements()->at(2) == upload_element1);
   EXPECT_TRUE(upload_data->elements()->at(3) == blob_element1);
   EXPECT_TRUE(upload_data->elements()->at(4) == blob_element2);
   EXPECT_TRUE(upload_data->elements()->at(5) == blob_element1);
   EXPECT_TRUE(upload_data->elements()->at(6) == blob_element2);
   EXPECT_TRUE(upload_data->elements()->at(7) == upload_element2);
 }
 
 }  // namespace webkit_blob