[BlobAsync] Asynchronous Blob Construction Final Patch

content/child/blob_storage/blob_transport_controller_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_controller.h"
 
 #include "base/memory/shared_memory.h"
 #include "content/child/blob_storage/blob_consolidation.h"
+#include "content/common/fileapi/webblob_messages.h"
+#include "ipc/ipc_sender.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 {
 
+class MockIPCSender : public IPC::Sender {
+ public:
+  MockIPCSender() {}
+  ~MockIPCSender() override {}
+
+  MOCK_METHOD1(Send, bool(IPC::Message* msg));
+};
+
 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) {
@@ skipping 62 matching lines @@
 
   // 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->blob_storage_.insert(
-      std::make_pair(kBlobUUID, make_scoped_ptr(consolidation)));
+  holder->blob_storage_[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,
@@ skipping 37 matching lines @@
 
   // 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->blob_storage_.insert(
-      std::make_pair(kBlobUUID, make_scoped_ptr(consolidation)));
+  holder->blob_storage_[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(
@@ skipping 17 matching lines @@
   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_F(BlobTransportControllerTest, TestPublicMethods) {
   const std::string kBlobUUID = "uuid";
   const std::string KRefBlobUUID = "refuuid";
   const std::string kBadBlobUUID = "uuuidBad";
+  MockIPCSender mock_sender;
+
   BlobTransportController* holder = BlobTransportController::GetInstance();
 
   BlobConsolidation* consolidation = new BlobConsolidation();
   consolidation->AddBlobItem(KRefBlobUUID, 10, 10);
+  IPC::Message* message = nullptr;
+  EXPECT_CALL(mock_sender, Send(testing::_))
+      .WillOnce(
+          testing::DoAll(testing::SaveArg<0>(&message), testing::Return(true)))
+      .RetiresOnSaturation();
   holder->InitiateBlobTransfer(kBlobUUID, "", make_scoped_ptr(consolidation),
-                               nullptr);
+                               &mock_sender);
+  EXPECT_NE(nullptr, message);
   holder->OnCancel(kBlobUUID,
                    storage::IPCBlobCreationCancelCode::OUT_OF_MEMORY);
   EXPECT_FALSE(holder->IsTransporting(kBlobUUID));
+  delete ((BlobStorageMsg_StartBuildingBlob*)message);
+  message = nullptr;
 
   consolidation = new BlobConsolidation();
   consolidation->AddBlobItem(KRefBlobUUID, 10, 10);
+  EXPECT_CALL(mock_sender, Send(testing::_))
+      .WillOnce(
+          testing::DoAll(testing::SaveArg<0>(&message), testing::Return(true)))
+      .RetiresOnSaturation();
   holder->InitiateBlobTransfer(kBlobUUID, "", make_scoped_ptr(consolidation),
-                               nullptr);
+                               &mock_sender);
+  EXPECT_NE(nullptr, message);
   holder->OnDone(kBlobUUID);
   EXPECT_FALSE(holder->IsTransporting(kBlobUUID));
+  delete ((BlobStorageMsg_StartBuildingBlob*)message);
 }
 
 TEST_F(BlobTransportControllerTest, ResponsesErrors) {
   using ResponsesStatus = BlobTransportController::ResponsesStatus;
   const std::string kBlobUUID = "uuid";
   const std::string KRefBlobUUID = "refuuid";
   const std::string kBadBlobUUID = "uuuidBad";
   BlobTransportController* holder = BlobTransportController::GetInstance();
 
   BlobConsolidation* consolidation = new BlobConsolidation();
   consolidation->AddBlobItem(KRefBlobUUID, 10, 10);
 
-  holder->blob_storage_.insert(
-      std::make_pair(kBlobUUID, make_scoped_ptr(consolidation)));
+  holder->blob_storage_[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));
   EXPECT_EQ(0u, output.size());
 }
 
 }  // namespace content