Adding methods that will be used by the quota management UI....

webkit/database/databases_table_unittest.cc

 // Copyright (c) 2009 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 "app/sql/connection.h"
 #include "app/sql/statement.h"
-#include "base/scoped_temp_dir.h"
 #include "base/string_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "webkit/database/databases_table.h"
 
-namespace webkit_database {
+namespace {
 
 class TestErrorDelegate : public sql::ErrorDelegate {
  public:
   virtual ~TestErrorDelegate() { }
   virtual int OnError(
       int error, sql::Connection* connection, sql::Statement* stmt) {
     return error;
   }
 };
 
+}  // namespace
+
+namespace webkit_database {
+
 static void CheckDetailsAreEqual(const DatabaseDetails& d1,
                                  const DatabaseDetails& d2) {
   EXPECT_EQ(d1.origin_identifier, d2.origin_identifier);
   EXPECT_EQ(d1.database_name, d2.database_name);
   EXPECT_EQ(d1.description, d2.description);
   EXPECT_EQ(d1.estimated_size, d2.estimated_size);
 }
 
 static bool DatabasesTableIsEmpty(sql::Connection* db) {
   sql::Statement statement(db->GetCachedStatement(
       SQL_FROM_HERE, "SELECT COUNT(*) FROM Databases"));
   return (statement.is_valid() && statement.Step() && !statement.ColumnInt(0));
 }
 
 TEST(DatabasesTableTest, TestIt) {
   // Initialize the 'Databases' table.
-  ScopedTempDir temp_dir;
   sql::Connection db;
 
   // Set an error delegate that will make all operations return false on error.
   scoped_refptr<TestErrorDelegate> error_delegate(new TestErrorDelegate());
   db.set_error_delegate(error_delegate);
 
   // Initialize the temp dir and the 'Databases' table.
-  EXPECT_TRUE(temp_dir.CreateUniqueTempDir());
-  EXPECT_TRUE(db.Open(temp_dir.path().Append(FILE_PATH_LITERAL("tracker.db"))));
+  EXPECT_TRUE(db.OpenInMemory());
   DatabasesTable databases_table(&db);
   EXPECT_TRUE(databases_table.Init());
 
   // The 'Databases' table should be empty.
   EXPECT_TRUE(DatabasesTableIsEmpty(&db));
 
   // Create the details for a databases.
   DatabaseDetails details_in1;
   DatabaseDetails details_out1;
   details_in1.origin_identifier = ASCIIToUTF16("origin1");
@@ skipping 50 matching lines @@
   EXPECT_TRUE(details_out_origin3.empty());
 
   // There should be only two databases with origin "origin1".
   std::vector<DatabaseDetails> details_out_origin1;
   EXPECT_TRUE(databases_table.GetAllDatabaseDetailsForOrigin(
       details_in1.origin_identifier, &details_out_origin1));
   EXPECT_EQ(size_t(2), details_out_origin1.size());
   CheckDetailsAreEqual(details_in1, details_out_origin1[0]);
   CheckDetailsAreEqual(details_in2, details_out_origin1[1]);
 
+  // Get the list of all origins: should be "origin1" and "origin2".
+  std::vector<string16> origins_out;
+  EXPECT_TRUE(databases_table.GetAllOrigins(&origins_out));
+  EXPECT_EQ(size_t(2), origins_out.size());
+  EXPECT_EQ(details_in1.origin_identifier, origins_out[0]);
+  EXPECT_EQ(details_in3.origin_identifier, origins_out[1]);
+
+  // Delete an origin and check that it's no longer in the table.
+  origins_out.clear();
+  EXPECT_TRUE(databases_table.DeleteOrigin(details_in3.origin_identifier));
+  EXPECT_TRUE(databases_table.GetAllOrigins(&origins_out));
+  EXPECT_EQ(size_t(1), origins_out.size());
+  EXPECT_EQ(details_in1.origin_identifier, origins_out[0]);
+
+  // Deleting an origin that doesn't have any record in this table should fail.
+  EXPECT_FALSE(databases_table.DeleteOrigin(ASCIIToUTF16("unknown_origin")));
+
   // Delete the details for 'db1' and check that they're no longer there.
   EXPECT_TRUE(databases_table.DeleteDatabaseDetails(
       details_in1.origin_identifier, details_in1.database_name));
   EXPECT_FALSE(databases_table.GetDatabaseDetails(
       details_in1.origin_identifier,
       details_in1.database_name,
       &details_out1));
 
   // Check that trying to delete a record that doesn't exist fails.
   EXPECT_FALSE(databases_table.DeleteDatabaseDetails(
       ASCIIToUTF16("unknown_origin"), ASCIIToUTF16("unknown_database")));
 }
 
 }  // namespace webkit_database