Add database recovery for FileSystemOriginDatabase

webkit/fileapi/file_system_origin_database_unittest.cc

 // Copyright (c) 2011 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 "testing/gtest/include/gtest/gtest.h"
 
 #include <string>
 
 #include "base/file_util.h"
 #include "base/scoped_temp_dir.h"
@@ skipping 145 matching lines @@
     EXPECT_EQ(origins[1].origin, origin1);
     EXPECT_EQ(origins[1].path, path1);
   } else {
     EXPECT_EQ(origins[0].origin, origin1);
     EXPECT_EQ(origins[0].path, path1);
     EXPECT_EQ(origins[1].origin, origin0);
     EXPECT_EQ(origins[1].path, path0);
   }
 }
 
+TEST(FileSystemOriginDatabaseTest, DatabaseRecoveryTest) {
+  ScopedTempDir dir;
+  ASSERT_TRUE(dir.CreateUniqueTempDir());
+  const FilePath kDBFile = dir.path().AppendASCII("fsod.db");
+  EXPECT_FALSE(file_util::PathExists(kDBFile));
+
+  const std::string origin("example.com");
+  {
+    FilePath path;
+    FileSystemOriginDatabase database(kDBFile);
+    EXPECT_FALSE(database.HasOriginPath(origin));
+    EXPECT_TRUE(database.GetPathForOrigin(origin, &path));
+    EXPECT_FALSE(path.empty());
+    EXPECT_TRUE(database.HasOriginPath(origin));
+  }
+
+  bool created = false;
+  base::PlatformFileError error = base::PLATFORM_FILE_ERROR_FAILED;
+  base::PlatformFile file = base::CreatePlatformFile(
+      kDBFile.AppendASCII("CURRENT"),
+      base::PLATFORM_FILE_CREATE_ALWAYS | base::PLATFORM_FILE_WRITE,
+      &created, &error);
+  EXPECT_EQ(base::PLATFORM_FILE_OK, error);
+  EXPECT_TRUE(created);
+  EXPECT_TRUE(base::ClosePlatformFile(file));
+
+  FilePath path;
+  FileSystemOriginDatabase database(kDBFile);
+  EXPECT_FALSE(database.HasOriginPath(origin));
+  EXPECT_TRUE(database.GetPathForOrigin(origin, &path));
+  EXPECT_FALSE(path.empty());
+  EXPECT_TRUE(database.HasOriginPath(origin));
+}
+
 }  // namespace fileapi