Add Dump{Quota,LastAccessTime}Table

webkit/quota/quota_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 <algorithm>
 #include <iterator>
 #include <set>
 
 #include "app/sql/connection.h"
+#include "base/bind.h"
+#include "base/callback.h"
 #include "base/file_util.h"
 #include "base/scoped_temp_dir.h"
 #include "googleurl/src/gurl.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "webkit/quota/quota_database.h"
 
 namespace {
 
 const base::Time kZeroTime;
 
@@ skipping 165 matching lines @@
     EXPECT_TRUE(db.RegisterOrigins(origins,
                                    kStorageTypeTemporary,
                                    base::Time()));
 
     used_count = -1;
     EXPECT_TRUE(db.FindOriginUsedCount(GURL("http://a/"),
                                        kStorageTypeTemporary,
                                        &used_count));
     EXPECT_EQ(1, used_count);
   }
+
+  template<typename T>
+  struct EntryVerifier {
+    std::set<T>* xs;
+    bool Run(const T& x) {
+      EXPECT_EQ(1u, xs->erase(x));
+      return true;
+    }
+  };
+
+  void DumpQuotaTable(const FilePath& kDbFile) {
+    typedef QuotaDatabase::QuotaTableEntry QuotaTableEntry;
+    typedef QuotaDatabase::QuotaTableCallback QuotaTableCallback;
+
+    QuotaTableEntry kQuotaTable[] = {
+      {"http://go/", kStorageTypeTemporary, 1},
+      {"http://oo/", kStorageTypeTemporary, 2},
+      {"http://gle/", kStorageTypePersistent, 3}
+    };
+
+    std::set<QuotaTableEntry> quota_table(
+        kQuotaTable, kQuotaTable + ARRAYSIZE_UNSAFE(kQuotaTable));
+    QuotaDatabase db(kDbFile);
+    EXPECT_TRUE(db.AssignQuotaTable(quota_table));
+
+    typedef EntryVerifier<QuotaTableEntry> Verifier;
+    Verifier verifier = {&quota_table};
+    QuotaTableCallback callback = base::Bind(&Verifier::Run,
+                                             base::Unretained(&verifier));
+    EXPECT_TRUE(db.DumpQuotaTable(&callback));
+    EXPECT_TRUE(quota_table.empty());
+  }
+
+  void DumpAccessTable(const FilePath& kDbFile) {
+    typedef QuotaDatabase::AccessTableEntry AccessTableEntry;
+    typedef QuotaDatabase::AccessTableCallback AccessTableCallback;
+
+    base::Time now(base::Time::Now());
+    AccessTableEntry kAccessTable[] = {
+      {GURL("http://go/"), kStorageTypeTemporary, 2147483647, now},
+      {GURL("http://oo/"), kStorageTypeTemporary, 0, now},
+      {GURL("http://gle/"), kStorageTypeTemporary, 1, now},
+    };
+
+    std::set<AccessTableEntry> access_table(
+        kAccessTable, kAccessTable + ARRAYSIZE_UNSAFE(kAccessTable));
+    QuotaDatabase db(kDbFile);
+    EXPECT_TRUE(db.AssignAccessTable(access_table));
+
+    typedef EntryVerifier<AccessTableEntry> Verifier;
+    Verifier verifier = {&access_table};
+    AccessTableCallback callback = base::Bind(&Verifier::Run,
+                                              base::Unretained(&verifier));
+    EXPECT_TRUE(db.DumpAccessTable(&callback));
+    EXPECT_TRUE(access_table.empty());
+  }
 };
 
 TEST_F(QuotaDatabaseTest, LazyOpen) {
   ScopedTempDir data_dir;
   ASSERT_TRUE(data_dir.CreateUniqueTempDir());
   const FilePath kDbFile = data_dir.path().AppendASCII("quota_manager.db");
   LazyOpen(kDbFile);
   LazyOpen(FilePath());
 }
 
@@ skipping 35 matching lines @@
   EXPECT_FALSE(db.IsOriginDatabaseBootstrapped());
 }
 
 TEST_F(QuotaDatabaseTest, RegisterOrigins) {
   ScopedTempDir data_dir;
   ASSERT_TRUE(data_dir.CreateUniqueTempDir());
   const FilePath kDbFile = data_dir.path().AppendASCII("quota_manager.db");
   RegisterOrigins(kDbFile);
   RegisterOrigins(FilePath());
 }
+
+TEST_F(QuotaDatabaseTest, DumpQuotaTable) {
+  ScopedTempDir data_dir;
+  ASSERT_TRUE(data_dir.CreateUniqueTempDir());
+  const FilePath kDbFile = data_dir.path().AppendASCII("quota_manager.db");
+  DumpQuotaTable(kDbFile);
+  DumpQuotaTable(FilePath());
+}
+
+TEST_F(QuotaDatabaseTest, DumpAccessTable) {
+  ScopedTempDir data_dir;
+  ASSERT_TRUE(data_dir.CreateUniqueTempDir());
+  const FilePath kDbFile = data_dir.path().AppendASCII("quota_manager.db");
+  DumpAccessTable(kDbFile);
+  DumpAccessTable(FilePath());
+}
 }  // namespace quota