| Index: content/browser/blob_storage/blob_flattener_unittest.cc
|
| diff --git a/content/browser/blob_storage/blob_flattener_unittest.cc b/content/browser/blob_storage/blob_flattener_unittest.cc
|
| index b67066aa51745a8c90c4793866a993fb834b7462..3e70dffc124387cd9db33c221376b10bd9bb01dd 100644
|
| --- a/content/browser/blob_storage/blob_flattener_unittest.cc
|
| +++ b/content/browser/blob_storage/blob_flattener_unittest.cc
|
| @@ -6,11 +6,9 @@
|
|
|
| #include <memory>
|
|
|
| -#include "base/bind.h"
|
| #include "base/files/file_path.h"
|
| #include "base/files/file_util.h"
|
| #include "base/files/scoped_temp_dir.h"
|
| -#include "base/memory/ptr_util.h"
|
| #include "base/message_loop/message_loop.h"
|
| #include "base/run_loop.h"
|
| #include "base/test/test_simple_task_runner.h"
|
| @@ -19,7 +17,6 @@
|
| #include "storage/browser/blob/blob_data_handle.h"
|
| #include "storage/browser/blob/blob_data_item.h"
|
| #include "storage/browser/blob/blob_entry.h"
|
| -#include "storage/browser/blob/blob_memory_controller.h"
|
| #include "storage/browser/blob/blob_storage_registry.h"
|
| #include "storage/browser/blob/shareable_blob_data_item.h"
|
| #include "storage/common/data_element.h"
|
| @@ -28,26 +25,9 @@
|
| namespace storage {
|
| namespace {
|
| using base::TestSimpleTaskRunner;
|
| -using FileCreationInfo = BlobMemoryController::FileCreationInfo;
|
|
|
| const char kType[] = "type";
|
| const char kDisposition[] = "";
|
| -const size_t kTestBlobStorageIPCThresholdBytes = 20;
|
| -const size_t kTestBlobStorageMaxSharedMemoryBytes = 50;
|
| -
|
| -const size_t kTestBlobStorageMaxBlobMemorySize = 400;
|
| -const uint64_t kTestBlobStorageMaxDiskSpace = 4000;
|
| -const uint64_t kTestBlobStorageMinFileSizeBytes = 10;
|
| -const uint64_t kTestBlobStorageMaxFileSizeBytes = 100;
|
| -
|
| -void SaveBlobStatusAndFiles(BlobStatus* status_ptr,
|
| - std::vector<FileCreationInfo>* files_ptr,
|
| - BlobStatus status,
|
| - std::vector<FileCreationInfo> files) {
|
| - EXPECT_FALSE(BlobStatusIsError(status));
|
| - *status_ptr = status;
|
| - std::move(files.begin(), files.end(), std::back_inserter(*files_ptr));
|
| -}
|
|
|
| } // namespace
|
|
|
| @@ -59,15 +39,11 @@
|
| : fake_file_path_(base::FilePath(FILE_PATH_LITERAL("kFakePath"))) {}
|
| ~BlobFlattenerTest() override {}
|
|
|
| - void SetUp() override {
|
| - ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
|
| - context_ = base::MakeUnique<BlobStorageContext>();
|
| - }
|
| + void SetUp() override { ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); }
|
|
|
| void TearDown() override {
|
| base::RunLoop().RunUntilIdle();
|
| file_runner_->RunPendingTasks();
|
| - base::RunLoop().RunUntilIdle();
|
| ASSERT_TRUE(temp_dir_.Delete());
|
| }
|
|
|
| @@ -90,21 +66,14 @@
|
| return scoped_refptr<BlobDataItem>(new BlobDataItem(std::move(element)));
|
| };
|
|
|
| - scoped_refptr<BlobDataItem> CreateFutureFileItem(size_t offset, size_t size) {
|
| - std::unique_ptr<DataElement> element(new DataElement());
|
| - element->SetToFilePathRange(BlobDataBuilder::GetFutureFileItemPath(0),
|
| - offset, size, base::Time());
|
| - return scoped_refptr<BlobDataItem>(new BlobDataItem(std::move(element)));
|
| - };
|
| -
|
| std::unique_ptr<BlobDataHandle> SetupBasicBlob(const std::string& id) {
|
| BlobDataBuilder builder(id);
|
| builder.AppendData("1", 1);
|
| builder.set_content_type("text/plain");
|
| - return context_->AddFinishedBlob(builder);
|
| - }
|
| -
|
| - BlobStorageRegistry* registry() { return context_->mutable_registry(); }
|
| + return context_.AddFinishedBlob(builder);
|
| + }
|
| +
|
| + BlobStorageRegistry* registry() { return context_.mutable_registry(); }
|
|
|
| const ShareableBlobDataItem& GetItemInBlob(const std::string& uuid,
|
| size_t index) {
|
| @@ -113,23 +82,12 @@
|
| return *entry->items()[index];
|
| }
|
|
|
| - void SetTestMemoryLimits() {
|
| - BlobStorageLimits limits;
|
| - limits.max_ipc_memory_size = kTestBlobStorageIPCThresholdBytes;
|
| - limits.max_shared_memory_size = kTestBlobStorageMaxSharedMemoryBytes;
|
| - limits.max_blob_in_memory_space = kTestBlobStorageMaxBlobMemorySize;
|
| - limits.max_blob_disk_space = kTestBlobStorageMaxDiskSpace;
|
| - limits.min_page_file_size = kTestBlobStorageMinFileSizeBytes;
|
| - limits.max_file_size = kTestBlobStorageMaxFileSizeBytes;
|
| - context_->mutable_memory_controller()->set_limits_for_testing(limits);
|
| - }
|
| -
|
| base::FilePath fake_file_path_;
|
| base::ScopedTempDir temp_dir_;
|
| scoped_refptr<TestSimpleTaskRunner> file_runner_ = new TestSimpleTaskRunner();
|
|
|
| base::MessageLoop fake_io_message_loop;
|
| - std::unique_ptr<BlobStorageContext> context_;
|
| + BlobStorageContext context_;
|
| };
|
|
|
| TEST_F(BlobFlattenerTest, NoBlobItems) {
|
| @@ -145,7 +103,7 @@
|
| EXPECT_EQ(0u, flattener.dependent_blobs.size());
|
| EXPECT_EQ(0u, flattener.copies.size());
|
| EXPECT_EQ(12u, flattener.total_size);
|
| - EXPECT_EQ(2u, flattener.transport_quota_needed);
|
| + EXPECT_EQ(2u, flattener.memory_quota_needed);
|
|
|
| ASSERT_EQ(2u, output.items().size());
|
| EXPECT_EQ(*CreateDataItem("hi", 2u), *output.items()[0]->item());
|
| @@ -189,7 +147,6 @@
|
| const std::string kBlobUUID = "kId";
|
| const std::string kDataBlob = "kId2";
|
| const std::string kFileBlob = "kId3";
|
| - const std::string kPendingFileBlob = "kId4";
|
|
|
| // We have the following:
|
| // * data,
|
| @@ -198,35 +155,19 @@
|
| // * full data blob,
|
| // * pending data,
|
|
|
| - context_ =
|
| - base::MakeUnique<BlobStorageContext>(temp_dir_.GetPath(), file_runner_);
|
| - SetTestMemoryLimits();
|
| -
|
| std::unique_ptr<BlobDataHandle> data_blob;
|
| {
|
| BlobDataBuilder builder(kDataBlob);
|
| builder.AppendData("12345", 5);
|
| builder.set_content_type("text/plain");
|
| - data_blob = context_->AddFinishedBlob(builder);
|
| + data_blob = context_.AddFinishedBlob(builder);
|
| }
|
|
|
| std::unique_ptr<BlobDataHandle> file_blob;
|
| {
|
| BlobDataBuilder builder(kFileBlob);
|
| builder.AppendFile(fake_file_path_, 1u, 10u, base::Time::Max());
|
| - file_blob = context_->AddFinishedBlob(builder);
|
| - }
|
| -
|
| - BlobStatus file_status = BlobStatus::ERR_INVALID_CONSTRUCTION_ARGUMENTS;
|
| - std::vector<FileCreationInfo> file_handles;
|
| - std::unique_ptr<BlobDataHandle> future_file_blob;
|
| - {
|
| - BlobDataBuilder builder(kPendingFileBlob);
|
| - builder.AppendFutureFile(0u, 2u, 0);
|
| - builder.AppendFutureFile(2u, 5u, 0);
|
| - future_file_blob = context_->BuildBlob(
|
| - builder,
|
| - base::Bind(&SaveBlobStatusAndFiles, &file_status, &file_handles));
|
| + file_blob = context_.AddFinishedBlob(builder);
|
| }
|
|
|
| BlobDataBuilder builder(kBlobUUID);
|
| @@ -236,47 +177,28 @@
|
| builder.AppendBlob(kDataBlob);
|
| builder.AppendBlob(kFileBlob, 1u, 3u);
|
| builder.AppendFutureData(12u);
|
| - builder.AppendBlob(kPendingFileBlob, 1u, 3u);
|
|
|
| BlobEntry output(kType, kDisposition);
|
| BlobFlattener flattener(builder, &output, registry());
|
| EXPECT_EQ(BlobStatus::PENDING_QUOTA, flattener.status);
|
|
|
| - EXPECT_EQ(3u, flattener.dependent_blobs.size());
|
| - EXPECT_EQ(32u, flattener.total_size);
|
| - EXPECT_EQ(14u, flattener.transport_quota_needed);
|
| - EXPECT_EQ(2u, flattener.copy_quota_needed);
|
| -
|
| - ASSERT_EQ(8u, output.items().size());
|
| + EXPECT_EQ(2u, flattener.dependent_blobs.size());
|
| + EXPECT_EQ(29u, flattener.total_size);
|
| + EXPECT_EQ(16u, flattener.memory_quota_needed);
|
| +
|
| + ASSERT_EQ(6u, output.items().size());
|
| EXPECT_EQ(*CreateDataItem("hi", 2u), *output.items()[0]->item());
|
| EXPECT_EQ(*CreateDataDescriptionItem(2u), *output.items()[1]->item());
|
| EXPECT_EQ(*CreateFileItem(3u, 5u), *output.items()[2]->item());
|
| EXPECT_EQ(GetItemInBlob(kDataBlob, 0), *output.items()[3]);
|
| EXPECT_EQ(*CreateFileItem(2u, 3u), *output.items()[4]->item());
|
| EXPECT_EQ(*CreateDataDescriptionItem(12u), *output.items()[5]->item());
|
| - EXPECT_EQ(*CreateFutureFileItem(1u, 1u), *output.items()[6]->item());
|
| - EXPECT_EQ(*CreateFutureFileItem(2u, 2u), *output.items()[7]->item());
|
| -
|
| - // We're copying items at index 1, 6, and 7.
|
| - ASSERT_EQ(3u, flattener.copies.size());
|
| +
|
| + // We're copying item at index 1
|
| + ASSERT_EQ(1u, flattener.copies.size());
|
| EXPECT_EQ(*flattener.copies[0].dest_item, *output.items()[1]);
|
| EXPECT_EQ(GetItemInBlob(kDataBlob, 0), *flattener.copies[0].source_item);
|
| EXPECT_EQ(1u, flattener.copies[0].source_item_offset);
|
| - EXPECT_EQ(*flattener.copies[1].dest_item, *output.items()[6]);
|
| - EXPECT_EQ(GetItemInBlob(kPendingFileBlob, 0),
|
| - *flattener.copies[1].source_item);
|
| - EXPECT_EQ(1u, flattener.copies[1].source_item_offset);
|
| - EXPECT_EQ(*flattener.copies[2].dest_item, *output.items()[7]);
|
| - EXPECT_EQ(GetItemInBlob(kPendingFileBlob, 1),
|
| - *flattener.copies[2].source_item);
|
| - EXPECT_EQ(0u, flattener.copies[2].source_item_offset);
|
| -
|
| - // Clean up temp files.
|
| - EXPECT_TRUE(file_runner_->HasPendingTask());
|
| - file_runner_->RunPendingTasks();
|
| - base::RunLoop().RunUntilIdle();
|
| - EXPECT_EQ(BlobStatus::PENDING_TRANSPORT, file_status);
|
| - EXPECT_FALSE(file_handles.empty());
|
| }
|
|
|
| } // namespace storage
|
|
|