OLD | NEW |
| (Empty) |
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. | |
4 | |
5 #include "base/bind.h" | |
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" | |
14 | |
15 using base::ASCIIToUTF16; | |
16 | |
17 namespace webkit_database { | |
18 | |
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); | |
25 } | |
26 | |
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)); | |
31 } | |
32 | |
33 TEST(DatabasesTableTest, TestIt) { | |
34 // Initialize the 'Databases' table. | |
35 sql::Connection db; | |
36 | |
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 | |
42 // changes were made. | |
43 ignore_errors.IgnoreError(SQLITE_CONSTRAINT); | |
44 | |
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()); | |
49 | |
50 // The 'Databases' table should be empty. | |
51 EXPECT_TRUE(DatabasesTableIsEmpty(&db)); | |
52 | |
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; | |
60 | |
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, | |
66 &details_out1)); | |
67 | |
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, | |
73 &details_out1)); | |
74 EXPECT_EQ(1, databases_table.GetDatabaseID(details_in1.origin_identifier, | |
75 details_in1.database_name)); | |
76 | |
77 // Check that the details were correctly written to the database. | |
78 CheckDetailsAreEqual(details_in1, details_out1); | |
79 | |
80 // Check that inserting a duplicate row fails. | |
81 EXPECT_FALSE(databases_table.InsertDatabaseDetails(details_in1)); | |
82 | |
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)); | |
92 | |
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)); | |
102 | |
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()); | |
108 | |
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]); | |
116 | |
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]); | |
123 | |
124 // Delete an origin and check that it's no longer in the table. | |
125 origins_out.clear(); | |
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]); | |
131 | |
132 // Deleting an origin that doesn't have any record in this table should fail. | |
133 EXPECT_FALSE(databases_table.DeleteOriginIdentifier("unknown_origin")); | |
134 | |
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, | |
141 &details_out1)); | |
142 | |
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"))); | |
146 | |
147 ASSERT_TRUE(ignore_errors.CheckIgnoredErrors()); | |
148 } | |
149 | |
150 } // namespace webkit_database | |
OLD | NEW |