Hook up DomStorageArea with a DomStorageDatabase.

webkit/dom_storage/dom_storage_area_unittest.cc

 // Copyright (c) 2012 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 "base/bind.h"
+#include "base/file_util.h"
+#include "base/message_loop.h"
+#include "base/message_loop_proxy.h"
+#include "base/scoped_temp_dir.h"
+#include "base/threading/sequenced_worker_pool.h"
+#include "base/time.h"
 #include "base/utf_string_conversions.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "webkit/dom_storage/dom_storage_area.h"
+#include "webkit/dom_storage/dom_storage_task_runner.h"
+#include "webkit/dom_storage/dom_storage_types.h"
 
 namespace dom_storage {
 
+void QuitMessageLoop() {
+  MessageLoop::current()->Quit();
+}
+
 TEST(DomStorageAreaTest, DomStorageAreaBasics) {
   const GURL kOrigin("http://dom_storage/");
   const string16 kKey(ASCIIToUTF16("key"));
   const string16 kValue(ASCIIToUTF16("value"));
   const string16 kKey2(ASCIIToUTF16("key2"));
   const string16 kValue2(ASCIIToUTF16("value2"));
 
   scoped_refptr<DomStorageArea> area(
       new DomStorageArea(1, kOrigin, FilePath(), NULL));
   string16 old_value;
@@ skipping 25 matching lines @@
   EXPECT_TRUE(area->SetItem(kKey, kValue, &old_nullable_value));
   EXPECT_NE(copy->map_.get(), area->map_.get());
   copy = area->ShallowCopy(2);
   EXPECT_EQ(copy->map_.get(), area->map_.get());
   EXPECT_NE(0u, area->Length());
   EXPECT_TRUE(area->Clear());
   EXPECT_EQ(0u, area->Length());
   EXPECT_NE(copy->map_.get(), area->map_.get());
 }
 
+TEST(DomStorageAreaTest, BackingDatabaseOpened) {
+  const int64 kNonLocalStorageNamespaceId = kLocalStorageNamespaceId + 1;
+  const GURL origin("http://www.google.com");
+  const string16 kKey = ASCIIToUTF16("test");
+  const string16 kKey2 = ASCIIToUTF16("test2");
+  const string16 kValue = ASCIIToUTF16("value");
+  ScopedTempDir temp_dir;
+  ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
+
+  // No directory, backing should be null.
+  {
+    scoped_refptr<DomStorageArea> area(
+        new DomStorageArea(kLocalStorageNamespaceId, origin, FilePath(),
+                           NULL));
+    EXPECT_EQ(NULL, area->backing_.get());
+    EXPECT_TRUE(area->initial_import_done_);
+    EXPECT_FALSE(file_util::PathExists(
+        temp_dir.path().Append(
+            DomStorageArea::DatabaseFileNameFromOrigin(origin))));
+  }
+
+  // Valid directory and origin but non-local namespace id. Backing should
+  // be null.
+  {
+    scoped_refptr<DomStorageArea> area(
+        new DomStorageArea(kNonLocalStorageNamespaceId, origin,
+                           temp_dir.path(), NULL));
+    EXPECT_EQ(NULL, area->backing_.get());
+    EXPECT_TRUE(area->initial_import_done_);
+
+    NullableString16 old_value;
+    EXPECT_TRUE(area->SetItem(kKey, kValue, &old_value));
+    ASSERT_TRUE(old_value.is_null());
+
+    // Check that saving a value has still left us without a backing database.
+    EXPECT_EQ(NULL, area->backing_.get());
+    EXPECT_FALSE(file_util::PathExists(
+        temp_dir.path().Append(
+            DomStorageArea::DatabaseFileNameFromOrigin(origin))));
+  }
+
+  // This should set up a DomStorageArea that is correctly backed to disk and
+  // uses a SequencedWorkerPool to commit the changes lazily.
+  {
+    base::SequencedWorkerPool pool(2, "dom_storage_area_test");
+    scoped_refptr<DomStorageArea> area(
+        new DomStorageArea(kLocalStorageNamespaceId, origin,
+            temp_dir.path(),
+            new DomStorageWorkerPoolTaskRunner(&pool,
+                base::MessageLoopProxy::current())));
+
+    EXPECT_TRUE(area->backing_.get());
+    EXPECT_FALSE(area->initial_import_done_);
+
+    // Need to write something to ensure that the database is created.
+    NullableString16 old_value;
+    EXPECT_TRUE(area->SetItem(kKey, kValue, &old_value));
+    ASSERT_TRUE(old_value.is_null());
+    EXPECT_TRUE(area->SetItem(kKey2, kValue, &old_value));
+    ASSERT_TRUE(old_value.is_null());
+    EXPECT_TRUE(area->initial_import_done_);
+
+    // Run the message loop to ensure that the delayed task to commit changes
+    // to disk makes it to the SequencedWorkerPool.
+    MessageLoop::current()->PostDelayedTask(
+        FROM_HERE, base::Bind(&QuitMessageLoop),
+        base::TimeDelta::FromSeconds(2));
+    MessageLoop::current()->Run();
+
+    // Now make sure that the SequencedWorkerPool finishes processing the
+    // commits.
+    pool.FlushForTesting();
+
+    EXPECT_TRUE(area->backing_->IsOpen());
+    EXPECT_EQ(2u, area->Length());
+    EXPECT_TRUE(file_util::PathExists(
+        temp_dir.path().Append(
+            DomStorageArea::DatabaseFileNameFromOrigin(origin))));
+    EXPECT_EQ(kValue, area->GetItem(kKey).string());
+    pool.Shutdown();
+  }
+}
+
+TEST(DomStorageAreaTest, TestDatabaseFilePath) {
+  EXPECT_EQ(FilePath().AppendASCII("file_path_to_0.localstorage"),
+      DomStorageArea::DatabaseFileNameFromOrigin(
+          GURL("file://path_to/index.html")));
+
+  EXPECT_EQ(FilePath().AppendASCII("https_www.google.com_0.localstorage"),
+      DomStorageArea::DatabaseFileNameFromOrigin(
+          GURL("https://www.google.com/")));
+
+  EXPECT_EQ(FilePath().AppendASCII("https_www.google.com_8080.localstorage"),
+      DomStorageArea::DatabaseFileNameFromOrigin(
+          GURL("https://www.google.com:8080")));
+}
+
 }  // namespace dom_storage