AppCache: Run a quick integrity check on the sqlite database when opening.

webkit/browser/appcache/appcache_database_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/files/scoped_temp_dir.h"
 #include "base/strings/stringprintf.h"
 #include "sql/connection.h"
 #include "sql/meta_table.h"
@@ skipping 55 matching lines @@
   EXPECT_TRUE(base::DirectoryExists(kNestedDir));
   EXPECT_TRUE(base::PathExists(kOtherFile));
 
   EXPECT_TRUE(db.DeleteExistingAndCreateNewDatabase());
 
   EXPECT_TRUE(base::PathExists(kDbFile));
   EXPECT_FALSE(base::DirectoryExists(kNestedDir));
   EXPECT_FALSE(base::PathExists(kOtherFile));
 }
 
+#ifdef NDEBUG
+// Only run in release builds because sql::Connection and familiy
+// crank up DLOG(FATAL)'ness and this test presents it with
+// intentionally bad data which causes debug builds to exit instead
+// of run to completion. In release builds, errors the are deliverd
+// to the consumer so  we can test the error handling of the consumer.
+TEST(AppCacheDatabaseTest, QuickIntegrityCheck) {
+  // Real files on disk for this test too, a corrupt database file.
+  base::ScopedTempDir temp_dir;
+  ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
+  const base::FilePath kDbFile = temp_dir.path().AppendASCII("appcache.db");
+  const std::string kCorruptData("deadbeef");
+  EXPECT_EQ(static_cast<int>(kCorruptData.length()),
+            file_util::WriteFile(
+                kDbFile, kCorruptData.data(), kCorruptData.length()));
+  const base::FilePath kOtherFile = temp_dir.path().AppendASCII("other_file");
+  EXPECT_EQ(3, file_util::WriteFile(kOtherFile, "foo", 3));
+
+  // See that the sql::Connection opens, but fails to pass the integrity check.
+  sql::Connection raw_db;
+  std::vector<std::string> messages;
+  EXPECT_TRUE(raw_db.Open(kDbFile));
+  EXPECT_FALSE(raw_db.QuickIntegrityCheck(&messages));
+  raw_db.Close();
+
+  // Should delete the corrupt data and start over.
+  AppCacheDatabase db(kDbFile);
+  EXPECT_TRUE(db.LazyOpen(true));
+  EXPECT_FALSE(base::PathExists(kOtherFile));
+  EXPECT_TRUE(base::PathExists(kDbFile));
+  db.CloseConnection();
+
+  // See that the Connection now passes its integrity check.
+  EXPECT_TRUE(raw_db.Open(kDbFile));
+  EXPECT_TRUE(raw_db.QuickIntegrityCheck(&messages));
+  ASSERT_EQ(1u, messages.size());
+  EXPECT_EQ(std::string("ok"), messages[0]);
+}
+#endif  // NDEBUG
+
 TEST(AppCacheDatabaseTest, ExperimentalFlags) {
   const char kExperimentFlagsKey[] = "ExperimentFlags";
   std::string kInjectedFlags("exp1,exp2");
 
   // Real files on disk for this test.
   base::ScopedTempDir temp_dir;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   const base::FilePath kDbFile = temp_dir.path().AppendASCII("appcache.db");
   const base::FilePath kOtherFile =  temp_dir.path().AppendASCII("other_file");
   EXPECT_EQ(3, file_util::WriteFile(kOtherFile, "foo", 3));
@@ skipping 1041 matching lines @@
     EXPECT_EQ(expected_namespace_url, fallbacks[i].namespace_.namespace_url);
     EXPECT_EQ(expected_target_url, fallbacks[i].namespace_.target_url);
     EXPECT_FALSE(fallbacks[i].namespace_.is_pattern);
     EXPECT_EQ(expected_whitelist_url, whitelists[i].namespace_url);
     EXPECT_FALSE(whitelists[i].is_pattern);
   }
 }
 #endif  // !APPCACHE_USE_SIMPLE_CACHE
 
 }  // namespace appcache