1 files changed, 81 insertions, 0 deletions
diff --git a/chromium/storage/browser/blob/blob_storage_registry_unittest.cc b/chromium/storage/browser/blob/blob_storage_registry_unittest.cc
new file mode 100644
index 00000000000..9cb2285f11f
--- /dev/null
+++ b/chromium/storage/browser/blob/blob_storage_registry_unittest.cc
@@ -0,0 +1,81 @@
+// 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 "storage/browser/blob/blob_storage_registry.h"
+
+#include "base/callback.h"
+#include "storage/browser/blob/blob_entry.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "url/gurl.h"
+
+namespace storage {
+namespace {
+
+TEST(BlobStorageRegistry, UUIDRegistration) {
+  const std::string kBlob1 = "Blob1";
+  const std::string kType = "type1";
+  const std::string kDisposition = "disp1";
+  BlobStorageRegistry registry;
+
+  EXPECT_FALSE(registry.DeleteEntry(kBlob1));
+  EXPECT_EQ(0u, registry.blob_count());
+
+  BlobEntry* entry = registry.CreateEntry(kBlob1, kType, kDisposition);
+  ASSERT_NE(nullptr, entry);
+  EXPECT_EQ(BlobStatus::PENDING_QUOTA, entry->status());
+  EXPECT_EQ(kType, entry->content_type());
+  EXPECT_EQ(kDisposition, entry->content_disposition());
+  EXPECT_EQ(0u, entry->refcount());
+
+  EXPECT_EQ(entry, registry.GetEntry(kBlob1));
+  EXPECT_TRUE(registry.DeleteEntry(kBlob1));
+  entry = registry.CreateEntry(kBlob1, kType, kDisposition);
+
+  EXPECT_EQ(1u, registry.blob_count());
+}
+
+TEST(BlobStorageRegistry, URLRegistration) {
+  const std::string kBlob = "Blob1";
+  const std::string kType = "type1";
+  const std::string kDisposition = "disp1";
+  const std::string kBlob2 = "Blob2";
+  const GURL kURL = GURL("blob://Blob1");
+  const GURL kURL2 = GURL("blob://Blob2");
+  BlobStorageRegistry registry;
+
+  EXPECT_FALSE(registry.IsURLMapped(kURL));
+  EXPECT_EQ(nullptr, registry.GetEntryFromURL(kURL, nullptr));
+  EXPECT_FALSE(registry.DeleteURLMapping(kURL, nullptr));
+  EXPECT_FALSE(registry.CreateUrlMapping(kURL, kBlob));
+  EXPECT_EQ(0u, registry.url_count());
+  BlobEntry* entry = registry.CreateEntry(kBlob, kType, kDisposition);
+
+  EXPECT_FALSE(registry.IsURLMapped(kURL));
+  EXPECT_TRUE(registry.CreateUrlMapping(kURL, kBlob));
+  EXPECT_FALSE(registry.CreateUrlMapping(kURL, kBlob2));
+
+  EXPECT_TRUE(registry.IsURLMapped(kURL));
+  EXPECT_EQ(entry, registry.GetEntryFromURL(kURL, nullptr));
+  std::string uuid;
+  EXPECT_EQ(entry, registry.GetEntryFromURL(kURL, &uuid));
+  EXPECT_EQ(kBlob, uuid);
+  EXPECT_EQ(1u, registry.url_count());
+
+  registry.CreateEntry(kBlob2, kType, kDisposition);
+  EXPECT_TRUE(registry.CreateUrlMapping(kURL2, kBlob2));
+  EXPECT_EQ(2u, registry.url_count());
+  EXPECT_TRUE(registry.DeleteURLMapping(kURL2, &uuid));
+  EXPECT_EQ(kBlob2, uuid);
+  EXPECT_FALSE(registry.IsURLMapped(kURL2));
+
+  // Both urls point to the same blob.
+  EXPECT_TRUE(registry.CreateUrlMapping(kURL2, kBlob));
+  std::string uuid2;
+  EXPECT_EQ(registry.GetEntryFromURL(kURL, &uuid),
+            registry.GetEntryFromURL(kURL2, &uuid2));
+  EXPECT_EQ(uuid, uuid2);
+}
+
+}  // namespace
+}  // namespace storage