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

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

Issue 2448353002: [BlobAsync] Moving async handling into BlobStorageContext & quota out. (Closed)
Patch Set: Cleaned up more Created 4 years, 1 month 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
new file mode 100644
index 0000000000000000000000000000000000000000..6731bdf63fd3498e7616fe397cd321b010c07124
--- /dev/null
+++ b/content/browser/blob_storage/blob_flattener_unittest.cc
@@ -0,0 +1,230 @@
+// 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/files/file_path.h"
+#include "base/files/file_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_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;
+
+const char kType[] = "type";
+const char kDisposition[] = "";
+
+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(base::FilePath("kFakePath"), 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(base::FilePath(BlobDataBuilder::kFutureFileName),
+ offset, size, base::Time::Max());
+ return scoped_refptr<BlobDataItem>(new BlobDataItem(std::move(element)));
+};
+
+} // namespace
+
+class BlobFlattenerTest : public testing::Test {
+ protected:
+ using BlobFlattener = BlobStorageContext::BlobFlattener;
+
+ BlobFlattenerTest() {}
+ ~BlobFlattenerTest() override {}
+
+ void SetUp() override {
+ ASSERT_TRUE(base::CreateNewTempDirectory("BlobFlattenerTest", &temp_dir_));
+ }
+
+ void TearDown() override {
+ base::RunLoop().RunUntilIdle();
+ file_runner_->RunPendingTasks();
+ ASSERT_EQ(true, base::DeleteFile(temp_dir_, true));
+ }
+
+ 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];
+ }
+
+ base::FilePath temp_dir_;
+ scoped_refptr<TestSimpleTaskRunner> file_runner_ = new TestSimpleTaskRunner();
+
+ base::MessageLoop fake_io_message_loop;
+ BlobStorageContext context_;
+};
+
+TEST_F(BlobFlattenerTest, NoBlobItems) {
+ const std::string kBlobUUID = "kId";
+
+ BlobDataBuilder builder(kBlobUUID);
+ builder.AppendData("hi", 2u);
+ builder.AppendFile(base::FilePath("kFakePath"), 0u, 10u, base::Time::Max());
+ BlobEntry output(kType, kDisposition);
+ BlobFlattener flattener(builder, &output, registry());
+
+ EXPECT_EQ(BlobStatus::DONE, flattener.status);
+ EXPECT_EQ(0u, flattener.dependent_blobs.size());
+ EXPECT_EQ(0u, flattener.copies.size());
+ EXPECT_EQ(12u, flattener.total_size.ValueOrDie());
+ EXPECT_EQ(2u, flattener.transport_quota_needed.ValueOrDie());
+
+ 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,
+ // * sliced pending file.
+
+ context_.EnableDisk(temp_dir_, file_runner_);
+
+ 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(base::FilePath("kFakePath"), 1u, 10u, base::Time::Max());
+ file_blob = context_.AddFinishedBlob(builder);
+ }
+
+ std::unique_ptr<BlobDataHandle> future_file_blob;
+ {
+ BlobDataBuilder builder(kPendingFileBlob);
+ builder.AppendFile(base::FilePath(BlobDataBuilder::kFutureFileName), 2u, 5u,
+ base::Time::Max());
+ future_file_blob = context_.AddFinishedBlob(builder);
+ }
+
+ BlobDataBuilder builder(kBlobUUID);
+ builder.AppendData("hi", 2u);
+ builder.AppendBlob(kDataBlob, 1u, 2u);
+ builder.AppendFile(base::FilePath("kFakePath"), 3u, 5u, base::Time::Max());
+ builder.AppendBlob(kDataBlob);
+ builder.AppendBlob(kFileBlob, 1u, 3u);
+ builder.AppendFutureData(12u);
+ builder.AppendBlob(kPendingFileBlob, 1u, 2u);
+
+ BlobEntry output(kType, kDisposition);
+ BlobFlattener flattener(builder, &output, registry());
+ EXPECT_EQ(BlobStatus::PENDING, flattener.status);
+
+ EXPECT_EQ(3u, flattener.dependent_blobs.size());
+ EXPECT_EQ(31u, flattener.total_size.ValueOrDie());
+ EXPECT_EQ(14u, flattener.transport_quota_needed.ValueOrDie());
+ EXPECT_EQ(2u, flattener.copy_quota_needed.ValueOrDie());
+
+ ASSERT_EQ(7u, 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(3u, 2u), *output.items()[6]->item());
+
+ // We're copying item at index 1 and 6.
+ ASSERT_EQ(2u, 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[0].source_item_offset);
+}
+
+} // namespace storage

Powered by Google App Engine
This is Rietveld 408576698