Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(215)

Unified Diff: content/browser/blob_storage/blob_flattener_unittest.cc

Issue 2774413002: Move some blob tests next to the files they cover. (Closed)
Patch Set: Created 3 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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
deleted file mode 100644
index 969fa8b6e1e67d7ec47a7f8d11cc3fd945b7ddb5..0000000000000000000000000000000000000000
--- a/content/browser/blob_storage/blob_flattener_unittest.cc
+++ /dev/null
@@ -1,283 +0,0 @@
-// Copyright 2016 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 "storage/browser/blob/blob_storage_context.h"
-
-#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"
-#include "base/time/time.h"
-#include "storage/browser/blob/blob_data_builder.h"
-#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"
-#include "testing/gtest/include/gtest/gtest.h"
-
-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
-
-class BlobFlattenerTest : public testing::Test {
- protected:
- using BlobFlattener = BlobStorageContext::BlobFlattener;
-
- BlobFlattenerTest()
- : fake_file_path_(base::FilePath(FILE_PATH_LITERAL("kFakePath"))) {}
- ~BlobFlattenerTest() override {}
-
- void SetUp() override {
- ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
- context_ = base::MakeUnique<BlobStorageContext>();
- }
-
- void TearDown() override {
- base::RunLoop().RunUntilIdle();
- file_runner_->RunPendingTasks();
- base::RunLoop().RunUntilIdle();
- ASSERT_TRUE(temp_dir_.Delete());
- }
-
- scoped_refptr<BlobDataItem> CreateDataDescriptionItem(size_t size) {
- std::unique_ptr<DataElement> element(new DataElement());
- element->SetToBytesDescription(size);
- return scoped_refptr<BlobDataItem>(new BlobDataItem(std::move(element)));
- };
-
- scoped_refptr<BlobDataItem> CreateDataItem(const char* memory, size_t size) {
- std::unique_ptr<DataElement> element(new DataElement());
- element->SetToBytes(memory, size);
- return scoped_refptr<BlobDataItem>(new BlobDataItem(std::move(element)));
- };
-
- scoped_refptr<BlobDataItem> CreateFileItem(size_t offset, size_t size) {
- std::unique_ptr<DataElement> element(new DataElement());
- element->SetToFilePathRange(fake_file_path_, offset, size,
- base::Time::Max());
- 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(); }
-
- const ShareableBlobDataItem& GetItemInBlob(const std::string& uuid,
- size_t index) {
- BlobEntry* entry = registry()->GetEntry(uuid);
- EXPECT_TRUE(entry);
- 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.desired_max_disk_space = kTestBlobStorageMaxDiskSpace;
- limits.effective_max_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_;
-};
-
-TEST_F(BlobFlattenerTest, NoBlobItems) {
- const std::string kBlobUUID = "kId";
-
- BlobDataBuilder builder(kBlobUUID);
- builder.AppendData("hi", 2u);
- builder.AppendFile(fake_file_path_, 0u, 10u, base::Time::Max());
- BlobEntry output(kType, kDisposition);
- BlobFlattener flattener(builder, &output, registry());
-
- EXPECT_EQ(BlobStatus::PENDING_QUOTA, flattener.status);
- 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);
-
- ASSERT_EQ(2u, output.items().size());
- EXPECT_EQ(*CreateDataItem("hi", 2u), *output.items()[0]->item());
- EXPECT_EQ(*CreateFileItem(0, 10u), *output.items()[1]->item());
-}
-
-TEST_F(BlobFlattenerTest, ErrorCases) {
- const std::string kBlobUUID = "kId";
- const std::string kBlob2UUID = "kId2";
-
- // Invalid blob reference.
- {
- BlobDataBuilder builder(kBlobUUID);
- builder.AppendBlob("doesnotexist");
- BlobEntry output(kType, kDisposition);
- BlobFlattener flattener(builder, &output, registry());
- EXPECT_EQ(BlobStatus::ERR_INVALID_CONSTRUCTION_ARGUMENTS, flattener.status);
- }
-
- // Circular reference.
- {
- BlobDataBuilder builder(kBlobUUID);
- builder.AppendBlob(kBlobUUID);
- BlobEntry output(kType, kDisposition);
- BlobFlattener flattener(builder, &output, registry());
- EXPECT_EQ(BlobStatus::ERR_INVALID_CONSTRUCTION_ARGUMENTS, flattener.status);
- }
-
- // Bad slice.
- {
- std::unique_ptr<BlobDataHandle> handle = SetupBasicBlob(kBlob2UUID);
- BlobDataBuilder builder(kBlobUUID);
- builder.AppendBlob(kBlob2UUID, 1, 2);
- BlobEntry output(kType, kDisposition);
- BlobFlattener flattener(builder, &output, registry());
- EXPECT_EQ(BlobStatus::ERR_INVALID_CONSTRUCTION_ARGUMENTS, flattener.status);
- }
-}
-
-TEST_F(BlobFlattenerTest, BlobWithSlices) {
- 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,
- // * sliced data blob,
- // * file
- // * 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);
- }
-
- 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));
- }
-
- BlobDataBuilder builder(kBlobUUID);
- builder.AppendData("hi", 2u);
- builder.AppendBlob(kDataBlob, 1u, 2u);
- builder.AppendFile(fake_file_path_, 3u, 5u, base::Time::Max());
- 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(*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());
- 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

Powered by Google App Engine
This is Rietveld 408576698