OLD | NEW |
| (Empty) |
1 // Copyright 2014 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 <stddef.h> | |
6 | |
7 #include "base/bind.h" | |
8 #include "base/strings/string_util.h" | |
9 #include "base/strings/utf_string_conversions.h" | |
10 #include "sql/connection.h" | |
11 #include "sql/statement.h" | |
12 #include "sql/test/scoped_error_expecter.h" | |
13 #include "storage/browser/database/databases_table.h" | |
14 #include "testing/gtest/include/gtest/gtest.h" | |
15 #include "third_party/sqlite/sqlite3.h" | |
16 | |
17 using base::ASCIIToUTF16; | |
18 using storage::DatabaseDetails; | |
19 using storage::DatabasesTable; | |
20 | |
21 namespace content { | |
22 | |
23 static void CheckDetailsAreEqual(const DatabaseDetails& d1, | |
24 const DatabaseDetails& d2) { | |
25 EXPECT_EQ(d1.origin_identifier, d2.origin_identifier); | |
26 EXPECT_EQ(d1.database_name, d2.database_name); | |
27 EXPECT_EQ(d1.description, d2.description); | |
28 EXPECT_EQ(d1.estimated_size, d2.estimated_size); | |
29 } | |
30 | |
31 static bool DatabasesTableIsEmpty(sql::Connection* db) { | |
32 sql::Statement statement(db->GetCachedStatement( | |
33 SQL_FROM_HERE, "SELECT COUNT(*) FROM Databases")); | |
34 return (statement.is_valid() && statement.Step() && !statement.ColumnInt(0)); | |
35 } | |
36 | |
37 TEST(DatabasesTableTest, TestIt) { | |
38 // Initialize the 'Databases' table. | |
39 sql::Connection db; | |
40 | |
41 sql::test::ScopedErrorExpecter expecter; | |
42 // TODO(shess): Suppressing SQLITE_CONSTRAINT because the code | |
43 // expects that and handles the resulting error. Consider revising | |
44 // the code to use INSERT OR IGNORE (which would not throw | |
45 // SQLITE_CONSTRAINT) and then check ChangeCount() to see if any | |
46 // changes were made. | |
47 expecter.ExpectError(SQLITE_CONSTRAINT); | |
48 | |
49 // Initialize the temp dir and the 'Databases' table. | |
50 EXPECT_TRUE(db.OpenInMemory()); | |
51 DatabasesTable databases_table(&db); | |
52 EXPECT_TRUE(databases_table.Init()); | |
53 | |
54 // The 'Databases' table should be empty. | |
55 EXPECT_TRUE(DatabasesTableIsEmpty(&db)); | |
56 | |
57 // Create the details for a databases. | |
58 DatabaseDetails details_in1; | |
59 DatabaseDetails details_out1; | |
60 details_in1.origin_identifier = "origin1"; | |
61 details_in1.database_name = ASCIIToUTF16("db1"); | |
62 details_in1.description = ASCIIToUTF16("description_db1"); | |
63 details_in1.estimated_size = 100; | |
64 | |
65 // Updating details for this database should fail. | |
66 EXPECT_FALSE(databases_table.UpdateDatabaseDetails(details_in1)); | |
67 EXPECT_FALSE(databases_table.GetDatabaseDetails( | |
68 details_in1.origin_identifier, | |
69 details_in1.database_name, | |
70 &details_out1)); | |
71 | |
72 // Inserting details for this database should pass. | |
73 EXPECT_TRUE(databases_table.InsertDatabaseDetails(details_in1)); | |
74 EXPECT_TRUE(databases_table.GetDatabaseDetails( | |
75 details_in1.origin_identifier, | |
76 details_in1.database_name, | |
77 &details_out1)); | |
78 EXPECT_EQ(1, databases_table.GetDatabaseID(details_in1.origin_identifier, | |
79 details_in1.database_name)); | |
80 | |
81 // Check that the details were correctly written to the database. | |
82 CheckDetailsAreEqual(details_in1, details_out1); | |
83 | |
84 // Check that inserting a duplicate row fails. | |
85 EXPECT_FALSE(databases_table.InsertDatabaseDetails(details_in1)); | |
86 | |
87 // Insert details for another database with the same origin. | |
88 DatabaseDetails details_in2; | |
89 details_in2.origin_identifier = "origin1"; | |
90 details_in2.database_name = ASCIIToUTF16("db2"); | |
91 details_in2.description = ASCIIToUTF16("description_db2"); | |
92 details_in2.estimated_size = 200; | |
93 EXPECT_TRUE(databases_table.InsertDatabaseDetails(details_in2)); | |
94 EXPECT_EQ(2, databases_table.GetDatabaseID(details_in2.origin_identifier, | |
95 details_in2.database_name)); | |
96 | |
97 // Insert details for a third database with a different origin. | |
98 DatabaseDetails details_in3; | |
99 details_in3.origin_identifier = "origin2"; | |
100 details_in3.database_name = ASCIIToUTF16("db3"); | |
101 details_in3.description = ASCIIToUTF16("description_db3"); | |
102 details_in3.estimated_size = 300; | |
103 EXPECT_TRUE(databases_table.InsertDatabaseDetails(details_in3)); | |
104 EXPECT_EQ(3, databases_table.GetDatabaseID(details_in3.origin_identifier, | |
105 details_in3.database_name)); | |
106 | |
107 // There should be no database with origin "origin3". | |
108 std::vector<DatabaseDetails> details_out_origin3; | |
109 EXPECT_TRUE(databases_table.GetAllDatabaseDetailsForOriginIdentifier( | |
110 "origin3", &details_out_origin3)); | |
111 EXPECT_TRUE(details_out_origin3.empty()); | |
112 | |
113 // There should be only two databases with origin "origin1". | |
114 std::vector<DatabaseDetails> details_out_origin1; | |
115 EXPECT_TRUE(databases_table.GetAllDatabaseDetailsForOriginIdentifier( | |
116 details_in1.origin_identifier, &details_out_origin1)); | |
117 EXPECT_EQ(size_t(2), details_out_origin1.size()); | |
118 CheckDetailsAreEqual(details_in1, details_out_origin1[0]); | |
119 CheckDetailsAreEqual(details_in2, details_out_origin1[1]); | |
120 | |
121 // Get the list of all origins: should be "origin1" and "origin2". | |
122 std::vector<std::string> origins_out; | |
123 EXPECT_TRUE(databases_table.GetAllOriginIdentifiers(&origins_out)); | |
124 EXPECT_EQ(size_t(2), origins_out.size()); | |
125 EXPECT_EQ(details_in1.origin_identifier, origins_out[0]); | |
126 EXPECT_EQ(details_in3.origin_identifier, origins_out[1]); | |
127 | |
128 // Delete an origin and check that it's no longer in the table. | |
129 origins_out.clear(); | |
130 EXPECT_TRUE(databases_table.DeleteOriginIdentifier( | |
131 details_in3.origin_identifier)); | |
132 EXPECT_TRUE(databases_table.GetAllOriginIdentifiers(&origins_out)); | |
133 EXPECT_EQ(size_t(1), origins_out.size()); | |
134 EXPECT_EQ(details_in1.origin_identifier, origins_out[0]); | |
135 | |
136 // Deleting an origin that doesn't have any record in this table should fail. | |
137 EXPECT_FALSE(databases_table.DeleteOriginIdentifier("unknown_origin")); | |
138 | |
139 // Delete the details for 'db1' and check that they're no longer there. | |
140 EXPECT_TRUE(databases_table.DeleteDatabaseDetails( | |
141 details_in1.origin_identifier, details_in1.database_name)); | |
142 EXPECT_FALSE(databases_table.GetDatabaseDetails( | |
143 details_in1.origin_identifier, | |
144 details_in1.database_name, | |
145 &details_out1)); | |
146 | |
147 // Check that trying to delete a record that doesn't exist fails. | |
148 EXPECT_FALSE(databases_table.DeleteDatabaseDetails( | |
149 "unknown_origin", ASCIIToUTF16("unknown_database"))); | |
150 | |
151 ASSERT_TRUE(expecter.SawExpectedErrors()); | |
152 } | |
153 | |
154 } // namespace content | |
OLD | NEW |