1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
6 #include "base/strings/string_util.h"
7 #include "base/strings/utf_string_conversions.h"
8 #include "sql/connection.h"
9 #include "sql/statement.h"
10 #include "sql/test/scoped_error_ignorer.h"
11 #include "testing/gtest/include/gtest/gtest.h"
12 #include "third_party/sqlite/sqlite3.h"
13 #include "webkit/browser/database/databases_table.h"
15 using base::ASCIIToUTF16;
17 namespace webkit_database {
19 static void CheckDetailsAreEqual(const DatabaseDetails& d1,
20 const DatabaseDetails& d2) {
21 EXPECT_EQ(d1.origin_identifier, d2.origin_identifier);
22 EXPECT_EQ(d1.database_name, d2.database_name);
23 EXPECT_EQ(d1.description, d2.description);
24 EXPECT_EQ(d1.estimated_size, d2.estimated_size);
27 static bool DatabasesTableIsEmpty(sql::Connection* db) {
28 sql::Statement statement(db->GetCachedStatement(
29 SQL_FROM_HERE, "SELECT COUNT(*) FROM Databases"));
30 return (statement.is_valid() && statement.Step() && !statement.ColumnInt(0));
33 TEST(DatabasesTableTest, TestIt) {
34 // Initialize the 'Databases' table.
37 sql::ScopedErrorIgnorer ignore_errors;
38 // TODO(shess): Suppressing SQLITE_CONSTRAINT because the code
39 // expects that and handles the resulting error. Consider revising
40 // the code to use INSERT OR IGNORE (which would not throw
41 // SQLITE_CONSTRAINT) and then check ChangeCount() to see if any
43 ignore_errors.IgnoreError(SQLITE_CONSTRAINT);
45 // Initialize the temp dir and the 'Databases' table.
46 EXPECT_TRUE(db.OpenInMemory());
47 DatabasesTable databases_table(&db);
48 EXPECT_TRUE(databases_table.Init());
50 // The 'Databases' table should be empty.
51 EXPECT_TRUE(DatabasesTableIsEmpty(&db));
53 // Create the details for a databases.
54 DatabaseDetails details_in1;
55 DatabaseDetails details_out1;
56 details_in1.origin_identifier = "origin1";
57 details_in1.database_name = ASCIIToUTF16("db1");
58 details_in1.description = ASCIIToUTF16("description_db1");
59 details_in1.estimated_size = 100;
61 // Updating details for this database should fail.
62 EXPECT_FALSE(databases_table.UpdateDatabaseDetails(details_in1));
63 EXPECT_FALSE(databases_table.GetDatabaseDetails(
64 details_in1.origin_identifier,
65 details_in1.database_name,
68 // Inserting details for this database should pass.
69 EXPECT_TRUE(databases_table.InsertDatabaseDetails(details_in1));
70 EXPECT_TRUE(databases_table.GetDatabaseDetails(
71 details_in1.origin_identifier,
72 details_in1.database_name,
74 EXPECT_EQ(1, databases_table.GetDatabaseID(details_in1.origin_identifier,
75 details_in1.database_name));
77 // Check that the details were correctly written to the database.
78 CheckDetailsAreEqual(details_in1, details_out1);
80 // Check that inserting a duplicate row fails.
81 EXPECT_FALSE(databases_table.InsertDatabaseDetails(details_in1));
83 // Insert details for another database with the same origin.
84 DatabaseDetails details_in2;
85 details_in2.origin_identifier = "origin1";
86 details_in2.database_name = ASCIIToUTF16("db2");
87 details_in2.description = ASCIIToUTF16("description_db2");
88 details_in2.estimated_size = 200;
89 EXPECT_TRUE(databases_table.InsertDatabaseDetails(details_in2));
90 EXPECT_EQ(2, databases_table.GetDatabaseID(details_in2.origin_identifier,
91 details_in2.database_name));
93 // Insert details for a third database with a different origin.
94 DatabaseDetails details_in3;
95 details_in3.origin_identifier = "origin2";
96 details_in3.database_name = ASCIIToUTF16("db3");
97 details_in3.description = ASCIIToUTF16("description_db3");
98 details_in3.estimated_size = 300;
99 EXPECT_TRUE(databases_table.InsertDatabaseDetails(details_in3));
100 EXPECT_EQ(3, databases_table.GetDatabaseID(details_in3.origin_identifier,
101 details_in3.database_name));
103 // There should be no database with origin "origin3".
104 std::vector<DatabaseDetails> details_out_origin3;
105 EXPECT_TRUE(databases_table.GetAllDatabaseDetailsForOriginIdentifier(
106 "origin3", &details_out_origin3));
107 EXPECT_TRUE(details_out_origin3.empty());
109 // There should be only two databases with origin "origin1".
110 std::vector<DatabaseDetails> details_out_origin1;
111 EXPECT_TRUE(databases_table.GetAllDatabaseDetailsForOriginIdentifier(
112 details_in1.origin_identifier, &details_out_origin1));
113 EXPECT_EQ(size_t(2), details_out_origin1.size());
114 CheckDetailsAreEqual(details_in1, details_out_origin1[0]);
115 CheckDetailsAreEqual(details_in2, details_out_origin1[1]);
117 // Get the list of all origins: should be "origin1" and "origin2".
118 std::vector<std::string> origins_out;
119 EXPECT_TRUE(databases_table.GetAllOriginIdentifiers(&origins_out));
120 EXPECT_EQ(size_t(2), origins_out.size());
121 EXPECT_EQ(details_in1.origin_identifier, origins_out[0]);
122 EXPECT_EQ(details_in3.origin_identifier, origins_out[1]);
124 // Delete an origin and check that it's no longer in the table.
126 EXPECT_TRUE(databases_table.DeleteOriginIdentifier(
127 details_in3.origin_identifier));
128 EXPECT_TRUE(databases_table.GetAllOriginIdentifiers(&origins_out));
129 EXPECT_EQ(size_t(1), origins_out.size());
130 EXPECT_EQ(details_in1.origin_identifier, origins_out[0]);
132 // Deleting an origin that doesn't have any record in this table should fail.
133 EXPECT_FALSE(databases_table.DeleteOriginIdentifier("unknown_origin"));
135 // Delete the details for 'db1' and check that they're no longer there.
136 EXPECT_TRUE(databases_table.DeleteDatabaseDetails(
137 details_in1.origin_identifier, details_in1.database_name));
138 EXPECT_FALSE(databases_table.GetDatabaseDetails(
139 details_in1.origin_identifier,
140 details_in1.database_name,
143 // Check that trying to delete a record that doesn't exist fails.
144 EXPECT_FALSE(databases_table.DeleteDatabaseDetails(
145 "unknown_origin", ASCIIToUTF16("unknown_database")));
147 ASSERT_TRUE(ignore_errors.CheckIgnoredErrors());
150 } // namespace webkit_database