[BlobAsync] Patch 3: Renderer Classes & Logic

content/child/blob_storage/blob_transport_temporary_holder_unittest.cc

+// Copyright 2015 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/child/blob_storage/blob_transport_temporary_holder.h"
+
+#include "base/memory/shared_memory.h"
+#include "content/child/blob_storage/blob_consolidation.h"
+#include "storage/common/blob_storage/blob_item_bytes_request.h"
+#include "storage/common/blob_storage/blob_item_bytes_response.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using storage::BlobItemBytesRequest;
+using storage::BlobItemBytesResponse;
+using storage::DataElement;
+
+namespace content {
+namespace {
+using ResponsesStatus = BlobTransportTemporaryHolder::ResponsesStatus;
+
+BlobItemBytesResponse ResponseWithData(size_t request_number,
+                                       const std::string& str) {
+  BlobItemBytesResponse response(request_number);
+  response.inline_data.assign(str.begin(), str.end());
+  return response;
+}
+
+static DataElement MakeBlobElement(const std::string& uuid,
+                                   uint64_t offset,
+                                   uint64_t size) {
+  DataElement element;
+  element.SetToBlobRange(uuid, offset, size);
+  return element;
+}
+
+static DataElement MakeDataDescriptionElement(size_t size) {
+  DataElement element;
+  element.SetToBytesDescription(size);
+  return element;
+}
+
+static DataElement MakeDataElement(const std::string& str) {
+  DataElement element;
+  element.SetToBytes(str.c_str(), str.size());
+  return element;
+}
+
+static blink::WebThreadSafeData CreateData(const std::string& str) {
+  return blink::WebThreadSafeData(str.c_str(), str.size());
+}
+
+TEST(BlobTransportTemporaryHolder, Descriptions) {
+  const std::string kBlobUUID = "uuid";
+  const std::string KRefBlobUUID = "refuuid";
+  const std::string kBadBlobUUID = "uuuidBad";
+  const size_t kShortcutSize = 11;
+  BlobTransportTemporaryHolder holder;
+
+  // The first two data elements should be combined and the data shortcut.
+  BlobConsolidation* consolidation = new BlobConsolidation();
+  consolidation->AddBlobItem(KRefBlobUUID, 10, 10);
+  consolidation->AddDataItem(CreateData("Hello"));
+  consolidation->AddDataItem(CreateData("Hello2"));
+  consolidation->AddBlobItem(KRefBlobUUID, 0, 10);
+  consolidation->AddDataItem(CreateData("Hello3"));
+
+  holder.HoldBlobConsolidation(kBlobUUID, make_scoped_ptr(consolidation));
+
+  std::vector<DataElement> out;
+  EXPECT_FALSE(holder.GetDescriptions(kBadBlobUUID, kShortcutSize, &out));
+  EXPECT_TRUE(holder.GetDescriptions(kBlobUUID, kShortcutSize, &out));
+
+  std::vector<DataElement> expected = {
+      MakeBlobElement(KRefBlobUUID, 10, 10), MakeDataElement("HelloHello2"),
+      MakeBlobElement(KRefBlobUUID, 0, 10), MakeDataDescriptionElement(6)};
+  EXPECT_EQ(expected, out);
+}
+
+TEST(BlobTransportTemporaryHolder, Responses) {
+  const std::string kBlobUUID = "uuid";
+  const std::string KRefBlobUUID = "refuuid";
+  const std::string kBadBlobUUID = "uuuidBad";
+  BlobTransportTemporaryHolder holder;
+
+  // The first two data elements should be combined.
+  BlobConsolidation* consolidation = new BlobConsolidation();
+  consolidation->AddBlobItem(KRefBlobUUID, 10, 10);
+  consolidation->AddDataItem(CreateData("Hello"));
+  consolidation->AddDataItem(CreateData("Hello2"));
+  consolidation->AddBlobItem(KRefBlobUUID, 0, 10);
+  consolidation->AddDataItem(CreateData("Hello3"));
+  // See the above test for the expected descriptions layout.
+
+  holder.HoldBlobConsolidation(kBlobUUID, make_scoped_ptr(consolidation));
+
+  std::vector<BlobItemBytesRequest> requests;
+  std::vector<base::SharedMemoryHandle> memory_handles;
+  std::vector<IPC::PlatformFileForTransit> file_handles;
+  std::vector<storage::BlobItemBytesResponse> output;
+
+  // Request for all of first data
+  requests.push_back(BlobItemBytesRequest::CreateIPCRequest(0, 1, 0, 11));
+  EXPECT_EQ(ResponsesStatus::SUCCESS,
+            holder.GetResponses(kBlobUUID, requests, &memory_handles,
+                                file_handles, &output));
+  EXPECT_EQ(1u, output.size());
+  std::vector<storage::BlobItemBytesResponse> expected = {
+      ResponseWithData(0, "HelloHello2")};
+  EXPECT_EQ(expected, output);
+
+  // Part of second data
+  output.clear();
+  requests[0] = BlobItemBytesRequest::CreateIPCRequest(1000, 3, 1, 5);
+  EXPECT_EQ(ResponsesStatus::SUCCESS,
+            holder.GetResponses(kBlobUUID, requests, &memory_handles,
+                                file_handles, &output));
+  EXPECT_EQ(1u, output.size());
+  expected = {ResponseWithData(1000, "ello3")};
+  EXPECT_EQ(expected, output);
+
+  // Both data segments
+  output.clear();
+  requests[0] = BlobItemBytesRequest::CreateIPCRequest(0, 1, 0, 11);
+  requests.push_back(BlobItemBytesRequest::CreateIPCRequest(1, 3, 0, 6));
+  EXPECT_EQ(ResponsesStatus::SUCCESS,
+            holder.GetResponses(kBlobUUID, requests, &memory_handles,
+                                file_handles, &output));
+  EXPECT_EQ(2u, output.size());
+  expected = {ResponseWithData(0, "HelloHello2"),
+              ResponseWithData(1, "Hello3")};
+  EXPECT_EQ(expected, output);
+}
+
+TEST(BlobTransportTemporaryHolder, SharedMemory) {
+  const std::string kBlobUUID = "uuid";
+  const std::string KRefBlobUUID = "refuuid";
+  const std::string kBadBlobUUID = "uuuidBad";
+  BlobTransportTemporaryHolder holder;
+
+  // The first two data elements should be combined.
+  BlobConsolidation* consolidation = new BlobConsolidation();
+  consolidation->AddBlobItem(KRefBlobUUID, 10, 10);
+  consolidation->AddDataItem(CreateData("Hello"));
+  consolidation->AddDataItem(CreateData("Hello2"));
+  consolidation->AddBlobItem(KRefBlobUUID, 0, 10);
+  consolidation->AddDataItem(CreateData("Hello3"));
+  // See the above test for the expected descriptions layout.
+
+  holder.HoldBlobConsolidation(kBlobUUID, make_scoped_ptr(consolidation));
+
+  std::vector<BlobItemBytesRequest> requests;
+  std::vector<base::SharedMemoryHandle> memory_handles;
+  std::vector<IPC::PlatformFileForTransit> file_handles;
+  std::vector<storage::BlobItemBytesResponse> output;
+
+  // Request for all data in shared memory
+  requests.push_back(
+      BlobItemBytesRequest::CreateSharedMemoryRequest(0, 1, 0, 11, 0, 0));
+  requests.push_back(
+      BlobItemBytesRequest::CreateSharedMemoryRequest(1, 3, 0, 6, 0, 11));
+  base::SharedMemory memory;
+  memory.CreateAndMapAnonymous(11 + 6);
+  base::SharedMemoryHandle handle =
+      base::SharedMemory::DuplicateHandle(memory.handle());
+  CHECK(base::SharedMemory::NULLHandle() != handle);
+  memory_handles.push_back(handle);
+
+  EXPECT_EQ(ResponsesStatus::SUCCESS,
+            holder.GetResponses(kBlobUUID, requests, &memory_handles,
+                                file_handles, &output));
+  EXPECT_EQ(2u, output.size());
+  std::vector<storage::BlobItemBytesResponse> expected = {
+      BlobItemBytesResponse(0), BlobItemBytesResponse(1)};
+  EXPECT_EQ(expected, output);
+  std::string expected_memory = "HelloHello2Hello3";
+  const char* mem_location = static_cast<const char*>(memory.memory());
+  std::vector<char> value(mem_location, mem_location + memory.requested_size());
+  EXPECT_THAT(value, testing::ElementsAreArray(expected_memory.c_str(),
+                                               expected_memory.size()));
+}
+
+TEST(BlobTransportTemporaryHolder, ResponsesErrors) {
+  const std::string kBlobUUID = "uuid";
+  const std::string KRefBlobUUID = "refuuid";
+  const std::string kBadBlobUUID = "uuuidBad";
+  BlobTransportTemporaryHolder holder;
+
+  // The first two data elements should be combined.
+  BlobConsolidation* consolidation = new BlobConsolidation();
+  consolidation->AddBlobItem(KRefBlobUUID, 10, 10);
+  consolidation->AddDataItem(CreateData("Hello"));
+  consolidation->AddDataItem(CreateData("Hello2"));
+  consolidation->AddBlobItem(KRefBlobUUID, 0, 10);
+  consolidation->AddDataItem(CreateData("Hello3"));
+  // See the above test for the expected descriptions layout.
+
+  holder.HoldBlobConsolidation(kBlobUUID, make_scoped_ptr(consolidation));
+
+  std::vector<BlobItemBytesRequest> requests;
+  std::vector<base::SharedMemoryHandle> memory_handles;
+  std::vector<IPC::PlatformFileForTransit> file_handles;
+  std::vector<storage::BlobItemBytesResponse> output;
+
+  // Error conditions
+  EXPECT_EQ(ResponsesStatus::BLOB_NOT_FOUND,
+            holder.GetResponses(kBadBlobUUID, requests, &memory_handles,
+                                file_handles, &output));
+
+  requests.push_back(BlobItemBytesRequest::CreateIPCRequest(0, 5, 0, 10));
+  EXPECT_EQ(ResponsesStatus::INVALID_ITEM_INDEX,
+            holder.GetResponses(kBlobUUID, requests, &memory_handles,
+                                file_handles, &output));
+  EXPECT_EQ(0u, output.size());
+
+  requests[0] = BlobItemBytesRequest::CreateIPCRequest(0, 0, 0, 10);
+  EXPECT_EQ(ResponsesStatus::INVALID_ITEM,
+            holder.GetResponses(kBlobUUID, requests, &memory_handles,
+                                file_handles, &output));
+  EXPECT_EQ(0u, output.size());
+
+  requests[0] = BlobItemBytesRequest::CreateIPCRequest(0, 1, 0, 20);
+  EXPECT_EQ(ResponsesStatus::INVALID_DATA_RANGE,
+            holder.GetResponses(kBlobUUID, requests, &memory_handles,
+                                file_handles, &output));
+  EXPECT_EQ(0u, output.size());
+}
+
+}  // namespace
+}  // namespace content